Update app.py

app.py

@@ -1918,9 +1918,13 @@ def get_active_sessions(language: str) -> List[Tuple[str, str]]:
         logger.error(f"Error getting active sessions: {str(e)}", exc_info=True)
         return []
 
-def resume_session(session_id: str, language: str) -> Generator[Tuple[str, str, str, str], None, None]:
+def resume_session(session_id: str, language: str) -> Generator[Tuple[str, str, str], None, None]:
     """Resume an existing session"""
     if not session_id:
+        if language == "Korean":
+            yield "", "", "❌ 세션을 선택해주세요."
+        else:
+            yield "", "", "❌ Please select a session."
         return
     
     # Process with existing session ID
@@ -2094,7 +2098,7 @@ def download_novel(novel_text: str, format: str, language: str, session_id: str
         logger.info(f"TXT saved: {filepath}")
         return filepath
 
-# Custom CSS with enhanced recovery status
+# Custom CSS
 custom_css = """
 .gradio-container {
     background: linear-gradient(135deg, #1e3c72, #2a5298);
@@ -2128,15 +2132,6 @@ custom_css = """
     color: white;
 }
 
-.recovery-status {
-    background-color: rgba(76, 175, 80, 0.2);
-    border: 1px solid #4CAF50;
-    padding: 10px;
-    border-radius: 5px;
-    margin: 10px 0;
-    color: #4CAF50;
-}
-
 #stages-display {
     background-color: rgba(255, 255, 255, 0.95);
     padding: 20px;
@@ -2197,13 +2192,6 @@ def create_interface():
         # State management
         current_session_id = gr.State(None)
         
-        # Recovery status display
-        recovery_status = gr.Markdown(
-            value="",
-            elem_classes=["recovery-status"],
-            visible=False
-        )
-        
         with gr.Row():
             with gr.Column(scale=1):
                 with gr.Group(elem_classes=["input-section"]):
@@ -2294,9 +2282,6 @@ def create_interface():
             )
         
         # Event handlers
-        def update_novel_state(stages, novel, status, recovery_msg):
-            return stages, novel, status, novel, recovery_msg
-        
         def refresh_sessions():
             try:
                 sessions = get_active_sessions("English")
@@ -2307,23 +2292,19 @@ def create_interface():
         
         def handle_auto_recover(language):
             session_id, message = auto_recover_session(language)
-            if session_id:
-                return session_id, gr.update(value=message, visible=True)
-            else:
-                if language == "Korean":
-                    message = "❌ 복구할 세션이 없습니다."
-                else:
-                    message = "❌ No session to recover."
-                return None, gr.update(value=message, visible=True)
+            return session_id
         
         submit_btn.click(
             fn=process_query,
             inputs=[query_input, language_select, current_session_id],
-            outputs=[stages_display, novel_output, status_text, recovery_status.value]
-        ).then(
-            fn=update_novel_state,
-            inputs=[stages_display, novel_output, status_text, recovery_status.value],
-            outputs=[stages_display, novel_output, status_text, novel_text_state, recovery_status]
+            outputs=[stages_display, novel_output, status_text]
+        )
+        
+        # Update novel text state when novel output changes
+        novel_output.change(
+            fn=lambda x: x,
+            inputs=[novel_output],
+            outputs=[novel_text_state]
         )
         
         resume_btn.click(
@@ -2333,17 +2314,17 @@ def create_interface():
         ).then(
             fn=resume_session,
             inputs=[current_session_id, language_select],
-            outputs=[stages_display, novel_output, status_text, recovery_status.value]
+            outputs=[stages_display, novel_output, status_text]
         )
         
         auto_recover_btn.click(
             fn=handle_auto_recover,
             inputs=[language_select],
-            outputs=[current_session_id, recovery_status]
+            outputs=[current_session_id]
         ).then(
             fn=resume_session,
             inputs=[current_session_id, language_select],
-            outputs=[stages_display, novel_output, status_text, recovery_status.value]
+            outputs=[stages_display, novel_output, status_text]
         )
         
         refresh_btn.click(
@@ -2352,11 +2333,11 @@ def create_interface():
         )
         
         clear_btn.click(
-            fn=lambda: ("", "", "🔄 Ready", "", None, gr.update(visible=False)),
-            outputs=[stages_display, novel_output, status_text, novel_text_state, current_session_id, recovery_status]
+            fn=lambda: ("", "", "🔄 Ready", "", None),
+            outputs=[stages_display, novel_output, status_text, novel_text_state, current_session_id]
         )
         
-        def handle_download(novel_text, format_type, language, session_id):
+        def handle_download(format_type, language, session_id, novel_text):
             if not novel_text and not session_id:
                 return gr.update(visible=False)
             
@@ -2368,24 +2349,14 @@ def create_interface():
         
         download_btn.click(
             fn=handle_download,
-            inputs=[novel_text_state, format_select, language_select, current_session_id],
+            inputs=[format_select, language_select, current_session_id, novel_text_state],
             outputs=[download_file]
         )
         
-        # Load sessions and check for auto-recovery on startup
-        def on_load():
-            sessions = get_active_sessions("English")
-            # Check for latest session
-            latest_session = NovelDatabase.get_latest_active_session()
-            if latest_session:
-                recovery_msg = f"💡 Found active session: '{latest_session['user_query'][:30]}...' - Click 'Auto Recover Latest' to continue"
-                return gr.update(choices=sessions), gr.update(value=recovery_msg, visible=True)
-            else:
-                return gr.update(choices=sessions), gr.update(visible=False)
-        
+        # Load sessions on startup
         interface.load(
-            fn=on_load,
-            outputs=[session_dropdown, recovery_status]
+            fn=refresh_sessions,
+            outputs=[session_dropdown]
         )
     
     return interface