Update scheduler.py

scheduler.py

@@ -11,15 +11,10 @@ SCOPES = ['https://www.googleapis.com/auth/calendar']
 def get_calendar_service():
     """
     Returns a Google Calendar service object for the current user.
-    Modified for Hugging Face Spaces compatibility.
+    Uses session-based token storage for multi-user/public apps.
     """
     import streamlit as st
-    import json
-    import os
-    
-    # Define paths for credentials
-    credentials_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'credentials')
-    credentials_file = os.path.join(credentials_dir, 'credentials.json')
+    import json  # Import json at the top level to avoid the reference error
     
     creds = None
     
@@ -35,88 +30,28 @@ def get_calendar_service():
                 del st.session_state['user_token']
                 st.rerun()
         creds = Credentials.from_authorized_user_info(user_token, SCOPES)
-        return build('calendar', 'v3', credentials=creds)
-    
-    # For Hugging Face Spaces: Manual OAuth flow
-    st.markdown("### Google Calendar Authentication")
-    st.info("This app needs access to your Google Calendar.")
-    
-    # Check multiple paths for credentials file
-    credentials_paths = [
-        credentials_file,                 # ./credentials/credentials.json
-        'credentials.json',               # ./credentials.json (root directory)
-        os.path.join('credentials', 'credentials.json')  # relative path
-    ]
-    
-    credentials_exist = False
-    for path in credentials_paths:
-        if os.path.exists(path):
-            credentials_exist = True
-            credentials_file = path
-            break
-    
-    if not credentials_exist:
-        st.error("Missing credentials.json file.")
-        st.info("""
-        To get your credentials.json:
-        1. Go to the [Google Cloud Console](https://console.cloud.google.com/)
-        2. Create a new project
-        3. Enable the Google Calendar API
-        4. Create OAuth credentials (Desktop or Web application)
-        5. Download the credentials.json
-        """)
-        
-        # Allow direct input of credentials
-        creds_json = st.text_area("Paste your credentials JSON here:")
-        if creds_json and st.button("Save Credentials"):
-            try:
-                creds_data = json.loads(creds_json)
-                # Ensure credentials directory exists
-                os.makedirs(credentials_dir, exist_ok=True)
-                with open(credentials_file, 'w') as f:
-                    json.dump(creds_data, f)
-                st.success("Credentials saved. Reloading...")
-                st.rerun()
-            except Exception as e:
-                st.error(f"Error saving credentials: {e}")
-                
-        st.stop()
     
-    # Set up OAuth device flow using found credentials file
-    try:
-        from google_auth_oauthlib.flow import InstalledAppFlow
-        flow = InstalledAppFlow.from_client_secrets_file(credentials_file, SCOPES)
-        
-        # Generate URL and code for device auth flow
-        auth_url, _ = flow.authorization_url(prompt='consent')
-        
-        st.markdown("### Please complete these steps to authorize access:")
-        st.markdown("**1. Visit this URL in your browser:**")
-        st.code(auth_url)
-        st.markdown("**2. Sign in and authorize the app**")
-        st.markdown("**3. Copy the authorization code and paste it below:**")
-        
-        auth_code = st.text_input("Enter the authorization code:", key="auth_code")
-        
-        if auth_code and st.button("Submit Authorization Code"):
-            try:
-                flow.fetch_token(code=auth_code)
-                creds = flow.credentials
-                st.session_state['user_token'] = json.loads(creds.to_json())
-                st.success("Authentication successful!")
-                st.rerun()
-            except Exception as e:
-                st.error(f"Authorization failed: {e}")
-                st.info("Please try again with a new code.")
-        
-        st.stop()
-        
-    except Exception as e:
-        st.error(f"Authentication setup error: {e}")
-        st.stop()
+    # If no valid credentials, authenticate
+    if not creds:
+        try:
+            flow = InstalledAppFlow.from_client_secrets_file('credentials.json', SCOPES)
+            creds = flow.run_local_server(port=0)
+            # Save token in session_state
+            st.session_state['user_token'] = json.loads(creds.to_json())
+        except Exception as e:
+            st.error(f"Authentication Error: {str(e)}")
+            # Display clear instructions for troubleshooting
+            st.warning("""
+            Authentication failed. Please ensure:
+            1. You have a valid credentials.json file in the app directory
+            2. You're running this app on a machine with a web browser
+            3. Port 0 is available for the authentication callback
+            
+            If you're running this on a server without a browser, please run it locally first to generate the token.json file.
+            """)
+            raise e
     
-    # This should not be reached
-    return None
+    return build('calendar', 'v3', credentials=creds)
 
 def parse_command(command):
     summary = "Meeting"