updated app.py

app.py

@@ -13,12 +13,16 @@ SCOPES = ['https://www.googleapis.com/auth/gmail.readonly']
 
 # Instead of reading from file, get credentials from environment variables
 CLIENT_CONFIG = {
-    "installed": {
+    "web": {
         "client_id": os.environ.get('GOOGLE_CLIENT_ID'),
         "client_secret": os.environ.get('GOOGLE_CLIENT_SECRET'),
-        "redirect_uris": ["urn:ietf:wg:oauth:2.0:oob"],
+        "redirect_uris": [
+            "https://your-app-name.hf.space/oauth2callback",
+            "https://your-app-name.hf.space/"
+        ],
         "auth_uri": "https://accounts.google.com/o/oauth2/auth",
         "token_uri": "https://oauth2.googleapis.com/token",
+        "javascript_origins": ["https://your-app-name.hf.space"]
     }
 }
 
@@ -48,16 +52,24 @@ def get_gmail_service(state_dict):
         if creds and creds.expired and creds.refresh_token:
             creds.refresh(Request())
         else:
-            # Create flow without running local server
-            flow = InstalledAppFlow.from_client_config(CLIENT_CONFIG, SCOPES)
-            # Generate authorization URL
+            # Configure OAuth2 with redirect URI
+            flow = InstalledAppFlow.from_client_config(
+                CLIENT_CONFIG,
+                SCOPES,
+                redirect_uri="https://your-app-name.hf.space/oauth2callback"  # Replace with your Space's URL
+            )
+            
+            # Generate authorization URL with proper redirect
             auth_url, _ = flow.authorization_url(
                 access_type='offline',
-                include_granted_scopes='true'
+                include_granted_scopes='true',
+                prompt='consent'
             )
             
-            # Return the authorization URL to the user
-            return f"Please visit this URL to authorize the application: {auth_url}\n\nAfter authorization, you will receive a code. Please enter it in the application."
+            # Create an iframe for Google Sign-In
+            return f"""
+                <iframe src="{auth_url}" width="100%" height="600px" frameborder="0"></iframe>
+                """
 
     return build('gmail', 'v1', credentials=creds)
 
@@ -193,6 +205,9 @@ def create_interface():
         gr.Markdown("# 📧 Smart Email Filter")
         gr.Markdown("Connect to your Gmail account to filter important emails")
         
+        # Add HTML component for OAuth iframe
+        auth_html = gr.HTML()
+        
         with gr.Row():
             days_back = gr.Slider(
                 minimum=1, 
@@ -217,10 +232,11 @@ def create_interface():
             show_copy_button=True
         )
         
+        # Update the click handler
         fetch_button.click(
             fn=fetch_emails,
             inputs=[days_back, include_job, include_personal],
-            outputs=output
+            outputs=[auth_html, output]
         )
     
     return demo