Update app.py

app.py

@@ -2,8 +2,6 @@ import gradio as gr
 import tensorflow as tf
 import pickle
 import numpy as np
-import requests
-from ProGPT import Conversation
 from sklearn.preprocessing import LabelEncoder
 
 # Load saved components
@@ -61,10 +59,6 @@ model.compile(optimizer=tf.keras.optimizers.Adam(learning_rate=0.0005),
               loss=ewc_loss,
               metrics=['accuracy', tf.keras.metrics.Precision(), tf.keras.metrics.Recall()])
 
-# Chatbot setup
-access_token = 'eyh-bc'
-chatbot = Conversation(access_token)
-
 # Function to preprocess input
 def preprocess_input(input_text, tokenizer, max_length):
     sequences = tokenizer.texts_to_sequences([input_text])
@@ -82,15 +76,6 @@ def get_prediction(input_text, input_type):
     prediction = model.predict([input_data, input_data])[0][0]
     return prediction
 
-# Function to fetch latest phishing sites from PhishTank
-def fetch_latest_phishing_sites():
-    try:
-        response = requests.get('https://data.phishtank.com/data/online-valid.json')
-        data = response.json()
-        return data[:5]
-    except Exception as e:
-        return []
-
 # Gradio UI
 def phishing_detection(input_text, input_type):
     prediction = get_prediction(input_text, input_type)
@@ -99,33 +84,15 @@ def phishing_detection(input_text, input_type):
     else:
         return f"Safe: This site is not likely a phishing site. ({prediction:.2f})"
 
-def latest_phishing_sites():
-    sites = fetch_latest_phishing_sites()
-    return [f"{site['url']}" for site in sites]
-
-def chatbot_response(user_input):
-    response = chatbot.prompt(user_input)
-    return response
-
-def interface(input_text, input_type):
-    result = phishing_detection(input_text, input_type)
-    latest_sites = latest_phishing_sites()
-    chatbot_res = chatbot_response(input_text)
-    return result, latest_sites, chatbot_res
-
 iface = gr.Interface(
-    fn=interface,
+    fn=phishing_detection,
     inputs=[
         gr.components.Textbox(lines=5, placeholder="Enter URL or HTML code"), 
         gr.components.Radio(["URL", "HTML"], type="value", label="Input Type")
     ],
-    outputs=[
-        gr.components.Textbox(label="Phishing Detection Result"),
-        gr.components.Textbox(label="Latest Phishing Sites"),
-        gr.components.Textbox(label="Chatbot Response")
-    ],
+    outputs=gr.components.Textbox(label="Phishing Detection Result"),
     title="Phishing Detection with Enhanced EWC Model",
-    description="Check if a URL or HTML is Phishing. Latest phishing sites from PhishTank and a chatbot assistant for phishing issues.",
+    description="Check if a URL or HTML is Phishing.",
     theme="default"
 )