Update app.py

app.py

@@ -6,6 +6,7 @@ import gradio as gr
 import spaces
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
+from pathlib import Path
 
 
 MAX_MAX_NEW_TOKENS = 2048
@@ -140,10 +141,26 @@ chat_interface = gr.ChatInterface(
 
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
-    gr.UploadButton(value="Upload a File", elem_id="uploadbutton")
     chat_interface.render()
     gr.Markdown(LICENSE)
 
+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)]
+
+with gr.Blocks() as demo:
+    gr.Markdown("First upload a file and and then you'll be able download it (but only once!)")
+    with gr.Row():
+        u = gr.UploadButton("Upload a file", file_count="single")
+        d = gr.DownloadButton("Download the file", visible=False)
+
+    u.upload(upload_file, u, [u, d])
+    d.click(download_file, None, [u, d])
+
+
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch()