Update app.py

app.py

@@ -348,7 +348,7 @@ def process_pdf(file_obj, progress=gr.Progress()) -> Tuple[Optional[str], str, s
             pass
         return None, "", f"❌ Error: {str(e)}"
 
-# Create Gradio interface with fixed download component
+# Create Gradio interface with fixed download functionality
 with gr.Blocks(
     title="PDF Splitter - Fast & Simple",
     theme=gr.themes.Base()
@@ -381,24 +381,42 @@ with gr.Blocks(
         stats_output = gr.HTML()
     
     with gr.Row():
-        # FIXED: Set interactive=True and use value parameter correctly
+        # CRITICAL FIX: Use gr.File with proper configuration
         download_file = gr.File(
             label="📦 Download ZIP (Contains only segments ≤5MB)",
             visible=True,
-            interactive=False,  # Keep false initially
-            type="filepath"  # Ensure it's set to filepath
+            type="filepath",  # Must be filepath for output
+            interactive=False,  # Non-interactive for output
+            elem_id="download_output"  # Add ID for debugging
         )
     
-    # Handle processing with proper output mapping
-    def process_and_update(file_obj, progress=gr.Progress()):
-        zip_path, stats_html, status_msg = process_pdf(file_obj, progress)
-        # Return the file path directly for gr.File component
-        return zip_path, stats_html, status_msg
+    # Fixed processing function that ensures file path is returned correctly
+    def process_pdf_wrapper(file_obj, progress=gr.Progress()):
+        """Wrapper to ensure proper file path return"""
+        if file_obj is None:
+            return None, "", "⚠️ Please upload a PDF file"
+        
+        try:
+            # Call your existing process_pdf function
+            zip_path, stats_html, status_msg = process_pdf(file_obj, progress)
+            
+            # CRITICAL: Ensure zip_path is a string and file exists
+            if zip_path and Path(zip_path).exists():
+                # Return the string path directly
+                return str(zip_path), stats_html, status_msg
+            else:
+                return None, stats_html, "❌ Error: ZIP file was not created"
+                
+        except Exception as e:
+            logger.error(f"Process wrapper error: {str(e)}")
+            return None, "", f"❌ Error: {str(e)}"
     
+    # Connect the button with proper output mapping
     split_btn.click(
-        fn=process_and_update,
+        fn=process_pdf_wrapper,
         inputs=[file_input],
-        outputs=[download_file, stats_output, status_text]
+        outputs=[download_file, stats_output, status_text],
+        show_progress=True
     )
 
 # Create Gradio interface with fixed theme