Update app.py

app.py

@@ -2,13 +2,14 @@ import gradio as gr
 from PIL import Image
 import time
 import threading
+import tqdm
 # import get_wsl
 
 
 # 所有button响应的函数
 
 # 任务处理函数
-def process_task(txt, inimage, p1, p2, p3, p4, progress=gr.Progress()):
+def process_task(txt, inimage, p1, p2, p3, p4, progress=gr.Progress(track_tqdm=True)):
     # 启动一个 15 秒的命令模拟
     def execute_command():
         # 这里用time.sleep(15)来模拟命令执行
@@ -20,15 +21,8 @@ def process_task(txt, inimage, p1, p2, p3, p4, progress=gr.Progress()):
     command_thread = threading.Thread(target=execute_command)
     command_thread.start()
 
-    # 启动进度条（匀速前进）
-    total_time = 15  # 总时间设置为15秒
-    num_steps = 60  # 15秒内分60步，每步0.25秒
-    step_time = total_time / num_steps  # 每次更新进度条的时间间隔
-
-    # 让进度条匀速前进
-    for i in range(num_steps):
-        progress.update(i / (num_steps - 1))  # 只传递值，不带desc
-        time.sleep(step_time)  # 等待一段时间，保持进度条匀速前进
+    for i in progress.tqdm(range(10)): # 这里填进度条段数
+        time.sleep(1) # 这里填多久改变一次
 
     command_thread.join()  # 等待命令执行完成
     return Image.open('img/bg.png'), gr.update(visible=True)