Update app.py

app.py

@@ -206,15 +206,22 @@ with gr.Blocks() as demo:
     with gr.Column():
         submit_btn = gr.Button("Submit")
         output_image = gr.Image(type="pil", label="Predicted Image")
-        raw_result = gr.Textbox(label="Raw Result", interactive=False)
-
-    submit_btn.click(fn=interface, inputs=[name, age, medical_record, sex, input_image], outputs=[output_image, raw_result])
 
     with gr.Row():
-        download_uploaded_btn = gr.Button("Download Uploaded Folder")
-        download_predicted_btn = gr.Button("Download Predicted Folder")
-
-    download_uploaded_btn.click(fn=download_uploaded_folder, inputs=[], outputs=gr.File())
-    download_predicted_btn.click(fn=download_predicted_folder, inputs=[], outputs=gr.File())
+        raw_result = gr.Textbox(label="Prediction Result")
+    
+    with gr.Row():
+        download_xlsx_btn = gr.Button("Download Patient Information (XLSX)")
+        download_uploaded_btn = gr.Button("Download Uploaded Images")
+        download_predicted_btn = gr.Button("Download Predicted Images")
+    
+    xlsx_file = gr.File(label="Patient Information XLSX 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, xlsx_file])
+    download_xlsx_btn.click(fn=lambda: str(xlsx_db_file), outputs=xlsx_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)
 
 demo.launch()