Update app.py

app.py

@@ -57,15 +57,16 @@ class GUI:
             out = gr.Markdown()
             inp = gr.File(file_types=['.epub'], visible=False, label="Upload ePub File")
             
-            # Event handlers - Updated for Gradio 5
+            # Event handlers - Simplified for Gradio 5
             demo.load(
                 fn=self.on_load,
                 outputs=[welcome_md, api_key_section, inp, clear_key_btn]
             )
             
-            # Updated login handling for Gradio 5
-            login_btn.click(
-                fn=self.handle_login_click,
+            # 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,
                 outputs=[welcome_md, api_key_section, inp, clear_key_btn]
             )
             
@@ -195,38 +196,68 @@ class GUI:
             )
         return self.handle_login_click(request)
 
-    def handle_login_click(self, request: gr.Request = None):
-        """Handle user login - Updated for Gradio 5"""
-        if not hasattr(request, 'username') or not request.username:
+    def check_login_state(self, request: gr.Request = None):
+        """Check login state periodically - workaround for Gradio 5"""
+        try:
+            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}")
+            return self.handle_user_not_logged_in()
+
+    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(visible=False),  # api_key_section
+            gr.update(visible=False),  # inp
+            gr.update(visible=False)   # clear_key_btn
+        )
+
+    def handle_user_logged_in(self, request):
+        """Handle when user is logged in"""
+        if not self.current_user or self.current_user != self._get_user_id(request):
+            # New login or different user
+            self.current_user = self._get_user_id(request)
+            user_name = request.username
+            
+            # Check if user has a saved API key
+            saved_key = self._get_saved_key(self.current_user)
+            if saved_key:
+                self.api_key = saved_key
+                if self._initialize_client():
+                    return (
+                        gr.update(value=f'# ePub Summarization Tool\n\nWelcome back {user_name}! ✅ Your saved API key is loaded and ready.'),
+                        gr.update(visible=False),  # api_key_section
+                        gr.update(visible=True),   # inp
+                        gr.update(visible=True)    # clear_key_btn
+                    )
+            
+            # No saved key or failed to initialize
             return (
-                gr.update(value='# ePub Summarization Tool\n\nPlease login to access the tool.'),
-                gr.update(visible=False),  # api_key_section
+                gr.update(value=f'# ePub Summarization Tool\n\nWelcome {user_name}! Please set up your Poe API key below.'),
+                gr.update(visible=True),   # api_key_section
                 gr.update(visible=False),  # inp
                 gr.update(visible=False)   # clear_key_btn
             )
         
-        self.current_user = self._get_user_id(request)
-        user_name = request.username
-        
-        # Check if user has a saved API key
-        saved_key = self._get_saved_key(self.current_user)
-        if saved_key:
-            self.api_key = saved_key
-            if self._initialize_client():
-                return (
-                    gr.update(value=f'# ePub Summarization Tool\n\nWelcome back {user_name}! ✅ Your saved API key is loaded and ready.'),
-                    gr.update(visible=False),  # api_key_section
-                    gr.update(visible=True),   # inp
-                    gr.update(visible=True)    # clear_key_btn
-                )
-        
-        # No saved key or failed to initialize
-        return (
-            gr.update(value=f'# ePub Summarization Tool\n\nWelcome {user_name}! Please set up your Poe API key below.'),
-            gr.update(visible=True),   # api_key_section
-            gr.update(visible=False),  # inp
-            gr.update(visible=False)   # clear_key_btn
-        )
+        # User already processed, return current state
+        if self.client:
+            return (
+                gr.update(value=f'# ePub Summarization Tool\n\nWelcome {request.username}! ✅ Ready to process ePub files.'),
+                gr.update(visible=False),  # api_key_section
+                gr.update(visible=True),   # inp
+                gr.update(visible=True)    # clear_key_btn
+            )
+        else:
+            return (
+                gr.update(value=f'# ePub Summarization Tool\n\nWelcome {request.username}! Please set up your Poe API key below.'),
+                gr.update(visible=True),   # api_key_section
+                gr.update(visible=False),  # inp
+                gr.update(visible=False)   # clear_key_btn
+            )
 
     def _initialize_client(self):
         """Initialize the Poe API client"""