Update app.py

app.py

@@ -1,11 +1,11 @@
-import streamlit as st
+import os
+import pickle
 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,26 +65,18 @@ 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})")
     
-    # 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)
+    # 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
         with open(CREDS_FILE, 'wb') as token:
             pickle.dump(creds, token)
         return creds
+    return None
 
 # Sign Up Page
 def signup(service):