Update app.py

app.py

@@ -5,17 +5,36 @@ from modelscope.utils.constant import Tasks
 # Load ModelScope's Text-to-Video Model
 video_model = pipeline(Tasks.text_to_video_synthesis, model='damo/text-to-video-synthesis')
 
-def generate_video(prompt):
-    result = video_model({'text': prompt})
+def generate_video(prompt, style, duration, image):
+    # Modify prompt with style
+    prompt = f"{prompt}, in {style} style"
+    
+    # Prepare inputs
+    inputs = {'text': prompt, 'duration': duration}
+    
+    if image:
+        inputs['image'] = image  # Use uploaded image if provided
+
+    # Generate video
+    result = video_model(inputs)
     return result["output_video"]
 
 # Gradio UI
-iface = gr.Interface(
-    fn=generate_video,
-    inputs=gr.Textbox(label="Enter text prompt"),
-    outputs=gr.Video(label="Generated Video"),
-    title="AI Text-to-Video Generator",
-    description="Type a prompt, and AI will create a video for you!",
-)
+with gr.Blocks() as iface:
+    gr.Markdown("# 🎥 AI Text-to-Video Generator")
+    gr.Markdown("Enter a prompt, choose a style, and generate a video!")
+
+    with gr.Row():
+        prompt = gr.Textbox(label="Enter text prompt", placeholder="A futuristic city with flying cars")
+        style = gr.Dropdown(["Realistic", "Cartoon", "Anime"], label="Select Video Style", value="Realistic")
+
+    duration = gr.Slider(1, 10, step=1, label="Video Duration (Seconds)", value=5)
+    image = gr.Image(label="Upload an Image (Optional)", type="filepath")
+
+    generate_btn = gr.Button("Generate Video")
+
+    output_video = gr.Video(label="Generated Video")
+
+    generate_btn.click(generate_video, inputs=[prompt, style, duration, image], outputs=output_video)
 
 iface.launch()