adding fake step on start

simple_app.py

@@ -14,12 +14,12 @@ snapshot_download(
 
 def infer(prompt, progress=gr.Progress(track_tqdm=True)):
     
-    # Configuration
+    # Configuration:
     total_process_steps = 11          # Total INFO messages expected
     irrelevant_steps = 4              # First 4 INFO messages are ignored
-    relevant_steps = total_process_steps - irrelevant_steps  # 7 overall (relevant) steps
+    relevant_steps = total_process_steps - irrelevant_steps  # 7 overall steps
 
-    # Create overall process progress bar (Level 1)
+    # Create overall progress bar (Level 1)
     overall_bar = tqdm(total=relevant_steps, desc="Overall Process", position=1,
                        ncols=120, dynamic_ncols=False, leave=True)
     processed_steps = 0
@@ -29,15 +29,16 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
     video_progress_bar = None
 
     # Variables for sub-step progress bar (Level 2)
-    # We use a tick total of 500 ticks = 20 seconds (each tick = 40ms)
+    # We use 500 ticks to represent 20 seconds (each tick = 40 ms)
     sub_bar = None
     sub_ticks = 0
     sub_tick_total = 500
-    # Flag indicating whether we're still waiting for the first relevant step.
+
+    # Flag to indicate we're still waiting for the first relevant INFO message.
     waiting_for_first_relevant = True
 
     command = [
-        "python", "-u", "-m", "generate",  # using -u for unbuffered output
+        "python", "-u", "-m", "generate",  # -u: unbuffered output
         "--task", "t2v-1.3B",
         "--size", "832*480",
         "--ckpt_dir", "./Wan2.1-T2V-1.3B",
@@ -54,7 +55,7 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
                                bufsize=1)
 
     while True:
-        # Poll stdout with a 40ms timeout.
+        # Poll stdout with a timeout of 40 ms.
         rlist, _, _ = select.select([process.stdout], [], [], 0.04)
         if rlist:
             line = process.stdout.readline()
@@ -64,10 +65,10 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
             if not stripped_line:
                 continue
 
-            # Check if line matches video generation progress (Level 3)
+            # Check for video generation progress (Level 3).
             progress_match = progress_pattern.search(stripped_line)
             if progress_match:
-                # Before entering video phase, cancel any active sub-step bar.
+                # If a sub-step bar is active, finish it before entering video phase.
                 if sub_bar is not None:
                     if sub_ticks < sub_tick_total:
                         sub_bar.update(sub_tick_total - sub_ticks)
@@ -76,7 +77,7 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
                     overall_bar.refresh()
                     sub_bar = None
                     sub_ticks = 0
-                # Enter video phase.
+                    waiting_for_first_relevant = False
                 current = int(progress_match.group(2))
                 total = int(progress_match.group(3))
                 if video_progress_bar is None:
@@ -84,7 +85,6 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
                                               ncols=120, dynamic_ncols=True, leave=True)
                 video_progress_bar.update(current - video_progress_bar.n)
                 video_progress_bar.refresh()
-                # When video generation completes, update overall bar.
                 if video_progress_bar.n >= video_progress_bar.total:
                     overall_bar.update(1)
                     overall_bar.refresh()
@@ -96,31 +96,40 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
             if "INFO:" in stripped_line:
                 parts = stripped_line.split("INFO:", 1)
                 msg = parts[1].strip() if len(parts) > 1 else ""
-                print(stripped_line)
+                print(stripped_line)  # Log the message
 
-                # For the first 4 INFO messages, we simply increment processed_steps.
                 if processed_steps < irrelevant_steps:
                     processed_steps += 1
-                    # If we're waiting for the first relevant step, start a waiting sub-bar if not already started.
+                    # While waiting for the first relevant INFO, ensure a waiting sub-bar is running.
                     if waiting_for_first_relevant and sub_bar is None:
                         sub_bar = tqdm(total=sub_tick_total, desc="Waiting for first step...", position=2,
                                        ncols=120, dynamic_ncols=False, leave=True)
                         sub_ticks = 0
-                    # Continue reading logs.
                     continue
                 else:
-                    # Now we are in the relevant phase.
-                    waiting_for_first_relevant = False
-                    # If a sub-bar exists (either waiting or from a previous step), finish it.
-                    if sub_bar is not None:
-                        if sub_ticks < sub_tick_total:
-                            sub_bar.update(sub_tick_total - sub_ticks)
-                        sub_bar.close()
-                        overall_bar.update(1)
-                        overall_bar.refresh()
-                        sub_bar = None
-                        sub_ticks = 0
-                    # Start a new sub-step bar with the current INFO message.
+                    # We have reached the first relevant INFO message.
+                    if waiting_for_first_relevant:
+                        waiting_for_first_relevant = False
+                        # Cancel the waiting sub-bar.
+                        if sub_bar is not None:
+                            if sub_ticks < sub_tick_total:
+                                sub_bar.update(sub_tick_total - sub_ticks)
+                            sub_bar.close()
+                            overall_bar.update(1)
+                            overall_bar.refresh()
+                            sub_bar = None
+                            sub_ticks = 0
+                    else:
+                        # If a sub-bar is active from a previous step, finish it.
+                        if sub_bar is not None:
+                            if sub_ticks < sub_tick_total:
+                                sub_bar.update(sub_tick_total - sub_ticks)
+                            sub_bar.close()
+                            overall_bar.update(1)
+                            overall_bar.refresh()
+                            sub_bar = None
+                            sub_ticks = 0
+                    # Now start a new sub-step bar for the current INFO message.
                     sub_bar = tqdm(total=sub_tick_total, desc=msg, position=2,
                                    ncols=120, dynamic_ncols=False, leave=True)
                     sub_ticks = 0
@@ -128,20 +137,22 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
             else:
                 print(stripped_line)
         else:
-            # No new data within 40ms.
-            # If a sub-bar is active, update it.
+            # No new stdout data within 40 ms.
+            # If we're waiting for the first relevant step and no sub-bar exists, create it.
+            if waiting_for_first_relevant and sub_bar is None:
+                sub_bar = tqdm(total=sub_tick_total, desc="Waiting for first step...", position=2,
+                               ncols=120, dynamic_ncols=False, leave=True)
+                sub_ticks = 0
             if sub_bar is not None:
                 sub_bar.update(1)
                 sub_ticks += 1
                 sub_bar.refresh()
                 if sub_ticks >= sub_tick_total:
-                    # 20 seconds have elapsed; finish this sub-step.
                     sub_bar.close()
                     overall_bar.update(1)
                     overall_bar.refresh()
                     sub_bar = None
                     sub_ticks = 0
-
         if process.poll() is not None:
             break