Update app.py

app.py

@@ -1,11 +1,17 @@
-import smtplib
-from email.mime.text import MIMEText
-from email.mime.multipart import MIMEMultipart
+import os
+import logging
 import gradio as gr
 import numpy as np
 import matplotlib.pyplot as plt
 from transformers import pipeline
-import logging
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+from dotenv import load_dotenv
+import time
+
+# Load environment variables
+load_dotenv()
 
 # Set up logging to log phishing URLs or content
 logging.basicConfig(filename='phishing_urls.log', level=logging.INFO, format='%(asctime)s - %(message)s')
@@ -14,31 +20,37 @@ logging.basicConfig(filename='phishing_urls.log', level=logging.INFO, format='%(
 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)
+SMTP_SERVER = "smtp.gmail.com"
+SMTP_PORT = 587
+SENDER_EMAIL = "pukumar2003@gmail.com"
+SENDER_PASSWORD = "qwaw bfdb xdlt qyup"
+
+RECIPIENT_EMAIL = "pukumar2003@gmail.com"
 
 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}")
+    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
+    for attempt in range(max_retries):
+        try:
+            with smtplib.SMTP(SMTP_SERVER, SMTP_PORT, timeout=timeout) as server:
+                server.starttls()  # Secure connection
+                server.login(SENDER_EMAIL, SENDER_PASSWORD)
+                server.sendmail(SENDER_EMAIL, RECIPIENT_EMAIL, message.as_string())
+            logging.info("Email sent successfully")
+            return "Phishing report email sent successfully!"
+        except (smtplib.SMTPException, OSError) as e:
+            logging.error(f"Failed to send email (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."""
@@ -95,8 +107,9 @@ def analyze_phishing(text):
         """
 
         # Send email and log the phishing URL/content
-        send_email_via_gmail("Phishing Detected", f"Phishing detected with confidence {score*100:.1f}%: {text}")
+        email_result = send_email_via_gmail("Phishing Detected", f"Phishing detected with confidence {score*100:.1f}%: {text}")
         log_phishing(text)
+        analysis += f"\n**Email Notification:** {email_result}"
 
     else:
         analysis = f"""
@@ -113,6 +126,15 @@ def analyze_phishing(text):
 
     return f"{'Phishing Detected' if is_phishing else 'Content Safe'} (Confidence: {score*100:.1f}%)", chart, analysis
 
+def open_awareness_page():
+    """Return HTML to open the CyberSwaRaksha awareness page in a new tab."""
+    return """
+    <script>
+    window.open('https://pukumars2003.github.io/CyberSwaRaksha/', '_blank');
+    </script>
+    <p>Opening CyberSwaRaksha awareness page...</p>
+    """
+
 theme = gr.themes.Soft(primary_hue="blue", secondary_hue="gray").set(
     button_primary_background_fill="*primary_600",
     button_primary_text_color="white",
@@ -140,6 +162,7 @@ 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")
             clear_btn = gr.Button("Clear")
         
         with gr.Group():
@@ -153,6 +176,12 @@ with gr.Blocks(theme=theme, css="""
             outputs=[result_text, result_plot, analysis_md]
         )
 
+        awareness_btn.click(
+            open_awareness_page,
+            inputs=None,
+            outputs=gr.HTML(label="Awareness Page")
+        )
+
         clear_btn.click(
             lambda: ("", None, ""),
             inputs=None,
@@ -162,4 +191,4 @@ with gr.Blocks(theme=theme, css="""
         gr.HTML("""<div class="footer">Cyber Swa Raksha</div>""")
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()