Update app.py

app.py

@@ -24,6 +24,7 @@ class GUI:
                 welcome_md = gr.Markdown()
                 # Updated for Gradio 5 - removed LogoutButton (LoginButton transforms automatically)
                 login_btn = gr.LoginButton()
+                refresh_btn = gr.Button("🔄 Refresh", size="sm", variant="secondary")
             
             # API Key input section (shown after login)
             api_key_section = gr.Column(visible=False)
@@ -57,16 +58,21 @@ class GUI:
             out = gr.Markdown()
             inp = gr.File(file_types=['.epub'], visible=False, label="Upload ePub File")
             
-            # Event handlers - Simplified for Gradio 5
+            # Event handlers - Fixed for Gradio 5
             demo.load(
                 fn=self.on_load,
                 outputs=[welcome_md, api_key_section, inp, clear_key_btn]
             )
             
-            # Use timer to check login state (workaround for Gradio 5 login issues)
-            timer = gr.Timer(value=2.0, active=True)
-            timer.tick(
-                fn=self.check_login_state,
+            # Direct login button event handling
+            login_btn.click(
+                fn=self.handle_login_state_change,
+                outputs=[welcome_md, api_key_section, inp, clear_key_btn]
+            )
+            
+            # Manual refresh button for debugging
+            refresh_btn.click(
+                fn=self.handle_login_state_change,
                 outputs=[welcome_md, api_key_section, inp, clear_key_btn]
             )
             
@@ -187,30 +193,31 @@ class GUI:
 
     def on_load(self, request: gr.Request = None):
         """Handle initial page load - Updated for Gradio 5"""
-        if not hasattr(request, 'username') or not request.username:
-            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.handle_login_click(request)
+        print(f"Page load - Username: {getattr(request, 'username', None)}")  # Debug
+        return self.handle_login_state_change(request)
 
-    def check_login_state(self, request: gr.Request = None):
-        """Check login state periodically - workaround for Gradio 5"""
+    def handle_login_state_change(self, request: gr.Request = None):
+        """Handle login state changes - works for both load and login button click"""
         try:
+            print(f"Login state check - Username: {getattr(request, 'username', None)}")  # Debug
+            print(f"Request attrs: {[attr for attr in dir(request) if not attr.startswith('_')]}")  # Debug
+            
             if hasattr(request, 'username') and request.username:
                 return self.handle_user_logged_in(request)
             else:
                 return self.handle_user_not_logged_in()
         except Exception as e:
-            print(f"Error checking login state: {e}")
+            print(f"Error in login state change: {e}")
             return self.handle_user_not_logged_in()
 
+    def check_login_state(self, request: gr.Request = None):
+        """Check login state - with debug info"""
+        return self.handle_login_state_change(request)
+
     def handle_user_not_logged_in(self):
         """Handle when user is not logged in"""
         return (
-            gr.update(value='# ePub Summarization Tool\n\nPlease login with Hugging Face to access the tool.'),
+            gr.update(value='# ePub Summarization Tool\n\nPlease login with Hugging Face to access the tool.\n\n*If you just logged in, click the 🔄 Refresh button above.*'),
             gr.update(visible=False),  # api_key_section
             gr.update(visible=False),  # inp
             gr.update(visible=False)   # clear_key_btn