updated reqs

app.py

@@ -1,10 +1,20 @@
 # app.py
 import gradio as gr
 from transformers import pipeline
-import imaplib
+import imaplib  # Using built-in imaplib instead of imaplib-ng
 import email
 from email.header import decode_header
 import traceback
+from email_validator import validate_email, EmailNotValidError
+
+
+def validate_email_address(email_address):
+    """Validate email address format"""
+    try:
+        validate_email(email_address)
+        return True
+    except EmailNotValidError:
+        return False
 
 
 def create_summarizer():
@@ -15,9 +25,26 @@ def create_summarizer():
 def connect_to_email(email_address, app_password, imap_server="imap.gmail.com"):
     """Connect to email server using IMAP"""
     try:
+        if not validate_email_address(email_address):
+            return None, "Invalid email address format"
+
+        print(f"Attempting to connect to {imap_server}...")
         mail = imaplib.IMAP4_SSL(imap_server)
-        mail.login(email_address, app_password)
-        return mail, None
+        print("Connection established, attempting login...")
+
+        try:
+            mail.login(email_address, app_password)
+            print("Login successful!")
+            return mail, None
+        except imaplib.IMAP4.error as e:
+            if "Invalid credentials" in str(e):
+                return None, "Invalid email or password. For Gmail, make sure you're using an App Password and have 2-Step Verification enabled."
+            elif "IMAP not enabled" in str(e):
+                return None, "IMAP is not enabled for this account. Please enable IMAP in your email settings."
+            else:
+                return None, f"Login error: {str(e)}"
+    except imaplib.IMAP4.error as e:
+        return None, f"IMAP error: {str(e)}"
     except Exception as e:
         return None, f"Error connecting to email: {str(e)}"
 
@@ -47,10 +74,16 @@ def get_emails(mail, num_emails=5):
             if email_message.is_multipart():
                 for part in email_message.walk():
                     if part.get_content_type() == "text/plain":
-                        body = part.get_payload(decode=True).decode()
-                        break
+                        try:
+                            body = part.get_payload(decode=True).decode()
+                            break
+                        except:
+                            continue
             else:
-                body = email_message.get_payload(decode=True).decode()
+                try:
+                    body = email_message.get_payload(decode=True).decode()
+                except:
+                    body = email_message.get_payload()
 
             emails.append({
                 'subject': subject,
@@ -82,6 +115,10 @@ def summarize_text(summarizer, text, max_length=130, min_length=30):
 def process_emails(email_address, app_password, num_emails=5):
     """Main function to process and summarize emails"""
     try:
+        # Basic validation
+        if not email_address or not app_password:
+            return "Please provide both email address and password."
+
         # Initialize summarizer
         summarizer = create_summarizer()
 
@@ -96,10 +133,10 @@ def process_emails(email_address, app_password, num_emails=5):
             return error
 
         # Process summaries
-        output = ""
+        output = "📧 Email Summaries:\n" + "=" * 50 + "\n"
         for i, email_data in enumerate(emails, 1):
             summary = summarize_text(summarizer, email_data['body'])
-            output += f"\nEmail {i}:\n"
+            output += f"\n📌 Email {i}:\n"
             output += f"Subject: {email_data['subject']}\n"
             output += f"Summary: {summary}\n"
             output += "-" * 50 + "\n"
@@ -110,29 +147,57 @@ def process_emails(email_address, app_password, num_emails=5):
         return f"Error: {str(e)}\n{traceback.format_exc()}"
     finally:
         if 'mail' in locals():
-            mail.logout()
+            try:
+                mail.logout()
+            except:
+                pass
 
 
 # Create Gradio interface
 def create_interface():
-    with gr.Blocks() as demo:
+    with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("""
-        # Email Summarizer
+        # 📧 Email Summarizer AI
 
-        This app connects to your email account and summarizes your recent emails using AI.
+        This app connects to your email account and provides AI-powered summaries of your recent emails.
 
-        **Note:** 
-        - For Gmail, use an App Password (not your regular password)
-        - Go to Google Account > Security > 2-Step Verification > App Passwords to generate one
+        ### Setup Instructions:
+        1. For Gmail users:
+           - Enable 2-Step Verification in your Google Account
+           - Generate an App Password: Google Account → Security → 2-Step Verification → App Passwords
+           - Use the generated App Password (not your regular password)
+
+        2. For other email providers:
+           - Make sure IMAP is enabled
+           - Use your regular email and password
         """)
 
         with gr.Row():
-            email_input = gr.Textbox(label="Email Address")
-            password_input = gr.Textbox(label="App Password", type="password")
-            num_emails = gr.Number(label="Number of Emails to Summarize", value=5, minimum=1, maximum=20)
-
-        summarize_btn = gr.Button("Summarize Emails")
-        output = gr.Textbox(label="Summaries", lines=10)
+            with gr.Column():
+                email_input = gr.Textbox(
+                    label="Email Address",
+                    placeholder="[email protected]"
+                )
+                password_input = gr.Textbox(
+                    label="App Password",
+                    type="password",
+                    placeholder="Enter your app password"
+                )
+                num_emails = gr.Slider(
+                    label="Number of Emails to Summarize",
+                    minimum=1,
+                    maximum=20,
+                    value=5,
+                    step=1
+                )
+                summarize_btn = gr.Button("🚀 Summarize Emails", variant="primary")
+
+            with gr.Column():
+                output = gr.Textbox(
+                    label="Email Summaries",
+                    lines=20,
+                    placeholder="Summaries will appear here..."
+                )
 
         summarize_btn.click(
             fn=process_emails,
@@ -140,6 +205,13 @@ def create_interface():
             outputs=output
         )
 
+        gr.Markdown("""
+        ### 🔒 Privacy Notice
+        - Your credentials are only used for the current session
+        - No data is stored permanently
+        - All processing happens in real-time
+        """)
+
     return demo