add upload & download file functions & gradio elements

app.py

@@ -123,6 +123,13 @@ def model_call(question, document, answer, client_base_url):
     combined_reasoning = " ".join(reasoning)[1:-1]
     return combined_reasoning, score
 
+def upload_file(filepath):
+    name = Path(filepath).name
+    return [gr.UploadButton(visible=False), gr.DownloadButton(label=f"Download {name}", value=filepath, visible=True)]
+
+def download_file():
+    return [gr.UploadButton(visible=True), gr.DownloadButton(visible=False)]
+
 # inputs = [
 #     gr.Textbox(label="Question"),
 #     gr.Textbox(label="Document"),
@@ -149,7 +156,12 @@ with gr.Blocks() as demo:
         with gr.Column(scale=1):
             reasoning = gr.Textbox(label="Reasoning")
             score = gr.Textbox(label="Score (FAIL if Hallucinated, PASS if not)")
+        u = gr.UploadButton("Upload a file", file_count="single")
+        d = gr.DownloadButton("Download the file", visible=False)
+
     model_dropdown.change(fn=update_client_base_url, inputs=[model_dropdown], outputs=[base_url_state])
+    u.upload(upload_file, u, [u, d])
+    d.click(download_file, None, [u, d])
 
     submit_button.click(fn=model_call, inputs=[question, document, answer, base_url_state], outputs=[reasoning, score])
     question.submit(fn=model_call, inputs=[question, document, answer, base_url_state], outputs=[reasoning, score])