Update simple_app.py

simple_app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import re 
+import re
 import subprocess
 import time
 import select
@@ -16,6 +16,7 @@ snapshot_download(
     local_dir="./Wan2.1-T2V-1.3B"
 )
 print("Model downloaded successfully.")
+
 def infer(prompt, progress=gr.Progress(track_tqdm=True)):
     # Configuration:
     total_process_steps = 11          # Total INFO messages expected
@@ -32,7 +33,6 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
     video_progress_bar = None
 
     # Variables for sub-step progress bar (Level 2)
-    # Now using 1000 ticks to represent 40 seconds (each tick = 40 ms)
     sub_bar = None
     sub_ticks = 0
     sub_tick_total = 1500
@@ -51,8 +51,6 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
     ]
     print("Starting video generation process...")
 
-
-
     process = subprocess.Popen(command,
                                stdout=subprocess.PIPE,
                                stderr=subprocess.STDOUT,
@@ -74,8 +72,8 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
 
             # Check for video generation progress (Level 3)
             progress_match = progress_pattern.search(stripped_line)
+            
             if progress_match:
-                # 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)
@@ -112,7 +110,6 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
                     continue
                 else:
                     # A new relevant INFO message has arrived.
-                    # If a sub-bar exists (whether full or not), finish it now.
                     if sub_bar is not None:
                         if sub_ticks < sub_tick_total:
                             sub_bar.update(sub_tick_total - sub_ticks)
@@ -135,7 +132,6 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
                     sub_bar.update(1)
                     sub_ticks += 1
                     sub_bar.refresh()
-                # If full (40 seconds reached), do not advance overall step—just remain waiting.
         if process.poll() is not None:
             break
 
@@ -148,9 +144,10 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
     if sub_bar is not None:
         sub_bar.close()
     overall_bar.close()
-    
+
+    # Add log for successful video generation
     if process.returncode == 0:
-        print("Command executed successfullyg.")
+        print("Video generation completed successfully.")
         return "generated_video.mp4"
     else:
         print("Error executing command.")
@@ -171,4 +168,4 @@ with gr.Blocks() as demo:
         outputs=[video_res]
     )
 
-demo.queue().launch(show_error=True, show_api=False, ssr_mode=False)
+demo.queue().launch(show_error=True, show_api=False, ssr_mode=False)