Update simple_app.py

simple_app.py

@@ -16,7 +16,7 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
     irrelevant_steps = 4
     relevant_steps = total_process_steps - irrelevant_steps  # 7 steps
 
-    # Create an overall progress bar for the 9 relevant steps.
+    # Create an overall progress bar for the relevant steps.
     overall_bar = tqdm(total=relevant_steps, desc="Overall Process", position=1, dynamic_ncols=True, leave=True)
     processed_steps = 0
 
@@ -24,7 +24,7 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
     info_pattern = re.compile(r"\[.*?\]\s+INFO:\s+(.*)")
     # Regex to capture video generation progress lines (e.g., " 10%|...| 5/50")
     progress_pattern = re.compile(r"(\d+)%\|.*\| (\d+)/(\d+)")
-    
+
     gen_progress_bar = None
 
     command = [
@@ -59,7 +59,7 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
             total = int(progress_match.group(3))
             if gen_progress_bar is None:
                 gen_progress_bar = tqdm(total=total, desc="Video Generation", position=0, dynamic_ncols=True, leave=True)
-            # Update the video generation progress bar
+            # Update video generation progress.
             gen_progress_bar.update(current - gen_progress_bar.n)
             gen_progress_bar.refresh()
             continue  # Skip further processing for progress lines
@@ -68,20 +68,19 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
         info_match = info_pattern.search(stripped_line)
         if info_match:
             msg = info_match.group(1)
-            # Always print the log line (using tqdm.write so it doesn't interfere with the bars)
+            # Print the raw log line.
             tqdm.write(stripped_line)
-            # For relevant steps (i.e. after the first three), update the overall progress.
+            # For relevant steps (after the first three), update the overall progress bar.
             if processed_steps < irrelevant_steps:
                 processed_steps += 1
             else:
                 overall_bar.update(1)
                 percentage = (overall_bar.n / overall_bar.total) * 100
-                # Update the description for the left part and set the postfix to the INFO message.
-                overall_bar.set_description(f"Overall Process - {percentage:.1f}%")
-                overall_bar.set_postfix_str(msg)
+                # Directly insert the INFO message in the description.
+                overall_bar.set_description(f"Overall Process - {percentage:.1f}% | {msg}")
                 overall_bar.refresh()
         else:
-            # For any other line, print it.
+            # For any other line, simply print it.
             tqdm.write(stripped_line)
 
     process.wait()