Update app.py

app.py

@@ -1,6 +1,5 @@
 import threading
 import time
-import random
 import os
 import gradio as gr
 from huggingface_hub import InferenceClient
@@ -11,31 +10,23 @@ client = InferenceClient(
     api_key=os.environ["HF_TOKEN"],
 )
 
-def run_edit(input_image, prompt, progress=gr.Progress()):
+def run_edit(input_image, prompt, progress_bar):
     stop_flag = {"stop": False}
 
     # --- background thread for smooth progress bar ---
     def progress_loop():
-        start = time.time()
         current = 0.0
-        total_time_guess = 10  # seconds, adjust to your model latency
+        total_time_guess = 10  # seconds, adjust to typical inference
 
         while not stop_flag["stop"]:
-            elapsed = time.time() - start
-
-            # Time-shaped curve approaching 99%
-            target = min(0.99, (elapsed / total_time_guess) ** 0.7)
-
-            # Smoothly move toward target
-            current = current * 0.8 + target * 0.2
-
-            progress(current, desc=f"Editing Image… {int(current*100)}%")
+            # Smooth progress approach
+            current += (0.99 - current) * 0.02
+            progress_bar.update(value=current, label=f"Editing Image… {int(current*100)}%")
             time.sleep(0.05)
 
-        # Finish immediately when model is done
-        progress(1.0, desc="Done!")
+        # Finish immediately
+        progress_bar.update(value=1.0, label="Done!")
 
-    # Start progress bar thread
     thread = threading.Thread(target=progress_loop)
     thread.start()
 
@@ -46,7 +37,6 @@ def run_edit(input_image, prompt, progress=gr.Progress()):
         model="Qwen/Qwen-Image-Edit",
     )
 
-    # Signal progress bar to finish
     stop_flag["stop"] = True
     thread.join()
 
@@ -62,15 +52,29 @@ 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="..."
+        value=""  # empty placeholder
     )
 
+    progress_bar = gr.Progress(label="Editing Image…", value=0.0)
+
     run_btn = gr.Button("Run Edit")
 
     run_btn.click(
         fn=run_edit,
-        inputs=[input_img, prompt],
+        inputs=[input_img, prompt, progress_bar],
         outputs=[output_img],
     )
 
-demo.launch()
+# Optional CSS styling for the progress bar
+custom_css = """
+#component-3 .gradio-progress-bar {
+    height: 25px;
+    border-radius: 12px;
+    background-color: #e0e0e0;
+}
+#component-3 .gradio-progress-bar-fill {
+    background-color: #4CAF50;
+    transition: width 0.05s linear;
+}
+"""
+demo.launch(share=True, inline=False, css=custom_css)