Update app.py

app.py

@@ -1,28 +1,36 @@
 import os
 import io
+# Workaround for PIL/Gradio bug :contentReference[oaicite:13]{index=13}
 import PIL.Image
 import gradio as gr
 from gradio_client import Client, handle_file
+
 from gradio_client.client import re
 from numpy import array
-
 # 1. Load your HF token from env
 HF_TOKEN = os.getenv("HF_TOKEN")  # export HF_TOKEN="hf_..."
+# 1) Connect to the Leffa Gradio app’s predict endpoint
+# Use the full "/call/predict" API path as shown on the View API page
 client = Client(
     "franciszzj/Leffa",
     hf_token=HF_TOKEN,
 )  # Gradio Python client
 
+
 def virtual_tryon(
     person_path,
     garment_path,
     garment_type,
 ):
-    # Wrap file inputs so Gradio client uploads them correctly
-    person_file = handle_file(person_path)
+    # 2) Wrap file inputs so Gradio client uploads them correctly
+    person_file = handle_file(
+        person_path
+    )  # handle_file uploads the image :contentReference[oaicite:6]{index=6}
     garment_file = handle_file(garment_path)
 
-    # Call the named endpoint with handle_file inputs
+    # 3) Build inputs in the exact order shown on the “Use via API” page :contentReference[oaicite:7]{index=7}
+
+    # 4) Call the named endpoint with handle_file inputs
     result = client.predict(
         person_file,  # Person Image
         garment_file,  # Garment Image
@@ -33,50 +41,43 @@ def virtual_tryon(
         vt_model_type="viton_hd",
         vt_garment_type=garment_type,
         vt_repaint=False,
-        api_name="/leffa_predict_vt",
-    )
+        api_name="/leffa_predict_vt",)
+    # result[0] is the generated image filepath on the server
     return result[0]  # Gradio will download & display this file
 
-# Password protection
-def check_password(pw):
-    return gr.update(visible=(pw == "supersecretpassword"))
+    # 5) Gradio UI
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## VIRTUAL TRY-ON DEMO")
-    
-    # Invisible password prompt and access button (minimized to zero size)
-    with gr.Row(visible=False):  # Invisible container
-        password_box = gr.Textbox(type="password", label="Enter access code", visible=False, elem_id="password_box")  # Invisible password field
-        access_button = gr.Button("Unlock", visible=False, elem_id="access_button")  # Invisible button
 
-    # Main UI elements (hidden until password is entered)
-    with gr.Column(visible=False) as main_ui:
-        with gr.Row():
-            with gr.Column():
-                gr.Markdown("### Upload Person Image")
-                src = gr.Image(sources="upload", type="filepath", label="Person Image")
-            with gr.Column():
-                gr.Markdown("### Upload Garment Image")
-                ref = gr.Image(sources="upload", type="filepath", label="Garment Image")
-            with gr.Column():
-                gr.Markdown("### Select Garment Type")
-                garment_type = gr.Radio(
-                    choices=[("Upper", "upper_body"), ("Lower", "lower_body"), ("Dress", "dresses")],
-                    value="upper_body",
-                    label="Garment Type",
-                )
-            with gr.Column():
-                gr.Markdown("### Output Image")
-                out = gr.Image(type="filepath", label="Result")
-                with gr.Row():
-                    btn = gr.Button("Generate")
+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"), ("Komplet", "dresses")],
+                value="upper_body",
+                label="Lloji i veshjes",
+            )
+        with gr.Column():
+            gr.Markdown("")
+            out = gr.Image(type="filepath", label="Rezultati",)
+            with gr.Row():
+                btn = gr.Button("VISHU")
 
         btn.click(virtual_tryon, [src, ref, garment_type], out)
 
-    access_button.click(fn=check_password, inputs=password_box, outputs=main_ui)
-
 demo.launch(
     share=True,
     show_error=True,
     pwa=True,
 )
+