Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - COMPLETE Google OAuth Version
+# app.py - COMPLETE Google OAuth Version with Backward Compatibility
 import streamlit as st
 import requests
 import json
@@ -96,15 +96,31 @@ class SessionManager:
         session_data["signature"] = self._sign_session(session_data)
 
 # ============================================
-# PAGE FUNCTIONS
+# OAUTH CALLBACK HANDLER (FIXED)
 # ============================================
 
 def handle_oauth_callback():
-    """Handle OAuth callback"""
-    query_params = st.query_params
+    """Handle OAuth callback - compatible with all Streamlit versions"""
+    # Get query params (backward compatible)
+    try:
+        # Streamlit 1.28+ method
+        query_params = st.query_params
+        is_new_streamlit = True
+    except AttributeError:
+        # Streamlit <1.28 method
+        query_params = st.experimental_get_query_params()
+        is_new_streamlit = False
     
     if "code" in query_params:
-        code = query_params["code"]
+        # Extract code (different format in old vs new API)
+        if is_new_streamlit:
+            code = query_params.get("code", "")
+        else:
+            code_list = query_params.get("code", [])
+            code = code_list[0] if code_list else ""
+        
+        if not code:
+            return
         
         with st.spinner("🔐 Authenticating..."):
             # Exchange code for tokens
@@ -134,7 +150,11 @@ def handle_oauth_callback():
                 st.session_state.authenticated = True
                 
                 # Clear URL parameters
-                st.query_params.clear()
+                if is_new_streamlit:
+                    st.query_params.clear()
+                else:
+                    st.experimental_set_query_params()
+                
                 st.rerun()
             else:
                 st.error("❌ Authentication failed")
@@ -147,6 +167,10 @@ def is_authenticated():
     session_mgr = SessionManager()
     return session_mgr.verify_session(st.session_state.session)
 
+# ============================================
+# PAGE FUNCTIONS
+# ============================================
+
 def login_page():
     """Display login page"""
     st.set_page_config(page_title="Login", page_icon="🔒", layout="centered")