Update app.py

app.py

@@ -93,6 +93,11 @@ def process_file(input_file):
         error_df = pd.DataFrame({"Error": [f"⚠️ {str(e)}"]})
         return error_df, None
 
+# Add the missing download_file function
+def download_file(out_path):
+    """Return the processed file path for download"""
+    return out_path
+
 # Gradio interface
 with gr.Blocks(title="Schedule Processor") as demo:
     gr.Markdown("# 📅 Schedule Processor")
@@ -107,7 +112,6 @@ with gr.Blocks(title="Schedule Processor") as demo:
     
     output_table = gr.Dataframe(label="Preview", wrap=True)
     download_button = gr.Button("Download Processed File", visible=False)
-    output_file = gr.File(label="Processed Schedule", visible=False)
     temp_file_path = gr.State(value=None)
 
     def reset_components():
@@ -116,32 +120,30 @@ with gr.Blocks(title="Schedule Processor") as demo:
             None,  # Clear file input
             pd.DataFrame(),  # Clear output table
             None,  # Reset temp file path
-            gr.update(visible=False),  # Hide download button
-            gr.update(visible=False)  # Hide output file
+            gr.update(visible=False)  # Hide download button
         ]
 
     def process_and_show(file):
         df, out_path = process_file(file)
         if out_path:
-            return df, out_path, gr.update(visible=True), gr.update(visible=False)
-        return df, None, gr.update(visible=False), gr.update(visible=False)
+            return df, out_path, gr.update(visible=True)
+        return df, None, gr.update(visible=False)
 
     process_btn.click(
         process_and_show,
         inputs=input_file,
-        outputs=[output_table, temp_file_path, download_button, output_file]
+        outputs=[output_table, temp_file_path, download_button]
     )
 
     reset_btn.click(
         reset_components,
-        outputs=[input_file, output_table, temp_file_path, download_button, output_file]
+        outputs=[input_file, output_table, temp_file_path, download_button]
     )
 
-    # Changed to just pass the file path directly
     download_button.click(
-        lambda x: x,  # Just pass the file path through
+        download_file,
         inputs=temp_file_path,
-        outputs=output_file
+        outputs=gr.File(label="Processed Schedule")
     )
 
 if __name__ == "__main__":