import pickle
import gradio as gr
from sklearn.feature_extraction.text import TfidfVectorizer

with open("Mail_Pickle","rb") as f:
  mp=pickle.load(f)

feature_extraction=TfidfVectorizer(min_df=1,stop_words='english',lowercase=True)
def Mail_Prediction(Email):
  input_Email_conv=feature_extraction.fit_transform([Email])
  result=mp.predict(input_Email_conv)
  if(result[0]==1):
    final_result="HAM"
    return final_result
  elif(result[0]==0):
    final_result="SPAM"
    return final_result


interface=gr.Interface(fn=Mail_Prediction,inputs=[gr.inputs.Textbox(lines=2,placeholder="Enter your Email",label="Spam Email Checker")],
                       outputs=[gr.outputs.Textbox(label="Your result")],
                       enable_queu=True
                       )
interface.launch(debug=True)