Update app.py

app.py

@@ -152,12 +152,12 @@ def save_patient_info_to_html(name, age, medical_record, sex, result):
     
     return html_file_path
 
-# Function to download the folders
-def download_folder(folder):
-    zip_path = os.path.join(tempfile.gettempdir(), f"{folder}.zip")
+# Function to download the folders (fix: pass string path instead of Path object)
+def download_folder(folder_path):
+    zip_path = os.path.join(tempfile.gettempdir(), f"{Path(folder_path).name}.zip")
     
     # Zip the folder
-    shutil.make_archive(zip_path.replace('.zip', ''), 'zip', folder)
+    shutil.make_archive(zip_path.replace('.zip', ''), 'zip', folder_path)
     
     return zip_path
 
@@ -192,10 +192,11 @@ with gr.Blocks() as demo:
     uploaded_folder_file = gr.File(label="Uploaded Images Zip File")
     predicted_folder_file = gr.File(label="Predicted Images Zip File")
 
+    # Fix: pass string representation of Path objects
     submit_btn.click(fn=predict_image, 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_folder, inputs=[uploaded_folder], outputs=uploaded_folder_file)
-    download_predicted_btn.click(fn=download_folder, inputs=[predicted_folder], outputs=predicted_folder_file)
+    download_uploaded_btn.click(fn=download_folder, inputs=[str(uploaded_folder)], outputs=uploaded_folder_file)
+    download_predicted_btn.click(fn=download_folder, inputs=[str(predicted_folder)], outputs=predicted_folder_file)
 
 # Launch Gradio app
 demo.launch()