Update app.py

app.py

@@ -9,7 +9,11 @@ client = Client(
     hf_token=HF_TOKEN,
 )
 
+# To store the last result path
+result_path = ""
+
 def virtual_tryon(person_path, garment_path, garment_type):
+    global result_path
     person_file = handle_file(person_path)
     garment_file = handle_file(garment_path)
 
@@ -25,24 +29,13 @@ def virtual_tryon(person_path, garment_path, garment_type):
         vt_repaint=False,
         api_name="/leffa_predict_vt",
     )
-    return result[0], gr.update(value=result[0], visible=True)  # show image + reveal "Shkarko"
+    result_path = result[0]
+    return result_path, gr.update(visible=True)
+
+def download_image():
+    return result_path
 
-with gr.Blocks(theme=gr.themes.Soft(), css="""
-#image-container { position: relative; display: inline-block; }
-#download-button {
-    position: absolute;
-    top: 10px;
-    left: 50%;
-    transform: translateX(-50%);
-    z-index: 10;
-    background-color: rgba(0, 0, 0, 0.6);
-    color: white;
-    padding: 10px 16px;
-    border-radius: 6px;
-    border: none;
-    cursor: pointer;
-}
-""") as demo:
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("")
     with gr.Row():
         with gr.Column():
@@ -62,12 +55,12 @@ with gr.Blocks(theme=gr.themes.Soft(), css="""
             )
         with gr.Column():
             gr.Markdown("")
-            with gr.Group(elem_id="image-container"):
-                download_btn = gr.File(label="Shkarko", elem_id="download-button", visible=False)
-                out = gr.Image(type="filepath", label="Rezultati")
+            out = gr.Image(type="filepath", label="Rezultati")
             btn = gr.Button("VISHU")
+            download_trigger = gr.Button("Shkarko", visible=False)
 
-    btn.click(virtual_tryon, [src, ref, garment_type], [out, download_btn])
+    btn.click(virtual_tryon, [src, ref, garment_type], [out, download_trigger])
+    download_trigger.click(download_image, outputs=gr.File())
 
 demo.launch(
     share=True,