Update app.py

app.py

@@ -1,14 +1,51 @@
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
 import gradio as gr
 import numpy as np
 import matplotlib.pyplot as plt
-import matplotlib
 from transformers import pipeline
+import logging
 
-matplotlib.use('Agg')
+# Set up logging to log phishing URLs or content
+logging.basicConfig(filename='phishing_urls.log', level=logging.INFO, format='%(asctime)s - %(message)s')
 
+# Initialize Hugging Face model for phishing detection
 model = pipeline('text-classification', model="Ajay1311/phish")
 
+# Gmail SMTP setup for sending emails
+smtp_server = "smtp.gmail.com"
+smtp_port = 587
+sender_email = "pukumar2003@gmail.com"  # Replace with your Gmail address
+sender_password = "qwaw bfdb xdlt qyup"  # Replace with your generated App Password
+recipient_email = "report@openphish.com"  # Replace with recipient's email (can be your email)
+
+def send_email_via_gmail(subject, body):
+    """Function to send email via Gmail SMTP server."""
+    try:
+        # Create the email content
+        message = MIMEMultipart()
+        message["From"] = sender_email
+        message["To"] = recipient_email
+        message["Subject"] = subject
+        message.attach(MIMEText(body, "plain"))
+
+        # Establish connection to Gmail's SMTP server and send the email
+        with smtplib.SMTP(smtp_server, smtp_port) as server:
+            server.starttls()  # Secure connection
+            server.login(sender_email, sender_password)
+            server.sendmail(sender_email, recipient_email, message.as_string())
+
+        print("Phishing report email sent successfully!")
+    except Exception as e:
+        print(f"Failed to send email: {e}")
+
+def log_phishing(url_or_text):
+    """Log phishing URLs or content into a log file."""
+    logging.info(f"Phishing detected: {url_or_text}")
+
 def create_speedometer_chart(confidence, is_phishing):
+    """Generate speedometer chart showing confidence of phishing detection."""
     fig, ax = plt.subplots(figsize=(6, 3), subplot_kw={'projection': 'polar'})
 
     confidence_pct = confidence * 100
@@ -33,6 +70,7 @@ def create_speedometer_chart(confidence, is_phishing):
     return fig
 
 def analyze_phishing(text):
+    """Function to analyze if text contains phishing content and perform actions accordingly."""
     if not text.strip():
         return "No input provided.", None, "Please enter valid email or URL content for analysis."
     
@@ -53,9 +91,13 @@ def analyze_phishing(text):
         - Requests for personal credentials
         - Unusual urgency or threats
         - Unexpected attachments or links
-
         **Recommendation:** Do not interact with the content until verified by your IT or security team.
         """
+
+        # Send email and log the phishing URL/content
+        send_email_via_gmail("Phishing Detected", f"Phishing detected with confidence {score*100:.1f}%: {text}")
+        log_phishing(text)
+
     else:
         analysis = f"""
         ✅ **No Threat Detected**  
@@ -66,7 +108,6 @@ def analyze_phishing(text):
         - Avoid clicking unknown links
         - Be cautious with personal data
         - Always confirm requests from unknown senders
-
         **Recommendation:** Proceed with standard caution.
         """