Debug 14

app.py

@@ -39,8 +39,11 @@ import gradio as gr
 def predict():
     return {"Death": 0.9, "Alive": 0.1}
 
-def download_patient(patient_id: str):
-    return f"Patient{patient_id}.csv"
+def download_patient(patient_id: str) -> str:
+    my_file = Path(f"Patient{patient_id}.csv")
+    if not my_file.is_file():
+        raise Exception(f"[{my_file}] not found.")
+    return str(my_file)
 
 with gr.Blocks() as demo:
     with gr.Row():
@@ -54,10 +57,10 @@ with gr.Blocks() as demo:
                     patient_1_download_btn = gr.DownloadButton(label="Download 1", value="1")
                 with gr.Column(min_width=100):
                     patient_2_input_btn = gr.Button("Patient 2")
-                    patient_2_download_btn = gr.DownloadButton("Download 2")
+                    patient_2_download_btn = gr.DownloadButton(label="Download 2",value="2")
                 with gr.Column(min_width=100):
                     patient_3_input_btn = gr.Button("Patient 3")
-                    patient_3_download_btn = gr.DownloadButton("Download 3")
+                    patient_3_download_btn = gr.DownloadButton(label="Download 3", value="3")
         with gr.Column():
             result = gr.Label(num_top_classes=2, label="Predictions")