Update app.py

app.py

@@ -223,13 +223,15 @@ with gr.Blocks() as demo:
         download_uploaded_btn = gr.Button("Download Uploaded Images")
         download_predicted_btn = gr.Button("Download Predicted Images")
     
-    # Add a file download output component
+    # Add file download output components for the uploaded and predicted images
     patient_info_file = gr.File(label="Patient Information HTML File")
+    uploaded_folder_file = gr.File(label="Uploaded Images Zip File")
+    predicted_folder_file = gr.File(label="Predicted Images Zip File")
 
     submit_btn.click(fn=interface, inputs=[name, age, medical_record, sex, input_image], outputs=[output_image, raw_result])
     download_html_btn.click(fn=save_patient_info_to_html, inputs=[name, age, medical_record, sex, raw_result], outputs=patient_info_file)
-    download_uploaded_btn.click(fn=download_uploaded_folder, outputs="file")
-    download_predicted_btn.click(fn=download_predicted_folder, outputs="file")
+    download_uploaded_btn.click(fn=download_uploaded_folder, outputs=uploaded_folder_file)
+    download_predicted_btn.click(fn=download_predicted_folder, outputs=predicted_folder_file)
 
 # Launch Gradio app
 demo.launch()