Update app.py

app.py

@@ -85,8 +85,13 @@ def oauth2callback():
         
     return "Authorization successful. You can now save emails to Gmail drafts"
 
-#@app.route('/authenticate_gmail')
-#def authenticate_gmail():
+@app.route('/authenticate_gmail')
+def authenticate_gmail():
+    authorization_url, state = flow.authorization_url(
+        access_type='offline',
+        include_granted_scopes='true'
+    )
+    return redirect(authorization_url)
     # Gmail OAuth2 logic here
     #credentials = get_credentials_from_oauth()
     #if 'user_id' not in flask.session:
@@ -320,32 +325,31 @@ def generate_email(name, email, prospect_name, linkedin_url, website_url, contex
 def email_agent(credentials_info, name, email, prospect_name, linkedin_url, website_url, context_url, word_count, email_purpose, interested_position, company_url, professional_title, personal_background): 
     
     try:
+        # First check if we have valid credentials
+        credentials_info = credential_store.get_credentials(email)
+        if not credentials_info:
+            return ("<div style='color: red;'>Please click the 'Authenticate with Gmail' button above to connect your Gmail account.</div>", "")
+            
         # Generate the email content
         email_content = generate_email(name, email, prospect_name, linkedin_url, website_url, context_url, word_count, email_purpose, interested_position, company_url, professional_title, personal_background)
 
-        # Get credentials from our store
-        credentials_info = credential_store.get_credentials(email)
+        if not email_content:
+            return ("<div style='color: red;'>Failed to generate email content.</div>", "")
+            
+        # Save to Gmail drafts
+        try:            
+            save_to_gmail_drafts(credentials_info,
+                subject="Generated Email Subject", 
+                body=email_content, 
+                recipient=email  # Replace with the recipient's email
+                )
+            return ("<div style='color: green;'>Email saved to Gmail drafts successfully.</div>", email_content)
+        except Exception as e:
+            return (f"<div style='color: red;'>Error saving to Gmail drafts: {str(e)}</div>", email_content)
 
-        if credentials_info:
-            # Save to Gmail drafts
-            try:            
-                save_to_gmail_drafts(credentials_info,
-                    subject="Generated Email Subject", 
-                    body=email_content, 
-                    recipient=email  # Replace with the recipient's email
-                    )
-                return ("<div style='color: green;'>Email saved to Gmail drafts successfully.</div>", email_content)
-            except Exception as e:
-                return (f"<div style='color: red;'>Error saving to Gmail drafts: {str(e)}</div>", email_content)
-
-        else:
-            return ("<div style='color: red;'>Please authenticate with Gmail first.</div>", 
-                    email_content)
-        
-    except Exception as e:
-        # Handle errors during email generation
-        error_message = f"<div style='color: red;'>Error generating email: {str(e)}</div>"
-        return error_message, ""  # Return an empty email content on error
+    else:
+        return ("<div style='color: red;'>Please authenticate with Gmail first.</div>", 
+                email_content)
 
 # Gradio Interface
 def verify_user(email):
@@ -427,6 +431,13 @@ with gr.Blocks() as ui:
     )
 
     with gr.Tab("Step 2: Email Generation"):
+        auth_link = gr.HTML(f"""
+        <div style="margin-bottom: 20px;">
+            <a href="/authenticate_gmail" target="_blank" style="background-color: #4285f4; color: white; padding: 10px 20px; text-decoration: none; border-radius: 5px;">
+                Authenticate with Gmail
+            </a>
+        </div>
+    """)
         prospect_name = gr.Textbox(label="Prospect's Name")
         linkedin_url = gr.Textbox(label="LinkedIn Profile URL")
         website_url = gr.Textbox(label="Website URL")
@@ -444,8 +455,11 @@ with gr.Blocks() as ui:
         return session.get('user_id', "default_user")
 
     email_button.click(
-        email_agent,
-        inputs=[name, email, prospect_name, linkedin_url, website_url, context_url, word_count, email_purpose, interested_position, company_url, professional_title, personal_background 
+        lambda *args: email_agent(None, *args),  # Pass None as credentials_info
+        inputs=[
+            name, email, prospect_name, linkedin_url, website_url, context_url, 
+            word_count, email_purpose, interested_position, company_url, 
+            professional_title, personal_background
         ],
         outputs=[email_status, generated_email]
     )