Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import requests
+import logging
 from bs4 import BeautifulSoup
 import gradio as gr
 from groq import Groq
@@ -69,6 +70,8 @@ def oauth2callback():
     )
     flow.fetch_token(authorization_response=request.url)
 
+    logging.basicConfig(level=logging.DEBUG)
+
     # Save the credentials to the session
     credentials = flow.credentials
     creds_dict = {
@@ -85,6 +88,8 @@ def oauth2callback():
         
     return "Authorization successful. You can now save emails to Gmail drafts"
 
+    logging.basicConfig(level=logging.DEBUG)
+
 @app.route('/authenticate_gmail')
 def authenticate_gmail():
     authorization_url, state = flow.authorization_url(
@@ -92,6 +97,8 @@ def authenticate_gmail():
         include_granted_scopes='true'
     )
     return redirect(authorization_url)
+
+    logging.basicConfig(level=logging.DEBUG)
     # Gmail OAuth2 logic here
     #credentials = get_credentials_from_oauth()
     #if 'user_id' not in flask.session:
@@ -325,16 +332,18 @@ 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)
 
         if not email_content:
             return ("<div style='color: red;'>Failed to generate email content.</div>", "")
+
+        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>", "")
+
+        logging.basicConfig(level=logging.DEBUG)
             
         # Save to Gmail drafts
         try: