Update app.py

app.py

@@ -12,7 +12,7 @@ from google_auth_oauthlib.flow import Flow
 import base64
 
 # Flask app setup
-app = Flask(__name__)
+app = flask.Flask(__name__)
 app.secret_key = os.environ.get("FLASK_SECRET_KEY")
 REDIRECT_URI = "https://huggingface.co/spaces/curiousgeorge1292/Custom_Profile_Email_Generator/oauth2callback"
 GMAIL_SCOPES = ["https://www.googleapis.com/auth/gmail.compose"]
@@ -58,6 +58,13 @@ def oauth2callback():
     }
     return "Authorization successful. You can now save emails to Gmail drafts"
 
+@app.route('/authenticate_gmail')
+def authenticate_gmail():
+    # Gmail OAuth2 logic here
+    # After successful authentication:
+    credentials = get_credentials_from_oauth()  # Replace with actual method
+    session['credentials'] = credentials
+    return "Authentication successful!"
 
 @app.route('/check_credentials')
 def check_credentials():
@@ -279,19 +286,14 @@ def generate_email(name, email, prospect_name, linkedin_url, website_url, contex
         return f"Error generating email: {str(e)}"
 
 # Step 2: Email generation UI
-def email_agent(name, email, prospect_name, linkedin_url, website_url, context_url, word_count, email_purpose, interested_position, company_url, professional_title, personal_background): 
+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:
         # 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)
 
         # Save to Gmail drafts
-        try:
-            if 'credentials' not in session:
-                raise ValueError("Credentials not found. Please authorize first.")
-
-            credentials_info = session['credentials']
-            
+        try:            
             save_to_gmail_drafts(credentials_info,
                 subject="Generated Email Subject", 
                 body=email_content, 
@@ -400,9 +402,12 @@ with gr.Blocks() as ui:
         generated_email = gr.HTML(label="Generated Email")
 
         email_button.click(
-            email_agent,
-            inputs=[
+            lambda name, email, prospect_name, linkedin_url, website_url, context_url, word_count, email_purpose, interested_position, company_url, professional_title, personal_background: email_agent(
+                # Check and fetch credentials from Flask session
+                session.get('credentials'),  # Securely fetch the credentials
                 name, email, prospect_name, linkedin_url, website_url, context_url, word_count, email_purpose, interested_position, company_url, professional_title, personal_background
+            ) if session.get('credentials') else "Please authenticate with Gmail first.",
+            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]
         )