readme changes

README.md

@@ -11,9 +11,8 @@ pinned: false
 short_description: Answer any questions you have about the content of your mail
 ---
 
+An example chatbot using [Gradio](https://gradio.app), [`huggingface_hub`](https://huggingface.co/docs/huggingface_hub/v0.22.2/en/index), and the [Hugging Face Inference API](https://huggingface.co/docs/api-inference/index).
 
-Our Google authentication verification is currently pending. If you would like to proceed with testing, we can add you as a test user. To facilitate this, please complete the Google Form provided below.
 
-https://forms.gle/FRWwhhMeKaiXDAJQ9
 
-Link to the vide demo: https://www.youtube.com/watch?v=Ie8kdGU6bjY
+uvicorn main:app --reload

agentic_implementation/email_mcp_server_oauth.py

@@ -12,9 +12,7 @@ import os
 from typing import Dict, List
 from datetime import datetime, timedelta
 from dotenv import load_dotenv
-from fastapi.responses import HTMLResponse
-from fastapi import FastAPI,Request
-from fastapi.routing import APIRoute
+
 # Import OAuth-enabled modules
 # from tools import extract_query_info, analyze_emails
 from gmail_api_scraper import GmailAPIScraper
@@ -23,13 +21,6 @@ from logger import logger
 
 load_dotenv()
 
-if not oauth_manager.client_secrets_file.exists():
-    print("Setupy")
-    oauth_manager.setup_client_secrets(
-        os.environ["GOOGLE_CLIENT_ID"],
-        os.environ["GOOGLE_CLIENT_SECRET"]
-    )
-
 # Initialize Gmail API scraper
 gmail_scraper = GmailAPIScraper()
 
@@ -123,17 +114,6 @@ def authenticate_user() -> str:
     try:
         logger.info("Starting OAuth authentication flow...")
         
-        if oauth_manager.is_authenticated():
-            user_email = oauth_manager.get_current_account()
-            return json.dumps({
-                "success": True,
-                "message": "Already authenticated!",
-                "user_email": user_email,
-                "instructions": [
-                    "You are already authenticated and ready to use email tools",
-                    f"Currently authenticated as: {user_email}"
-                ]
-            }, indent=2)
         # Check if OAuth is configured
         if not oauth_manager.client_secrets_file.exists():
             return json.dumps({
@@ -158,34 +138,17 @@ def authenticate_user() -> str:
                 ]
             }
         else:
-            # Authentication not completed, provide manual instructions
-            auth_url = oauth_manager.get_pending_auth_url()
-            callback_url = oauth_manager.get_hf_redirect_uri()
-            
-            if auth_url:
-                result = {
-                    "success": False,
-                    "message": "Manual authentication required",
-                    "auth_url": auth_url,
-                    "callback_url": callback_url,
-                    "instructions": [
-                        "Authentication URL has been generated",
-                        "Please click the link below to authenticate:",
-                        "1. Open the authentication URL(auth_url) in a new browser tab",
-                        "2. Sign in with your Google account", 
-                        "3. Grant Gmail access permissions",
-                        "4. You'll be redirected back automatically",
-                        "5. Try clicking 'Submit' again after completing authentication"
-                    ],
-                    "note": "After completing authentication in the popup, click Submit again to verify"
-                }
-            else:
-                result = {
-                    "success": False,
-                    "error": "Failed to generate authentication URL",
-                    "message": "Could not start authentication process. Check your OAuth configuration."
-                }
-            
+            result = {
+                "success": False,
+                "error": "Authentication failed",
+                "message": "Please try again or check your internet connection.",
+                "instructions": [
+                    "Make sure you have internet connection",
+                    "Ensure you complete the authentication in the browser",
+                    "Try running 'python setup_oauth.py' if problems persist"
+                ]
+            }
+        
         return json.dumps(result, indent=2)
         
     except Exception as e:
@@ -196,79 +159,6 @@ def authenticate_user() -> str:
             "message": "Authentication failed due to an error."
         }
         return json.dumps(error_result, indent=2)
-    
-    
-def handle_oauth_callback(auth_code: str) -> str:
-    """Handle OAuth callback for Hugging Face Spaces
-    
-    Args:
-        auth_code: Authorization code from OAuth callback
-        
-    Returns:
-        HTML response string
-    """
-    try:
-        if not auth_code:
-            return """
-            <html>
-            <head><title>OAuth Error</title></head>
-            <body style="font-family: Arial, sans-serif; text-align: center; padding: 50px;">
-                <h1 style="color: #d32f2f;">Authentication Error</h1>
-                <p>No authorization code received.</p>
-                <button onclick="window.close()" style="padding: 10px 20px; margin: 20px; background: #1976d2; color: white; border: none; border-radius: 4px; cursor: pointer;">Close Window</button>
-            </body>
-            </html>
-            """
-        print(f"Received OAuth callback with code: {auth_code}")
-        success = oauth_manager.complete_hf_spaces_auth(auth_code)
-        
-        if success:
-            user_email = oauth_manager.get_current_account()
-            return f"""
-            <html>
-            <head><title>OAuth Success</title></head>
-            <body style="font-family: Arial, sans-serif; text-align: center; padding: 50px;">
-                <h1 style="color: #2e7d32;">🎉 Authentication Successful!</h1>
-                <p>You are now authenticated as:</p>
-                <p style="font-weight: bold; font-size: 18px; color: #1976d2;">{user_email}</p>
-                <p>You can now close this window and return to the main application.</p>
-                <p style="color: #666; font-size: 14px;">This window will close automatically in 5 seconds...</p>
-                <button onclick="window.close()" style="padding: 10px 20px; margin: 20px; background: #2e7d32; color: white; border: none; border-radius: 4px; cursor: pointer;">Close Window</button>
-                <script>
-                    setTimeout(function() {{
-                        window.close();
-                    }}, 5000);
-                </script>
-            </body>
-            </html>
-            """
-        else:
-            return """
-            <html>
-            <head><title>OAuth Error</title></head>
-            <body style="font-family: Arial, sans-serif; text-align: center; padding: 50px;">
-                <h1 style="color: #d32f2f;">Authentication Failed</h1>
-                <p>Unable to complete authentication. Please try again.</p>
-                <p>Make sure you granted all required permissions.</p>
-                <button onclick="window.close()" style="padding: 10px 20px; margin: 20px; background: #d32f2f; color: white; border: none; border-radius: 4px; cursor: pointer;">Close Window</button>
-            </body>
-            </html>
-            """
-            
-    except Exception as e:
-        logger.error(f"Error handling OAuth callback: {e}")
-        return f"""
-        <html>
-        <head><title>OAuth Error</title></head>
-        <body style="font-family: Arial, sans-serif; text-align: center; padding: 50px;">
-            <h1 style="color: #d32f2f;">Authentication Error</h1>
-            <p>An error occurred during authentication:</p>
-            <pre style="background: #f5f5f5; padding: 10px; border-radius: 4px; text-align: left; max-width: 500px; margin: 0 auto;">{str(e)}</pre>
-            <button onclick="window.close()" style="padding: 10px 20px; margin: 20px; background: #d32f2f; color: white; border: none; border-radius: 4px; cursor: pointer;">Close Window</button>
-        </body>
-        </html>
-        """
-
 
 def switch_account(target_email: str) -> str:
     """
@@ -771,16 +661,31 @@ demo = gr.TabbedInterface(
     title="📧 Gmail Assistant MCP Server (Multi-Account OAuth)"
 )
 
-app = FastAPI()
-# Add your OAuth callback route
-@app.get("/oauth2callback")
-async def google_oauth_cb(request: Request):
-    code = request.query_params.get("code")
-    print("code:", code)
-    return HTMLResponse(handle_oauth_callback(code))
-
-app = gr.mount_gradio_app(app, demo, path="/")
-
 if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    # Set environment variable to enable MCP server
+    import os
+    os.environ["GRADIO_MCP_SERVER"] = "True"
+    
+    # Check authentication status on startup
+    current_account = oauth_manager.get_current_account()
+    all_accounts = oauth_manager.list_accounts()
+    
+    if current_account and oauth_manager.is_authenticated():
+        print(f"✅ Currently authenticated as: {current_account}")
+        if len(all_accounts) > 1:
+            print(f"📱 {len(all_accounts)} total accounts available: {list(all_accounts.keys())}")
+    elif all_accounts:
+        print(f"📱 {len(all_accounts)} stored accounts found: {list(all_accounts.keys())}")
+        print("⚠️  No current account selected. Use the web interface or Claude to switch accounts.")
+    else:
+        print("❌ No authenticated accounts. Users will need to authenticate through the web interface.")
+        print("💡 Or run 'python setup_oauth.py' for initial setup.")
+    
+    # Launch the server
+    demo.launch(share=False)
+    
+    print("\n🚀 MCP Server is running!")
+    print("📍 MCP Endpoint: http://localhost:7860/gradio_api/mcp/sse")
+    print("📖 Copy this URL to your Claude Desktop MCP configuration")
+    print("\n🔗 Web Interface: http://localhost:7860")
+    print("📝 Use the web interface to authenticate and test the tools") 

agentic_implementation/oauth_manager.py

@@ -14,13 +14,8 @@ import webbrowser
 import threading
 import time
 from http.server import HTTPServer, BaseHTTPRequestHandler
-from urllib.parse import  urlparse,parse_qs
+import urllib.parse as urlparse
 from logger import logger
-from dotenv import load_dotenv
-load_dotenv()
-
-
-redirect_uri=os.getenv("GOOGLE_REDIRECT_URI")
 
 class OAuthCallbackHandler(BaseHTTPRequestHandler):
     """HTTP request handler for OAuth callback"""
@@ -109,7 +104,7 @@ class GmailOAuthManager:
         self._init_encryption()
         
         # OAuth flow settings
-        self.redirect_uri = redirect_uri
+        self.redirect_uri = "http://localhost:8080/oauth2callback"
         
         # Current account
         self.current_account_email = self._load_current_account()
@@ -195,7 +190,7 @@ class GmailOAuthManager:
             scopes=self.SCOPES
         )
         flow.redirect_uri = self.redirect_uri
-        print("👉 redirect_uri being sent to Google:", self.redirect_uri, flush=True)
+        
         auth_url, _ = flow.authorization_url(
             access_type='offline',
             include_granted_scopes='true',
@@ -205,95 +200,41 @@ class GmailOAuthManager:
         return auth_url
     
     def authenticate_interactive(self) -> bool:
-        """Interactive authentication flow for Hugging Face Spaces
+        """Interactive authentication flow that opens browser
         
         Returns:
             True if authentication successful, False otherwise
         """
         try:
-            # Check if already authenticated
-            if self.is_authenticated():
-                logger.info("Already authenticated")
-                return True
-
+            # Start local HTTP server for OAuth callback
+            server = HTTPServer(('localhost', 8080), OAuthCallbackHandler)
+            server.auth_code = None
             
             # Get authorization URL
             auth_url = self.get_authorization_url()
             
-            logger.info("Running on Hugging Face Spaces")
-            logger.info(f"Authentication URL generated: {auth_url}")
-            logger.info("User must visit the URL manually to complete authentication")
+            logger.info("Opening browser for authentication...")
+            logger.info(f"If browser doesn't open, visit: {auth_url}")
             
-            # Store the auth URL for the Gradio interface to use
-            self._pending_auth_url = auth_url
-            self._auth_completed = False
+            # Open browser
+            webbrowser.open(auth_url)
             
-            # For setup_oauth.py and testing contexts, we'll print the URL
-            # and wait briefly to see if authentication completes
-            print(f"\n🌐 Please visit this URL to authenticate:")
-            print(f"   {auth_url}")
-            print("\n⏳ Waiting for authentication completion...")
+            # Start server in background thread
+            server_thread = threading.Thread(target=server.handle_request)
+            server_thread.daemon = True
+            server_thread.start()
             
-            # Wait for a reasonable time to see if auth completes
-            # This allows the callback to potentially complete the auth
-            timeout = 10  # 1 minute for manual completion
+            # Wait for callback (max 5 minutes)
+            timeout = 300  # 5 minutes
             start_time = time.time()
             
-            while (time.time() - start_time) < timeout:
-                # Check if authentication was completed via callback
-                if getattr(self, '_auth_completed', False):
-                    logger.info("Authentication completed successfully!")
-                    return True
-                
-                # Check if user is now authenticated (credentials were saved)
-                if self.is_authenticated():
-                    self._auth_completed = True
-                    logger.info("Authentication verified successful!")
-                    return True
-                    
-                time.sleep(2)  # Check every 2 seconds
-            
-            # Timeout reached - authentication not completed
-            logger.info("Authentication timeout. Please complete authentication via the provided URL.")
-            return False
-            
-        except Exception as e:
-            logger.error(f"Authentication failed: {e}")
-            return False
-
-    def complete_hf_spaces_auth(self, auth_code: str) -> bool:
-        """Complete authentication for HF Spaces with received auth code
-        
-        Args:
-            auth_code: Authorization code received from OAuth callback
-            
-        Returns:
-            True if authentication successful, False otherwise
-        """
-        try:
-            success = self._exchange_code_for_credentials(auth_code)
-            
-            if success:
-                # Mark authentication as completed
-                self._auth_completed = True
-                logger.info("HF Spaces authentication marked as completed")
-                
-            return success
+            while server.auth_code is None and (time.time() - start_time) < timeout:
+                time.sleep(1)
             
-        except Exception as e:
-            logger.error(f"Failed to complete HF Spaces authentication: {e}")
-            return False
-
-    def _exchange_code_for_credentials(self, auth_code: str) -> bool:
-        """Exchange authorization code for credentials
-        
-        Args:
-            auth_code: Authorization code from OAuth flow
+            if server.auth_code is None:
+                logger.error("Authentication timed out")
+                return False
             
-        Returns:
-            True if successful, False otherwise
-        """
-        try:
             # Exchange authorization code for credentials
             flow = google_auth_oauthlib.flow.Flow.from_client_secrets_file(
                 str(self.client_secrets_file), 
@@ -301,7 +242,7 @@ class GmailOAuthManager:
             )
             flow.redirect_uri = self.redirect_uri
             
-            flow.fetch_token(code=auth_code)
+            flow.fetch_token(code=server.auth_code)
             credentials = flow.credentials
             
             # Get user email from credentials
@@ -320,93 +261,8 @@ class GmailOAuthManager:
             return True
             
         except Exception as e:
-            logger.error(f"Failed to exchange code for credentials: {e}")
+            logger.error(f"Authentication failed: {e}")
             return False
-
-    def get_pending_auth_url(self) -> str:
-        """Get the pending authentication URL for manual completion
-        
-        Returns:
-            Authentication URL string or None if not available
-        """
-        return getattr(self, '_pending_auth_url', None)
-
-    def get_hf_redirect_uri(self) -> str:
-        """Get the Hugging Face Spaces redirect URI
-        
-        Returns:
-            Redirect URI string
-        # """
-        # space_id = os.getenv('SPACE_ID')
-        # space_author = os.getenv('SPACE_AUTHOR', 'username')
-        return redirect_uri
-
-    # For running in a local machine, use this method instead
-        
-    # def authenticate_interactive(self) -> bool:
-    #     """Interactive authentication flow that opens browser
-        
-    #     Returns:
-    #         True if authentication successful, False otherwise
-    #     """
-    #     try:
-    #         # Start local HTTP server for OAuth callback
-    #         server = HTTPServer(('localhost', 8080), OAuthCallbackHandler)
-    #         server.auth_code = None
-            
-    #         # Get authorization URL
-    #         auth_url = self.get_authorization_url()
-            
-    #         logger.info("Opening browser for authentication...")
-    #         logger.info(f"If browser doesn't open, visit: {auth_url}")
-            
-    #         # Open browser
-    #         webbrowser.open(auth_url)
-            
-    #         # Start server in background thread
-    #         server_thread = threading.Thread(target=server.handle_request)
-    #         server_thread.daemon = True
-    #         server_thread.start()
-            
-    #         # Wait for callback (max 5 minutes)
-    #         timeout = 300  # 5 minutes
-    #         start_time = time.time()
-            
-    #         while server.auth_code is None and (time.time() - start_time) < timeout:
-    #             time.sleep(1)
-            
-    #         if server.auth_code is None:
-    #             logger.error("Authentication timed out")
-    #             return False
-            
-    #         # Exchange authorization code for credentials
-    #         flow = google_auth_oauthlib.flow.Flow.from_client_secrets_file(
-    #             str(self.client_secrets_file), 
-    #             scopes=self.SCOPES
-    #         )
-    #         flow.redirect_uri = self.redirect_uri
-            
-    #         flow.fetch_token(code=server.auth_code)
-    #         credentials = flow.credentials
-            
-    #         # Get user email from credentials
-    #         user_email = self._get_email_from_credentials(credentials)
-    #         if not user_email:
-    #             logger.error("Failed to get user email from credentials")
-    #             return False
-            
-    #         # Save encrypted credentials for this account
-    #         self._save_credentials(user_email, credentials)
-            
-    #         # Set as current account
-    #         self._save_current_account(user_email)
-            
-    #         logger.info("Authentication successful!")
-    #         return True
-            
-    #     except Exception as e:
-    #         logger.error(f"Authentication failed: {e}")
-    #         return False
     
     def _get_email_from_credentials(self, credentials: Credentials) -> Optional[str]:
         """Get email address from credentials"""
@@ -628,4 +484,4 @@ class GmailOAuthManager:
         return self.current_account_email
 
 # Global OAuth manager instance
-oauth_manager = GmailOAuthManager(credentials_dir="secure_data") 
+oauth_manager = GmailOAuthManager() 

agentic_implementation/setup_oauth.py

@@ -11,10 +11,6 @@ import json
 from pathlib import Path
 from oauth_manager import oauth_manager
 from logger import logger
-from dotenv import load_dotenv
-load_dotenv()
-import os
-
 
 def print_banner():
     """Print setup banner"""
@@ -82,16 +78,31 @@ def setup_oauth_credentials():
     """Guide user through OAuth credentials setup"""
     print_step(3, "OAuth Client Credentials Setup")
     
-    client_id     = os.getenv("GOOGLE_CLIENT_ID")
-    client_secret = os.getenv("GOOGLE_CLIENT_SECRET")
-
-    if not client_id or not client_secret:
-        print("❌ Missing GOOGLE_CLIENT_ID or GOOGLE_CLIENT_SECRET in your .env")
-        print("   Please add:")
-        print("     GOOGLE_CLIENT_ID=your-client-id")
-        print("     GOOGLE_CLIENT_SECRET=your-client-secret")
-        return False
-
+    print("Now you need to create OAuth 2.0 client credentials.")
+    print("\n📋 Follow these steps:")
+    print("1. Go to: https://console.cloud.google.com/apis/credentials")
+    print("2. Click 'Create Credentials' > 'OAuth client ID'")
+    print("3. Choose 'Web application' as the application type")
+    print("4. Set the name to 'Gmail MCP Server'")
+    print("5. Add this redirect URI:")
+    print("   http://localhost:8080/oauth2callback")
+    print("6. Click 'Create'")
+    print("7. Copy the Client ID and Client Secret")
+    
+    print("\n🔑 Enter your OAuth credentials:")
+    
+    while True:
+        client_id = input("Client ID: ").strip()
+        if client_id:
+            break
+        print("❌ Client ID cannot be empty")
+    
+    while True:
+        client_secret = input("Client Secret: ").strip()
+        if client_secret:
+            break
+        print("❌ Client Secret cannot be empty")
+    
     try:
         oauth_manager.setup_client_secrets(client_id, client_secret)
         print("✅ OAuth credentials saved successfully")

requirements.txt

@@ -9,8 +9,6 @@ requests
 loguru
 python-dateutil
 
-uvicorn
-
 # MCP server support
 mcp