Update app.py

app.py

@@ -71,14 +71,99 @@ def generate_video(
     generation_id = uuid.uuid4().hex[:8]
     print(f"Generation ID: {generation_id}")
 
-    # Prepare parameters for the video generation request
-    # Note: Different providers may have different parameter requirements
-    parameters = {
-        "prompt": prompt,
+    # Initial parameters dictionary - we'll customize it for each provider
+    base_parameters = {
         "negative_prompt": negative_prompt,
-        "num_frames": num_frames,
-        "fps": fps,
-        "width": width,
+        "num_inference_steps": num_inference_steps,
+        "guidance_scale": guidance_scale,
+    }
+    
+    # Add seed if specified
+    if seed is not None and seed != -1:
+        base_parameters["seed"] = seed
+    
+    # Provider-specific parameter handling
+    if provider == "hf-inference":
+        print("Using Hugging Face Inference API, adapting parameters...")
+        # HF Inference API supports these parameters
+        parameters = {
+            "negative_prompt": negative_prompt,
+            "num_inference_steps": num_inference_steps,
+            "guidance_scale": guidance_scale,
+            "num_frames": num_frames,
+            # HF Inference does NOT support fps
+            # "fps": fps,  
+            "width": width,
+            "height": height,
+        }
+        
+        # Add motion_bucket_id for SVD models if applicable
+        if "stable-video-diffusion" in model_to_use:
+            parameters["motion_bucket_id"] = motion_bucket_id
+            
+        # Add seed if specified
+        if seed is not None and seed != -1:
+            parameters["seed"] = seed
+    
+    elif provider == "fal-ai":
+        print("Using FalAI provider, adapting parameters...")
+        # According to Fal-AI API specification, only these parameters are supported
+        parameters = {
+            "negative_prompt": negative_prompt,
+            "num_frames": num_frames,
+            "num_inference_steps": num_inference_steps,
+            "guidance_scale": guidance_scale,
+        }
+        
+        # Add seed if specified
+        if seed is not None and seed != -1:
+            parameters["seed"] = seed
+        
+        # Note: width and height are not supported by Fal-AI's text_to_video API
+        print("Note: width and height parameters are not supported by Fal-AI and will be ignored")
+    
+    elif provider == "novita":
+        print("Using Novita provider, adapting parameters...")
+        # Based on documentation, Novita uses specific parameters
+        parameters = {
+            "negative_prompt": negative_prompt,
+            "num_frames": num_frames,
+            "num_inference_steps": num_inference_steps,
+            "guidance_scale": guidance_scale,
+            "width": width,
+            "height": height
+        }
+        
+        # Add seed if specified
+        if seed is not None and seed != -1:
+            parameters["seed"] = seed
+    
+    elif provider == "replicate":
+        print("Using Replicate provider, adapting parameters...")
+        parameters = {
+            "negative_prompt": negative_prompt,
+            "num_frames": num_frames,
+            "num_inference_steps": num_inference_steps,
+            "guidance_scale": guidance_scale,
+            "width": width,
+            "height": height
+        }
+        
+        # Add seed if specified
+        if seed is not None and seed != -1:
+            parameters["seed"] = seed
+        
+        # Replicate supports fps in some models
+        if "stable-video-diffusion" in model_to_use:
+            parameters["fps"] = fps
+    
+    else:
+        # Default parameters for any other provider
+        print(f"Using generic parameters for provider: {provider}")
+        parameters = base_parameters
+        parameters["num_frames"] = num_frames
+        parameters["width"] = width
+        parameters["height"] = height
         "height": height,
         "num_inference_steps": num_inference_steps,
         "guidance_scale": guidance_scale,
@@ -284,6 +369,40 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme") as demo:
                     lines=2
                 )
                 
+                with gr.Row():
+                    num_inference_steps = gr.Slider(
+                        minimum=1,
+                        maximum=100,
+                        value=25,
+                        step=1,
+                        label="Inference Steps"
+                    )
+                    
+                    guidance_scale = gr.Slider(
+                        minimum=1.0,
+                        maximum=20.0,
+                        value=7.5,
+                        step=0.5,
+                        label="Guidance Scale"
+                    )
+                
+                with gr.Row():
+                    motion_bucket_id = gr.Slider(
+                        minimum=1,
+                        maximum=255,
+                        value=127,
+                        step=1,
+                        label="Motion Bucket ID (for SVD models)"
+                    )
+                    
+                    seed = gr.Slider(
+                        minimum=-1,
+                        maximum=2147483647,
+                        value=-1,
+                        step=1,
+                        label="Seed (-1 for random)"
+                    )
+                
                 with gr.Row():
                     width = gr.Slider(
                         minimum=256,