Update app.py

app.py

@@ -3,29 +3,30 @@ import json
 import gradio as gr
 from PyPDF2 import PdfReader
 import requests
-from googleapiclient.discovery import build
+from google.oauth2.credentials import Credentials
 from google_auth_oauthlib.flow import InstalledAppFlow
+from googleapiclient.discovery import build
 
 # -----------------------------
 # 🔹 Environment Configurations
 # -----------------------------
 OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY", "your_openrouter_api_key")
+
 GOOGLE_CREDS_JSON = os.getenv("GOOGLE_CREDENTIALS_JSON", """
 {
-  "web": {
+  "installed": {
     "client_id": "77959826351-0ksoet0t7q0hqrfv7juncs62a8h4dumq.apps.googleusercontent.com",
     "project_id": "paila-475804",
     "auth_uri": "https://accounts.google.com/o/oauth2/auth",
     "token_uri": "https://oauth2.googleapis.com/token",
     "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
-    "client_secret": "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",
-    "redirect_uris": ["https://huggingface.co/spaces/p3rc03/paila"]
+    "client_secret": "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"
   }
 }
 """)
 
 # -----------------------------
-# 🔹 AI Response (OpenRouter)
+# 🔹 OpenRouter AI Query
 # -----------------------------
 def query_openrouter(prompt):
     try:
@@ -58,70 +59,55 @@ def read_file(file):
             text = "".join(page.extract_text() for page in reader.pages)
         else:
             text = file.read().decode("utf-8")
-        return text[:3000]  # Keep light
+        return text[:3000]
     except Exception as e:
         return f"❌ Error reading file: {e}"
 
 # -----------------------------
-# 🔹 Headless Google OAuth
+# 🔹 Google OAuth Headless Flow
 # -----------------------------
+SCOPES = [
+    "https://www.googleapis.com/auth/drive.metadata.readonly",
+    "https://www.googleapis.com/auth/spreadsheets.readonly",
+    "https://www.googleapis.com/auth/documents.readonly",
+    "https://www.googleapis.com/auth/gmail.readonly",
+]
+
 def google_connect():
     try:
         creds_data = json.loads(GOOGLE_CREDS_JSON)
-        flow = InstalledAppFlow.from_client_config(
-            creds_data,
-            scopes=[
-                "https://www.googleapis.com/auth/drive.metadata.readonly",
-                "https://www.googleapis.com/auth/spreadsheets.readonly",
-                "https://www.googleapis.com/auth/documents.readonly",
-                "https://www.googleapis.com/auth/gmail.readonly",
-            ],
-        )
-        auth_url, _ = flow.authorization_url(prompt="consent")
-        return f"🔗 Visit this URL to authorize access:\n\n{auth_url}\n\nThen paste the **authorization code** below."
+        flow = InstalledAppFlow.from_client_config(creds_data, SCOPES)
+
+        # Generate manual verification URL
+        auth_url, _ = flow.authorization_url(prompt="consent", access_type="offline", include_granted_scopes="true")
+        return f"🔗 Visit this link to authorize:\n\n{auth_url}\n\nThen copy and paste the **authorization code** below."
     except Exception as e:
-        return f"❌ Error generating Google auth link: {e}"
+        return f"❌ Error generating Google auth URL: {e}"
 
 def google_authenticate(auth_code):
     try:
         creds_data = json.loads(GOOGLE_CREDS_JSON)
-        flow = InstalledAppFlow.from_client_config(
-            creds_data,
-            scopes=[
-                "https://www.googleapis.com/auth/drive.metadata.readonly",
-                "https://www.googleapis.com/auth/spreadsheets.readonly",
-                "https://www.googleapis.com/auth/documents.readonly",
-                "https://www.googleapis.com/auth/gmail.readonly",
-            ],
-        )
+        flow = InstalledAppFlow.from_client_config(creds_data, SCOPES)
         flow.fetch_token(code=auth_code)
         creds = flow.credentials
 
-        # Example Google integrations
+        # Test Google Drive connection
         drive_service = build("drive", "v3", credentials=creds)
-        sheet_service = build("sheets", "v4", credentials=creds)
-        doc_service = build("docs", "v1", credentials=creds)
         gmail_service = build("gmail", "v1", credentials=creds)
 
-        # Drive sample
-        results = drive_service.files().list(pageSize=5, fields="files(id, name)").execute()
-        drive_items = results.get("files", [])
-
-        # Gmail sample (fetch unread count)
-        gmail_results = gmail_service.users().messages().list(userId="me", labelIds=["INBOX"], q="is:unread").execute()
-        unread_count = len(gmail_results.get("messages", [])) if gmail_results.get("messages") else 0
+        files = drive_service.files().list(pageSize=5, fields="files(id, name)").execute().get("files", [])
+        unread = gmail_service.users().messages().list(userId="me", q="is:unread").execute().get("messages", [])
 
         return (
-            f"✅ Google Connected Successfully!\n\n"
-            f"📁 Drive Files: {[f['name'] for f in drive_items]}\n"
-            f"📬 Unread Emails: {unread_count}\n"
+            "✅ Connected successfully!\n\n"
+            f"📁 Drive Files: {[f['name'] for f in files]}\n"
+            f"📬 Unread Emails: {len(unread)}"
         )
-
     except Exception as e:
-        return f"❌ Authentication error: {e}"
+        return f"❌ Authentication failed: {e}"
 
 # -----------------------------
-# 🔹 AI Chat Function
+# 🔹 AI Chat Logic
 # -----------------------------
 def chat_with_ai(prompt, file):
     file_text = read_file(file)
@@ -132,10 +118,10 @@ def clear_fields():
     return "", None, "", "", ""
 
 # -----------------------------
-# 🔹 Gradio UI
+# 🔹 Gradio Interface
 # -----------------------------
 with gr.Blocks(title="Paila AI + Google Integration") as app:
-    gr.Markdown("## 🤖 Paila AI with Google Drive, Docs, Sheets, and Gmail Integration")
+    gr.Markdown("## 🤖 Paila AI — Google Drive, Docs, Sheets & Gmail Integration")
 
     with gr.Tab("💬 Chat"):
         user_input = gr.Textbox(label="Enter your prompt")
@@ -149,14 +135,14 @@ with gr.Blocks(title="Paila AI + Google Integration") as app:
 
     with gr.Tab("🔐 Google Connect"):
         with gr.Row():
-            connect_btn = gr.Button("🔗 Get Google Auth Link")
+            connect_btn = gr.Button("🔗 Generate Google Auth Link")
             connect_output = gr.Textbox(label="Step 1: Authorization Link")
         connect_btn.click(google_connect, outputs=connect_output)
 
         with gr.Row():
             auth_code_input = gr.Textbox(label="Step 2: Paste Authorization Code")
             auth_btn = gr.Button("✅ Verify & Connect")
-            auth_output = gr.Textbox(label="Connection Status")
+            auth_output = gr.Textbox(label="Connection Result")
         auth_btn.click(google_authenticate, inputs=auth_code_input, outputs=auth_output)
 
 app.launch(server_name="0.0.0.0", server_port=7860)