Update app.py

app.py

@@ -77,33 +77,31 @@ 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)
+    temp_file_path = gr.State(value=None)
     btn = gr.Button("Process File")
     
-    # State to store processed file path
-    temp_file_path = gr.State()
-
     def process_and_show(file):
         df, out_path = process_file(file)
         if out_path:
-            temp_file_path.set(out_path)  # Save file path for download
-            return df, gr.update(visible=True)
-        return df, gr.update(visible=False)
+            return df, out_path, gr.update(visible=True)
+        return df, None, gr.update(visible=False)
     
-    def download_file():
-        return temp_file_path.get()  # Return the stored file path
+    def download_file(out_path):
+        return out_path
 
     btn.click(
         process_and_show,
         inputs=input_file,
-        outputs=[output_table, download_button]
+        outputs=[output_table, temp_file_path, download_button]
     )
 
     download_button.click(
         download_file,
-        inputs=None,
+        inputs=temp_file_path,
         outputs=gr.File(label="Processed Schedule")
     )
 
 if __name__ == "__main__":
     demo.launch()
 
+