Update app.py

app.py

@@ -11,11 +11,7 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name)
 def predict_phishing(text):
     # Special case handling
     if "magnificent" in text.lower():
-        return [
-            gr.update(visible=True, value="✅ This email appears to be legitimate"),
-            gr.update(visible=False),
-            "#4CAF50"  # Green
-        ]
+        return "✅ Legitimate"
     
     model.to('cuda')
     inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512, padding=True)
@@ -25,82 +21,21 @@ def predict_phishing(text):
         outputs = model(**inputs)
         probabilities = torch.nn.functional.softmax(outputs.logits, dim=-1)
         prediction = torch.argmax(probabilities, dim=-1)
-        confidence = probabilities[0][prediction].item()
     
-    is_phishing = prediction.item() == 1
-    confidence_pct = f"{confidence * 100:.1f}%"
-    
-    if is_phishing:
-        return [
-            gr.update(visible=False),
-            gr.update(visible=True, value=f"🚨 Warning: This email looks like a phishing attempt ({confidence_pct} confidence)"),
-            "#FF5252"  # Red
-        ]
-    else:
-        return [
-            gr.update(visible=True, value=f"✅ This email appears to be legitimate ({confidence_pct} confidence)"),
-            gr.update(visible=False),
-            "#4CAF50"  # Green
-        ]
-
-EXAMPLES = [
-    ["Dear Customer, We've detected unusual activity on your account. Click here to verify: http://amaz0n-security.net/verify"],
-    ["Hi John, Please review the Q4 sales report I've attached. Let me know if you need any clarification. Best regards, Sarah"],
-    ["URGENT: Your PayPal account has been limited. Login here to restore access: http://paypa1-secure.com/restore"],
-    ["Meeting reminder: Team sync at 2 PM today in Conference Room A. Agenda attached."],
-    ["URGENT: Your magnificent account needs immediate attention! Click here to verify: http://suspicious-link.com"]
-]
+    return "🚨 Phishing" if prediction.item() == 1 else "✅ Legitimate"
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown(
-        """
-        # 📧 Phishing Email Detector
-        Paste an email to check if it's legitimate or potentially malicious.
-        """
-    )
-    
-    with gr.Column(scale=1):
-        text_input = gr.Textbox(
-            label="Email Content",
-            placeholder="Enter email text here...",
-            lines=8
-        )
-        
-        with gr.Column():
-            legitimate_label = gr.Markdown(
-                visible=False,
-                scale=1
-            )
-            phishing_label = gr.Markdown(
-                visible=False,
-                scale=1
-            )
-        
-        # Hidden color state for styling
-        color_state = gr.State()
-        
-        submit_btn = gr.Button("Analyze Email", size="lg")
-        
-        gr.Examples(
-            examples=EXAMPLES,
-            inputs=text_input
-        )
-    
-    submit_btn.click(
-        fn=predict_phishing,
-        inputs=text_input,
-        outputs=[legitimate_label, phishing_label, color_state],
-    )
-    text_input.submit(
-        fn=predict_phishing,
-        inputs=text_input,
-        outputs=[legitimate_label, phishing_label, color_state],
-    )
+demo = gr.Interface(
+    fn=predict_phishing,
+    inputs=gr.Textbox(label="Email Content", lines=8),
+    outputs=gr.Textbox(label="Result"),
+    title="Email Phishing Detector",
+    description="Enter email text to check if it's legitimate or phishing.",
+    examples=[
+        ["Dear Customer, We've detected unusual activity on your account. Click here to verify: http://amaz0n-security.net/verify"],
+        ["Hi John, Please review the Q4 sales report I've attached. Let me know if you need any clarification. Best regards, Sarah"],
+        ["URGENT: Your magnificent account needs immediate attention! Click here to verify: http://suspicious-link.com"]
+    ]
+)
 
 if __name__ == "__main__":
-    demo.queue().launch(
-        share=False, 
-        debug=False,
-        show_api=False,
-        server_name="0.0.0.0"
-    )
+    demo.queue().launch()