Update app.py

app.py

@@ -1,57 +1,53 @@
-import threading
+
+import gradio as gr
 import time
+import random
 import os
-import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Initialize Hugging Face client
+# Initialize client
 client = InferenceClient(
     provider="fal-ai",
     api_key=os.environ["HF_TOKEN"],
 )
 
-def run_edit(input_image, prompt, progress_html):
-    stop_flag = {"stop": False}
-
-    # --- background thread for smooth progress bar ---
-    def progress_loop():
-        current = 0.0
-        while not stop_flag["stop"]:
-            # Smooth progress approach
-            current += (0.99 - current) * 0.02
-            bar_html = f"""
-            <div style='width: 100%; background-color: #e0e0e0; border-radius: 12px;'>
-                <div style='width: {current*100:.1f}%; background-color: #4CAF50; height: 25px; border-radius: 12px; transition: width 0.05s linear;'></div>
-            </div>
-            <div style='margin-top: 5px;'>Editing Image… {int(current*100)}%</div>
-            """
-            progress_html.update(bar_html)
-            time.sleep(0.05)
-
-        # Finish immediately
-        bar_html = f"""
-        <div style='width: 100%; background-color: #e0e0e0; border-radius: 12px;'>
-            <div style='width: 100%; background-color: #4CAF50; height: 25px; border-radius: 12px;'></div>
-        </div>
-        <div style='margin-top: 5px;'>Done!</div>
-        """
-        progress_html.update(bar_html)
-
-    thread = threading.Thread(target=progress_loop)
-    thread.start()
-
-    # --- actual model call ---
+def smooth_progress_bar():
+    """
+    Generates a smooth, natural-looking progress curve that slowly approaches 99%.
+    """
+    progress = gr.Progress(track_tqdm=False)
+
+    current = 0.0
+    while current < 0.99:
+        # Small random increments for a natural feel
+        increment = random.uniform(0.003, 0.007)  # ~0.3%–0.7%
+
+        # Speed adjusts as we get closer to 99%
+        delay = random.uniform(0.01, 0.03)
+
+        current = min(0.99, current + increment)
+        progress(current, desc=f"Editing image… {int(current*100)}%")
+
+        time.sleep(delay)
+
+    return progress
+
+
+def run_edit(input_image, prompt):
+    progress = smooth_progress_bar()
+
+    # Real model call
     output = client.image_to_image(
         input_image,
         prompt=prompt,
         model="Qwen/Qwen-Image-Edit",
     )
 
-    stop_flag["stop"] = True
-    thread.join()
-
+    # Instantly finish
+    progress(1.0, desc="Done!")
     return output
 
+
 # --- Gradio UI ---
 with gr.Blocks(theme="TejAndrewsACC/ACC", title="Image Edit Demo") as demo:
     gr.Markdown("## ACC AI Image Editor")
@@ -62,17 +58,15 @@ with gr.Blocks(theme="TejAndrewsACC/ACC", title="Image Edit Demo") as demo:
 
     prompt = gr.Textbox(
         label="How do you want the image to be changed?",
-        value=""  # empty by default
+        value=""
     )
 
-    progress_html = gr.HTML("<div style='width:100%; height:25px; background-color:#e0e0e0; border-radius:12px;'></div>")
-
     run_btn = gr.Button("Run Edit")
 
     run_btn.click(
-        fn=run_edit,
-        inputs=[input_img, prompt, progress_html],
+        fn=lambda img_path, pr: run_edit(open(img_path, "rb").read(), pr),
+        inputs=[input_img, prompt],
         outputs=[output_img],
     )
 
-demo.launch()
+demo.launch()