Update app.py

app.py

@@ -26,6 +26,14 @@ def download_report(report_text):
         return None
     return generate_pdf(report_text)
 
+def reset_on_clear():
+    return (
+        None,                           # scan_display
+        "",                             # report_box
+        gr.Button(interactive=False),   # generate_btn
+        gr.DownloadButton(visible=False) # download_btn
+    )
+
 # Build the UI
 with gr.Blocks(title="MRI Brain Report Generator") as app:
     
@@ -81,4 +89,10 @@ with gr.Blocks(title="MRI Brain Report Generator") as app:
         outputs=download_btn
     )
 
+    upload_btn.clear(
+    fn=reset_on_clear,
+    inputs=None,
+    outputs=[scan_display, report_box, generate_btn, download_btn]
+)
+
 app.launch()