Update app.py

app.py

@@ -1655,7 +1655,7 @@ def create_gradio_app():
 
         if not file:
             logger.warning("Missing file")
-            return "", "", "", None, session_state
+            return "", "", "", gr.Column(visible=False), session_state
 
         # Validate file (file.name contains Gradio's temp path)
         logger.info(f"🔍 VALIDATING FILE: {file.name}")
@@ -1664,7 +1664,7 @@ def create_gradio_app():
 
         if not validation["valid"]:
             logger.error(f"❌ FILE VALIDATION FAILED: {validation['error']}")
-            return "", "", "", None, session_state
+            return "", "", "", gr.Column(visible=False), session_state
 
         # Save file to our session directory
         logger.info("💾 Saving uploaded file to session directory...")
@@ -1735,7 +1735,7 @@ def create_gradio_app():
             logger.info(f"🎯 WORKFLOW INITIALIZATION - Session: {ui.session_id}")
             logger.info(f"📝 Document: {temp_path}")
             logger.info("⚡ Starting multi-step financial analysis workflow...")
-            yield (progress_html, create_step_html("extraction"), "", gr.Column(visible=False))
+            yield (progress_html, create_step_html("extraction"), "", gr.Column(visible=False), session_state)
             
             time.sleep(1)  # Brief pause for UI update
             
@@ -1745,7 +1745,7 @@ def create_gradio_app():
             logger.info("=" * 60)
             logger.info("📋 Initializing financial data extraction agent...")
             progress_html = "🔍 <strong>Step 1/4: Extracting financial data from document...</strong>"
-            yield (progress_html, create_step_html("extraction"), "", gr.Column(visible=False))
+            yield (progress_html, create_step_html("extraction"), "", gr.Column(visible=False), session_state)
             
             # Check for cached extraction
             if "extracted_data" in ui.workflow.session_state:
@@ -1794,7 +1794,7 @@ def create_gradio_app():
             logger.info("📊 STEP 2/4: DATA ORGANIZATION PHASE")
             logger.info("=" * 60)
             progress_html = "📊 <strong>Step 2/4: Organizing and analyzing financial data...</strong>"
-            yield (progress_html, create_step_html("arrangement"), "", gr.Column(visible=False))
+            yield (progress_html, create_step_html("arrangement"), "", gr.Column(visible=False), session_state)
             
             if "arrangement_response" in ui.workflow.session_state:
                 logger.info("💾 Using cached data arrangement from previous run")
@@ -1856,7 +1856,7 @@ def create_gradio_app():
             # Step 3: Code Generation
             logger.info("Step 3: Starting code generation...")
             progress_html = "💻 <strong>Step 3/4: Generating Python code for Excel reports...</strong>"
-            yield (progress_html, create_step_html("code_generation"), "", gr.Column(visible=False))
+            yield (progress_html, create_step_html("code_generation"), "", gr.Column(visible=False), session_state)
             
             if "code_generation_response" in ui.workflow.session_state:
                 logger.info("Using cached code generation results from previous run")
@@ -1917,7 +1917,7 @@ def create_gradio_app():
             # Step 4: Final Results
             logger.info("Step 4: Preparing final results...")
             progress_html = "📊 <strong>Step 4/4: Creating final Excel report...</strong>"
-            yield (progress_html, create_step_html("execution"), "", gr.Column(visible=False))
+            yield (progress_html, create_step_html("execution"), "", gr.Column(visible=False), session_state)
             
             time.sleep(1)  # Brief pause to show step
             
@@ -2351,7 +2351,7 @@ def main():
         app.launch(
             server_name="0.0.0.0",
             server_port=7860,
-            share=False,
+            share=True,
             debug=False,
             show_error=True,
         )