Update app.py

app.py

@@ -15,36 +15,27 @@ result_path = ""
 
 def rename_to_image(input_path, output_dir="Uploads"):
     os.makedirs(output_dir, exist_ok=True)
-    # Rename to Image_<random>.png
     random_id = os.urandom(4).hex()
     output_filename = f"Image_{random_id}.png"
     output_path = os.path.join(output_dir, output_filename)
     try:
         with Image.open(input_path) as img:
             img.convert('RGB').save(output_path, 'PNG', quality=100)
-        return output_path, ""
+        return output_path
     except Exception as e:
-        return None, f"Error renaming image: {str(e)}"
-
-def store_person_upload(person_path):
-    if person_path:
-        person_png, error = rename_to_image(person_path)
-        return person_png, error
-    return None, ""
-
-def store_garment_upload(garment_path):
-    if garment_path:
-        garment_png, error = rename_to_image(garment_path)
-        return garment_png, error
-    return None, ""
+        raise ValueError(f"Error renaming image: {str(e)}")
 
 def virtual_tryon(person_path, garment_path, garment_type):
     global result_path
     if not person_path or not garment_path:
-        return None, gr.update(visible=False), gr.update(visible=True), "Ju lutem ngarkoni të dyja imazhet."
+        return None, gr.update(visible=False), "Ju lutem ngarkoni të dyja imazhet."
+
+    # Rename uploads to Image_<random>.png
+    person_png = rename_to_image(person_path)
+    garment_png = rename_to_image(garment_path)
 
-    person_file = handle_file(person_path)
-    garment_file = handle_file(garment_path)
+    person_file = handle_file(person_png)
+    garment_file = handle_file(garment_png)
 
     try:
         result = client.predict(
@@ -67,9 +58,9 @@ def virtual_tryon(person_path, garment_path, garment_type):
         with Image.open(input_image_path) as img:
             img.convert('RGB').save(output_image_path, 'PNG', quality=95)
         result_path = output_image_path
-        return result_path, gr.update(visible=True), gr.update(visible=False), ""
+        return result_path, gr.update(visible=True), ""
     except Exception as e:
-        return None, gr.update(visible=False), gr.update(visible=True), f"Error processing: {str(e)}"
+        return None, gr.update(visible=False), f"Error processing: {str(e)}"
 
 def trigger_download():
     return gr.update(value=result_path, visible=True, interactive=True)
@@ -77,10 +68,6 @@ def trigger_download():
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("")
     error_msg = gr.Markdown("", visible=False)
-    # States to store image paths
-    person_state = gr.State()
-    garment_state = gr.State()
-
     with gr.Row():
         with gr.Column():
             src = gr.Image(sources="upload", type="filepath", label="Foto e personit")
@@ -100,24 +87,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             downloadable = gr.File(visible=False)
             out = gr.Image(type="filepath", label="Rezultati")
 
-    # Store renamed images on upload
-    src.upload(
-        store_person_upload,
-        inputs=src,
-        outputs=[person_state, error_msg]
-    )
-    ref.upload(
-        store_garment_upload,
-        inputs=ref,
-        outputs=[garment_state, error_msg]
-    )
-    # Process and show result
     btn.click(
         virtual_tryon,
-        inputs=[person_state, garment_state, garment_type],
-        outputs=[out, shkarko_btn, downloadable, error_msg]
+        inputs=[src, ref, garment_type],
+        outputs=[out, shkarko_btn, error_msg]
     )
-    # Trigger download
     shkarko_btn.click(trigger_download, outputs=downloadable)
 
 demo.launch(