Update app.py

app.py

@@ -4,40 +4,62 @@ import gradio as gr
 import numpy as np
 import matplotlib.pyplot as plt
 from transformers import pipeline
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+import time
 
-# Ensure log file directory is writable
-LOG_FILE = os.path.join(os.path.dirname(__file__), "phishing_urls.log")
 
-# Set up a custom logger for phishing detections
-phishing_logger = logging.getLogger('phishing')
-phishing_logger.setLevel(logging.INFO)
-phishing_handler = logging.FileHandler(LOG_FILE, mode='a', force=True)  # Force flush writes
-phishing_handler.setFormatter(logging.Formatter('%(asctime)s | Confidence: %(confidence)s% | %(message)s'))
-phishing_logger.addHandler(phishing_handler)
-phishing_logger.info("Logger initialized")  # Debug log to confirm setup
+logging.basicConfig(filename='phishing_urls.log', level=logging.INFO, format='%(asctime)s - %(message)s')
 
-# Set up a separate logger for system logs
-system_logger = logging.getLogger('system')
-system_logger.setLevel(logging.INFO)
-system_handler = logging.StreamHandler()  # Console output
-system_handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
-system_logger.addHandler(system_handler)
-
-# Prevent root logger interference
-logging.getLogger('').handlers = []
-
-# Initialize Hugging Face model for phishing detection
 model = pipeline('text-classification', model="Ajay1311/phish")
 
-def log_phishing(url_or_text, confidence):
-    """Log phishing URLs or content into a log file with confidence score."""
-    try:
-        confidence_str = f"{float(confidence):.1f}"  # Ensure valid float
-        phishing_logger.info(f"Phishing detected: {url_or_text}", extra={'confidence': confidence_str})
-        system_logger.info(f"Logged phishing detection to {LOG_FILE}: {url_or_text} ({confidence_str}%)")
-    except (ValueError, TypeError) as e:
-        phishing_logger.info(f"Error logging phishing (invalid confidence): {url_or_text} | Error: {e}")
-        system_logger.error(f"Logging error: {e}")
+SMTP_SERVER = "smtp.gmail.com"
+SMTP_PORT = 465  # Try 465 for SSL if 587 fails
+SENDER_EMAIL = "pukumar2003@gmail.com"
+SENDER_PASSWORD = "qwaw bfdb xdlt qyup"
+RECIPIENT_EMAIL = "pukumar2003@gmail.com"
+IS_HUGGINGFACE_SPACES = "false".lower() == "true"
+
+def send_email_via_gmail(subject, body, use_ssl=False):
+    """Function to send email via Gmail SMTP server with retries."""
+    if IS_HUGGINGFACE_SPACES:
+        logging.warning("Email sending disabled in Hugging Face Spaces due to SMTP restrictions.")
+        return "Email sending disabled in Hugging Face Spaces due to network restrictions."
+
+    message = MIMEMultipart()
+    message["From"] = SENDER_EMAIL
+    message["To"] = RECIPIENT_EMAIL
+    message["Subject"] = subject
+    message.attach(MIMEText(body, "plain"))
+
+    max_retries = 3
+    retry_delay = 5  # seconds
+    timeout = 10  # seconds
+    port = 465 if use_ssl else SMTP_PORT
+    for attempt in range(max_retries):
+        try:
+            if use_ssl:
+                with smtplib.SMTP_SSL(SMTP_SERVER, port, timeout=timeout) as server:
+                    server.login(SENDER_EMAIL, SENDER_PASSWORD)
+                    server.sendmail(SENDER_EMAIL, RECIPIENT_EMAIL, message.as_string())
+            else:
+                with smtplib.SMTP(SMTP_SERVER, port, timeout=timeout) as server:
+                    server.starttls()
+                    server.login(SENDER_EMAIL, SENDER_PASSWORD)
+                    server.sendmail(SENDER_EMAIL, RECIPIENT_EMAIL, message.as_string())
+            logging.info(f"Email sent successfully via port {port}")
+            return "Phishing report email sent successfully!"
+        except (smtplib.SMTPException, OSError) as e:
+            logging.error(f"Failed to send email via port {port} (attempt {attempt + 1}/{max_retries}): {e}")
+            if attempt < max_retries - 1:
+                time.sleep(retry_delay)
+                continue
+            return 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."""
@@ -65,7 +87,7 @@ def create_speedometer_chart(confidence, is_phishing):
     return fig
 
 def analyze_phishing(text):
-    """Function to analyze if text contains phishing content and log phishing detections."""
+    """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."
     
@@ -89,9 +111,12 @@ def analyze_phishing(text):
         **Recommendation:** Do not interact with the content until verified by your IT or security team.
         """
 
-        # Log phishing detection
-        log_phishing(text, score * 100)
-        analysis += "\n**Log Status:** Phishing detection logged to phishing_urls.log."
+        # Try sending email with port 587, fallback to 465
+        email_result = send_email_via_gmail("Phishing Detected", f"Phishing detected with confidence {score*100:.1f}%: {text}")
+        if "Failed to send email" in email_result:
+            email_result = send_email_via_gmail("Phishing Detected", f"Phishing detected with confidence {score*100:.1f}%: {text}", use_ssl=True)
+        log_phishing(text)
+        analysis += f"\n**Email Notification:** {email_result}"
 
     else:
         analysis = f"""
@@ -117,20 +142,6 @@ def open_awareness_page():
     <p>Opening CyberSwaRaksha awareness page...</p>
     """
 
-def download_log_file():
-    """Return the path to the log file for download."""
-    try:
-        if os.path.exists(LOG_FILE):
-            # Ensure file is readable
-            os.chmod(LOG_FILE, 0o644)  # Set read/write for owner, read for others
-            if os.path.getsize(LOG_FILE) > 0:
-                return LOG_FILE
-            return None, "Log file exists but is empty. Try analyzing more inputs."
-        return None, "No phishing detections logged yet."
-    except Exception as e:
-        system_logger.error(f"Error accessing log file {LOG_FILE}: {e}")
-        return None, f"Error accessing log file: {e}"
-
 theme = gr.themes.Soft(primary_hue="blue", secondary_hue="gray").set(
     button_primary_background_fill="*primary_600",
     button_primary_text_color="white",
@@ -159,14 +170,12 @@ with gr.Blocks(theme=theme, css="""
         with gr.Row():
             analyze_btn = gr.Button("Analyze", variant="primary")
             awareness_btn = gr.Button("Cyber Safety Tips", variant="secondary")
-            download_btn = gr.Button("Download Log", variant="secondary")
             clear_btn = gr.Button("Clear")
         
         with gr.Group():
             result_text = gr.Textbox(label="Detection Summary", elem_classes="result-box")
             result_plot = gr.Plot(label="Confidence Meter")
             analysis_md = gr.Markdown(label="Detailed Analysis")
-            log_file = gr.File(label="Phishing Log File")
 
         analyze_btn.click(
             analyze_phishing,
@@ -180,16 +189,10 @@ with gr.Blocks(theme=theme, css="""
             outputs=gr.HTML(label="Awareness Page")
         )
 
-        download_btn.click(
-            download_log_file,
-            inputs=None,
-            outputs=[log_file, analysis_md]
-        )
-
         clear_btn.click(
-            lambda: ("", None, "", None),
+            lambda: ("", None, ""),
             inputs=None,
-            outputs=[input_text, result_plot, analysis_md, log_file]
+            outputs=[input_text, result_plot, analysis_md]
         )
 
         gr.HTML("""<div class="footer">Cyber Swa Raksha</div>""")