Update simple_app.py

simple_app.py

@@ -62,19 +62,24 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
 
         # Check for INFO lines.
         if "INFO:" in stripped_line:
+            # Extract the text after "INFO:"
             parts = stripped_line.split("INFO:", 1)
             msg = parts[1].strip() if len(parts) > 1 else ""
-            tqdm.write(stripped_line)  # Use tqdm.write() instead of print()
+            # Print the log line.
+            print(stripped_line)
+            # Skip updating the overall bar for the first few irrelevant steps.
             if processed_steps < irrelevant_steps:
                 processed_steps += 1
             else:
+                # Create a sub-progress bar with a total of 1 for this step.
+                sub_bar = tqdm(total=1, desc=msg, position=2, ncols=120, dynamic_ncols=False, leave=True)
+                sub_bar.update(1)
+                sub_bar.close()
+
+                # Update the overall progress bar.
                 overall_bar.update(1)
-                percentage = (overall_bar.n / overall_bar.total) * 100
-                # Combine percentage and INFO message into one description.
-                overall_bar.desc = f"Overall Process MICHEL- {percentage:.1f}% | {msg}"
                 overall_bar.refresh()
         else:
-            # Print any other lines.
             print(stripped_line)
 
     process.wait()