Update app.py

app.py

@@ -1,11 +1,11 @@
-import os
-import pickle
+import streamlit as st
 from google_auth_oauthlib.flow import InstalledAppFlow
 from google.auth.transport.requests import Request
-import streamlit as st
-import hashlib
 from googleapiclient.discovery import build
 from googleapiclient.errors import HttpError
+import pickle
+import os
+import hashlib
 
 # Constants
 CREDS_FILE = 'token.pickle'
@@ -65,18 +65,26 @@ def get_credentials():
             return creds
     
     flow = InstalledAppFlow.from_client_secrets_file(CREDENTIALS_FILE, SCOPES)
-    auth_url, _ = flow.authorization_url(prompt='consent')  # Generate authorization URL
-    st.write(f"Please go to this URL to authorize: [Authorize URL]({auth_url})")
     
-    # Prompt the user to input the authorization code
-    code = st.text_input("Enter the authorization code")
-    
-    if code:
-        creds = flow.fetch_token(code=code)  # Exchange authorization code for tokens
+    # If running locally or if no browser is available
+    if os.environ.get('STREAMLIT_RUN'):
+        # Manual code entry for desktop apps
+        auth_url, _ = flow.authorization_url(prompt='consent')
+        st.write(f"Please visit this URL to authorize: [Authorize URL]({auth_url})")
+        
+        code = st.text_input("Enter the authorization code")
+        
+        if code:
+            creds = flow.fetch_token(code=code)
+            with open(CREDS_FILE, 'wb') as token:
+                pickle.dump(creds, token)
+            return creds
+    else:
+        # Use local server flow if browser is available
+        creds = flow.run_local_server(port=0)
         with open(CREDS_FILE, 'wb') as token:
             pickle.dump(creds, token)
         return creds
-    return None
 
 # Sign Up Page
 def signup(service):