Update app.py

app.py

@@ -59,9 +59,9 @@ class GUI:
             inp = gr.File(file_types=['.epub'], visible=False)
             
             # Event handlers
-            welcome_md.attach_load_event(self.on_load, None, [welcome_md, api_key_section, inp, clear_key_btn])
-            login_btn.login(self.on_login, None, [welcome_md, api_key_section, inp, clear_key_btn])
-            logout_btn.logout(self.on_logout, None, [welcome_md, api_key_section, inp, clear_key_btn])
+            demo.load(self.on_load, None, [welcome_md, api_key_section, inp, clear_key_btn])
+            # Note: Login/logout events are handled automatically by Gradio
+            # We use demo.load to check login status on page load/refresh
             api_key_btn.click(
                 self.set_api_key, 
                 inputs=[api_key_input, remember_key], 
@@ -151,21 +151,33 @@ class GUI:
             print(f"Error deleting saved key: {e}")
         return False
 
-    def on_load(self, profile: gr.OAuthProfile | None = None):
-        """Handle initial page load"""
-        if profile is None:
+    def on_load(self, request: gr.Request = None):
+        """Handle initial page load and login status check"""
+        # Get user info from request if available
+        profile = getattr(request, 'username', None) if request else None
+        
+        if not profile:
             return (
                 gr.update(value='# ePub Summarization Tool\n\nPlease login to access the tool.'),
                 gr.update(visible=False),  # api_key_section
                 gr.update(visible=False),  # inp
                 gr.update(visible=False)   # clear_key_btn
             )
-        return self.on_login(profile)
+        
+        # Create a mock profile object for compatibility
+        class MockProfile:
+            def __init__(self, username):
+                self.username = username
+                self.name = username
+                self.email = f"{username}@example.com"  # Fallback email
+        
+        mock_profile = MockProfile(profile)
+        return self.handle_user_login(mock_profile)
 
-    def on_login(self, profile: gr.OAuthProfile | None = None):
-        """Handle user login"""
+    def handle_user_login(self, profile):
+        """Handle user login logic"""
         if not profile:
-            return self.on_logout()
+            return self.handle_user_logout()
         
         self.current_user = self._get_user_id(profile)
         user_name = profile.name if hasattr(profile, 'name') else profile.username
@@ -190,8 +202,8 @@ class GUI:
             gr.update(visible=False)   # clear_key_btn
         )
 
-    def on_logout(self, profile: gr.OAuthProfile | None = None):
-        """Handle user logout"""
+    def handle_user_logout(self):
+        """Handle user logout logic"""
         self.current_user = None
         self.api_key = None
         self.client = None
@@ -334,7 +346,10 @@ class GUI:
             print(f"Error calling Poe API: {e}")
             return f"Error processing text: {str(e)}"
 
-    def process(self, file, profile: gr.OAuthProfile | None = None):
+    def process(self, file, request: gr.Request = None):
+        # Get user info from request if available
+        profile = getattr(request, 'username', None) if request else None
+        
         if profile is None:
             return gr.update(value='⚠️ Please login to access the tool.')