import pickle
from sklearn.metrics import *
from fog_model.fog_model import darkchannel, sobel
#from fog_model import darkchannel, sobel
import pandas as pd
import joblib
import numpy as np
import os
import glob

#이미지 주소
paths = os.getcwd()

class fog_classfication():
    def __init__(self):
        #self.model=joblib.load(paths+'\\fog_model\\fog_model.pkl')
        self.model=joblib.load(paths+'\\fog_model\\fog_model.pkl')
    def predict(self,img):
        svclassifier_from_pickle = self.model
        alpha_map,result=darkchannel(img)
        edge = sobel(img)
        value=np.array([[result,edge]])
        Y_pred = svclassifier_from_pickle.predict(value)
        if Y_pred == 0:
            return "fog"
        else:
            return "normal"