Update app.py

app.py

@@ -10,9 +10,6 @@ client = Client(
     hf_token=HF_TOKEN,
 )
 
-# Variable to hold result path
-result_path = ""
-
 def rename_to_image(input_path, output_dir="Uploads"):
     os.makedirs(output_dir, exist_ok=True)
     random_id = os.urandom(4).hex()
@@ -23,12 +20,11 @@ def rename_to_image(input_path, output_dir="Uploads"):
             img.convert('RGB').save(output_path, 'PNG', quality=100)
         return output_path
     except Exception as e:
-        raise ValueError(f"Error renaming image: {str(e)}")
+        raise ValueError(f"Error renaming: {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), "Ju lutem ngarkoni të dyja imazhet."
+        return None, "Ju lutem ngarkoni të dyja imazhet."
 
     # Rename uploads to Image_<random>.png
     person_png = rename_to_image(person_path)
@@ -43,27 +39,23 @@ def virtual_tryon(person_path, garment_path, garment_type):
             garment_file,
             ref_acceleration=False,
             step=40,
-            scale=2.5,
+            scale=2.5,  # Corrected to 2.5 (was 42 in last version)
             seed=42,
             vt_model_type="viton_hd",
             vt_garment_type=garment_type,
             vt_repaint=False,
             api_name="/leffa_predict_vt",
         )
-        # Rename result to medyfytyra_<random>.png
+        # Rename result to uniqueproc_<random>.png
         input_image_path = result[0]
         random_id = os.urandom(4).hex()
-        output_image_path = os.path.join("Outputs", f"medyfytyra_{random_id}.png")
+        output_image_path = os.path.join("Outputs", f"uniqueproc_{random_id}.png")
         os.makedirs("Outputs", exist_ok=True)
         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), ""
+        return output_image_path, ""
     except Exception as 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)
+        return None, f"Error processing: {str(e)}"
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("")
@@ -83,16 +75,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             )
         with gr.Column():
             btn = gr.Button("VISHU")
-            shkarko_btn = gr.Button("SHKARKO", visible=False)
-            downloadable = gr.File(visible=False)
             out = gr.Image(type="filepath", label="Rezultati")
 
     btn.click(
         virtual_tryon,
         inputs=[src, ref, garment_type],
-        outputs=[out, shkarko_btn, error_msg]
+        outputs=[out, error_msg]
     )
-    shkarko_btn.click(trigger_download, outputs=downloadable)
 
 demo.launch(
     share=True,