Update app.py

app.py

@@ -243,6 +243,7 @@ cleanup_thread.start()
 def process_pdf(file_obj, progress=gr.Progress()) -> Tuple[Optional[str], str, str]:
     """
     Main processing function for Gradio interface
+    Returns: (zip_file_path, statistics_html, status_message)
     """
     if file_obj is None:
         return None, "", "⚠️ Please upload a PDF file"
@@ -292,7 +293,8 @@ def process_pdf(file_obj, progress=gr.Progress()) -> Tuple[Optional[str], str, s
         
         # Create ZIP file
         progress(0.9, "Creating ZIP archive...")
-        zip_path = session_dir / f"pdf_segments_{session_id}.zip"
+        zip_filename = f"pdf_segments_{session_id}.zip"
+        zip_path = session_dir / zip_filename
         
         with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
             for file_path in output_files:
@@ -315,28 +317,9 @@ def process_pdf(file_obj, progress=gr.Progress()) -> Tuple[Optional[str], str, s
                     <td style="padding: 10px; font-weight: bold; color: #334155;">❌ Segments Discarded (>5MB):</td>
                     <td style="padding: 10px; text-align: right; color: #dc2626; font-weight: 600;">{stats['segments_discarded']}</td>
                 </tr>
-                <tr style="border-bottom: 1px solid #e2e8f0; background: #f8fafc;">
-                    <td style="padding: 10px; font-weight: bold; color: #334155;">📦 Original Size:</td>
-                    <td style="padding: 10px; text-align: right; color: #475569; font-weight: 600;">{stats['original_size_mb']:.2f} MB</td>
-                </tr>
-                <tr style="border-bottom: 1px solid #e2e8f0;">
-                    <td style="padding: 10px; font-weight: bold; color: #334155;">📁 Total Output Size:</td>
-                    <td style="padding: 10px; text-align: right; color: #475569; font-weight: 600;">{stats['total_output_size_mb']:.2f} MB</td>
-                </tr>
-                <tr style="border-bottom: 1px solid #e2e8f0; background: #f8fafc;">
-                    <td style="padding: 10px; font-weight: bold; color: #334155;">📈 Largest Segment:</td>
-                    <td style="padding: 10px; text-align: right; color: #475569; font-weight: 600;">{stats['largest_segment_mb']:.2f} MB</td>
-                </tr>
-                <tr style="background: white;">
-                    <td style="padding: 10px; font-weight: bold; color: #334155;">📉 Smallest Segment:</td>
-                    <td style="padding: 10px; text-align: right; color: #475569; font-weight: 600;">{stats['smallest_segment_mb']:.2f} MB</td>
-                </tr>
             </table>
             <p style="margin-top: 15px; color: #059669; font-weight: bold;">
-                ✨ Your file has been split successfully!
-            </p>
-            <p style="margin-top: 10px; color: #6b7280; font-size: 0.9em;">
-                ⏱️ Files will be automatically deleted after {CLEANUP_AFTER_MINUTES} minutes
+                ✨ Your file has been split successfully! Click the download button below.
             </p>
         </div>
         """
@@ -346,7 +329,8 @@ def process_pdf(file_obj, progress=gr.Progress()) -> Tuple[Optional[str], str, s
         # Clean up input file to save space
         input_path.unlink()
         
-        return str(zip_path), stats_html, "✅ Processing complete! Download your ZIP file below."
+        # IMPORTANT: Return the actual file path as a string for gr.File to handle
+        return str(zip_path), stats_html, "✅ Processing complete! Your ZIP file is ready for download."
         
     except Exception as e:
         logger.error(f"Processing error: {str(e)}")
@@ -357,6 +341,59 @@ 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
+with gr.Blocks(
+    title="PDF Splitter - Fast & Simple",
+    theme=gr.themes.Base()
+) as app:
+    
+    gr.Markdown("""
+    # 📄 PDF Splitter Tool
+    
+    **Split large PDFs into 4.5MB segments - Files over 5MB are automatically discarded!**
+    """)
+    
+    with gr.Row():
+        with gr.Column():
+            file_input = gr.File(
+                label="Upload PDF",
+                file_types=[".pdf"],
+                type="filepath"
+            )
+            
+            split_btn = gr.Button(
+                "🚀 Split PDF into 4.5MB Segments", 
+                variant="primary",
+                size="lg"
+            )
+    
+    with gr.Row():
+        status_text = gr.Markdown("Ready to process your PDF...")
+    
+    with gr.Row():
+        stats_output = gr.HTML()
+    
+    with gr.Row():
+        # FIXED: Set interactive=True and use value parameter correctly
+        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
+        )
+    
+    # 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
+    
+    split_btn.click(
+        fn=process_and_update,
+        inputs=[file_input],
+        outputs=[download_file, stats_output, status_text]
+    )
+
 # Create Gradio interface with fixed theme
 with gr.Blocks(
     title="PDF Splitter - Fast & Simple",