Update simple_app.py

simple_app.py

@@ -15,7 +15,7 @@ snapshot_download(
     repo_id="Wan-AI/Wan2.1-T2V-1.3B",
     local_dir="./Wan2.1-T2V-1.3B"
 )
-print("Model downloaded successfully.")
+print("✅ Model downloaded successfully.")
 
 def infer(prompt, progress=gr.Progress(track_tqdm=True)):
     total_process_steps = 11
@@ -34,8 +34,9 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
     sub_tick_total = 1500
     video_phase = False
 
+    # ✅ Use generate.py directly
     command = [
-        "python", "-u", "-m", "generate",
+        "python", "-u", "generate.py",  # <- Make sure generate.py is in the same folder
         "--task", "t2v-1.3B",
         "--size", "480*480",
         "--ckpt_dir", "./Wan2.1-T2V-1.3B",
@@ -47,7 +48,7 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
         "--save_file", "generated_video.mp4"
     ]
 
-    print("Starting video generation process...")
+    print("🚀 Starting video generation process...")
     process = subprocess.Popen(
         command,
         stdout=subprocess.PIPE,
@@ -60,9 +61,12 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
 
     for line in iter(stdout.readline, ''):
         stripped_line = line.strip()
+        print(f"[SUBPROCESS]: {stripped_line}")  # Debug print
+
         if not stripped_line:
             continue
 
+        # Match video generation progress (like tqdm)
         progress_match = progress_pattern.search(stripped_line)
         if progress_match:
             if sub_bar is not None and sub_ticks < sub_tick_total:
@@ -89,7 +93,6 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
             parts = stripped_line.split("INFO:", 1)
             msg = parts[1].strip() if len(parts) > 1 else ""
             print(f"[INFO]: {msg}")
-
             if processed_steps < irrelevant_steps:
                 processed_steps += 1
                 continue
@@ -104,31 +107,35 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
                                ncols=120, dynamic_ncols=False, leave=True)
                 sub_ticks = 0
             continue
-        else:
-            print(stripped_line)
 
     process.wait()
 
+    # Final cleanup
     if video_progress_bar is not None:
         video_progress_bar.close()
     if sub_bar is not None:
         sub_bar.close()
     overall_bar.close()
 
-    if process.returncode == 0 and os.path.exists("generated_video.mp4"):
-        print("Video generation completed successfully.")
-        return "generated_video.mp4"
+    # ✅ Output validation
+    if process.returncode == 0:
+        if os.path.exists("generated_video.mp4"):
+            print("✅ Video generation completed successfully.")
+            return "generated_video.mp4"
+        else:
+            print("❌ Video generation finished but output file is missing.")
+            raise gr.Error("Output video not found after generation.")
     else:
-        print("Error: Video generation failed.")
-        raise gr.Error("Video generation failed. Check logs for details.")
+        print("❌ Subprocess failed.")
+        raise gr.Error("Video generation failed. Check logs above.")
 
-# Gradio UI
+# ✅ Gradio UI
 with gr.Blocks() as demo:
     with gr.Column():
-        gr.Markdown("# Wan 2.1 1.3B")
-        gr.Markdown("Enjoy this simple working UI, duplicate the space to skip the queue :)")
-        prompt = gr.Textbox(label="Prompt")
-        submit_btn = gr.Button("Submit")
+        gr.Markdown("# Wan 2.1 1.3B - Text to Video")
+        gr.Markdown("Generate short videos from prompts. Duplicate this space to avoid queue limits.")
+        prompt = gr.Textbox(label="Enter your prompt")
+        submit_btn = gr.Button("Generate Video")
         video_res = gr.Video(label="Generated Video")
 
     submit_btn.click(