Upload app.py

app.py

@@ -84,7 +84,8 @@ def query_agent(description, question, image0, image1):
         return None, None
 
 def simulate(image0, image1):
-    print(image0, image1)
+    upload_file(image0)
+    upload_file(image1)
     preferences = []
     reasons = []
     probs = []
@@ -111,6 +112,22 @@ def simulate(image0, image1):
 
 subtitle = "Upload two images of emails and see which is generally preferred by Petco customers!"
 
+
+from pathlib import Path
+
+def upload_file(filepath):
+    name = Path(filepath).name
+    return [gr.UploadButton(visible=False), gr.DownloadButton(label=f"{name}", value=filepath, visible=True)]
+
+# 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])
+
 demo = gr.Interface(fn=simulate, 
                     inputs=[gr.UploadButton("Click to Upload Email 0", file_types=["image"], file_count="single"), 
                             gr.UploadButton("Click to Upload Email 1", file_types=["image"], file_count="single")],