Update app.py

app.py

@@ -9,7 +9,7 @@ client = Client(
     hf_token=HF_TOKEN,
 )
 
-# To store the last result path
+# Variable to hold result path
 result_path = ""
 
 def virtual_tryon(person_path, garment_path, garment_type):
@@ -30,22 +30,19 @@ def virtual_tryon(person_path, garment_path, garment_type):
         api_name="/leffa_predict_vt",
     )
     result_path = result[0]
-    return result_path, gr.update(visible=True)
+    return result_path, gr.update(visible=True), gr.update(visible=False)
 
-def download_image():
-    return result_path
+def trigger_download():
+    return gr.update(value=result_path, visible=True)
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("")
     with gr.Row():
         with gr.Column():
-            gr.Markdown("")
             src = gr.Image(sources="upload", type="filepath", label="Foto e personit")
         with gr.Column():
-            gr.Markdown("")
             ref = gr.Image(sources="upload", type="filepath", label="Foto e veshjes")
         with gr.Column():
-            gr.Markdown("")
             garment_type = gr.Radio(
                 choices=[("Siper", "upper_body"),
                          ("Posht", "lower_body"),
@@ -54,13 +51,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 label="Lloji i veshjes",
             )
         with gr.Column():
-            gr.Markdown("")
-            out = gr.Image(type="filepath", label="Rezultati")
             btn = gr.Button("VISHU")
-            download_trigger = gr.Button("Shkarko", visible=False)
+            shkarko_btn = gr.Button("SHKARKO", visible=False)
+            downloadable = gr.File(visible=False)
+            out = gr.Image(type="filepath", label="Rezultati")
 
-    btn.click(virtual_tryon, [src, ref, garment_type], [out, download_trigger])
-    download_trigger.click(download_image, outputs=gr.File())
+    btn.click(virtual_tryon, [src, ref, garment_type], [out, shkarko_btn, downloadable])
+    shkarko_btn.click(trigger_download, outputs=downloadable)
 
 demo.launch(
     share=True,