Update app.py

app.py

@@ -1,32 +1,66 @@
-import streamlit as st
+import gradio as gr
 import subprocess
 import os
 
-# Title
-st.title("🎥 WAN 2.1 - 14B AI Text-to-Video Generator")
-
-# Input fields
-prompt = st.text_area("Enter your text prompt:", "A cat in military dress wearing headphones, laughing and walking.")
-frame_num = st.slider("Number of frames:", min_value=30, max_value=120, value=60, step=10)
-resolution = st.selectbox("Select resolution:", ["832*480", "1280*720"])
-sample_steps = st.slider("Sampling steps:", min_value=10, max_value=50, value=20, step=5)
-
-# Button to generate video
-if st.button("Generate Video"):
-    st.info("Generating video... This may take a few minutes.")
-
-    # Run WAN 2.1 - 14B Model
-    command = f"python generate.py --task t2v-14B --size {resolution} --frame_num {frame_num} --sample_steps {sample_steps} --ckpt_dir ./Wan2.1-T2V-14B --offload_model True --prompt \"{prompt}\""
+def generate_video(prompt, frame_num, resolution, sample_steps):
+    # Inform user that generation has started
+    status = "Generating video... This may take a few minutes."
+    
+    # Build command with proper quoting for the prompt
+    command = (
+        f"python generate.py --task t2v-14B --size {resolution} "
+        f"--frame_num {frame_num} --sample_steps {sample_steps} "
+        f"--ckpt_dir ./Wan2.1-T2V-14B --offload_model True --prompt \"{prompt}\""
+    )
     
+    # Run the command and capture stdout and stderr
     process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     stdout, stderr = process.communicate()
-
-    # Print logs for debugging
-    st.text_area("📜 Logs", stdout.decode() + stderr.decode())
-
-    # Check if video was created
+    logs = stdout.decode() + stderr.decode()
+    
+    # Check if video file exists
     if os.path.exists("output.mp4"):
-        st.video("output.mp4")
-        st.success("✅ Video generated successfully!")
+        video_file = "output.mp4"
+        status = "✅ Video generated successfully!"
     else:
-        st.error("❌ Video generation failed! Check logs above.")
+        video_file = None
+        status = "❌ Video generation failed! Check logs above."
+        
+    return video_file, logs, status
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🎥 WAN 2.1 - 14B AI Text-to-Video Generator")
+    
+    with gr.Row():
+        prompt_input = gr.Textbox(
+            label="Enter your text prompt:",
+            value="A cat in military dress wearing headphones, laughing and walking.",
+            lines=4
+        )
+    
+    with gr.Row():
+        frame_slider = gr.Slider(30, 120, step=10, value=60, label="Number of frames:")
+    
+    with gr.Row():
+        resolution_choice = gr.Radio(
+            choices=["832*480", "1280*720"],
+            label="Select resolution:",
+            value="832*480"
+        )
+    
+    with gr.Row():
+        steps_slider = gr.Slider(10, 50, step=5, value=20, label="Sampling steps:")
+    
+    generate_button = gr.Button("Generate Video")
+    
+    video_output = gr.Video(label="Generated Video")
+    logs_output = gr.Textbox(label="📜 Logs", lines=10)
+    status_output = gr.Textbox(label="Status")
+    
+    generate_button.click(
+        generate_video,
+        inputs=[prompt_input, frame_slider, resolution_choice, steps_slider],
+        outputs=[video_output, logs_output, status_output]
+    )
+
+demo.launch()