Add advanced video generation parameters to Gradio interface

app.py

@@ -14,7 +14,7 @@ TRANSFORMER_MODELS = [
 ]
 
 @spaces.GPU
-def generate_video(transformer_model, prompt, negative_prompt):
+def generate_video(transformer_model, prompt, negative_prompt, num_frames, height, width, num_inference_steps):
     # Cargar el modelo del transformer seleccionado
     transformer = CogVideoXTransformer3DModel.from_pretrained(
         transformer_model, 
@@ -32,10 +32,10 @@ def generate_video(transformer_model, prompt, negative_prompt):
     video_frames = pipeline(
         prompt=prompt,
         negative_prompt=negative_prompt,
-        num_frames=81,
-        height=512, 
-        width=768,
-        num_inference_steps=50
+        num_frames=num_frames,
+        height=height, 
+        width=width,
+        num_inference_steps=num_inference_steps
     ).frames[0]
     
     # Guardar el video en un archivo temporal
@@ -65,6 +65,41 @@ with gr.Blocks() as demo:
                 label="Prompt Negativo",
                 value="inconsistent motion, blurry motion, worse quality, degenerate outputs, deformed outputs"
             )
+            
+            with gr.Accordion("Parámetros Avanzados", open=False):
+                num_frames = gr.Slider(
+                    minimum=8,
+                    maximum=128,
+                    value=8,
+                    step=1,
+                    label="Número de Frames",
+                    info="Cantidad de frames en el video"
+                )
+                height = gr.Slider(
+                    minimum=32,
+                    maximum=1024,
+                    value=32,
+                    step=64,
+                    label="Altura",
+                    info="Altura del video en píxeles"
+                )
+                width = gr.Slider(
+                    minimum=32,
+                    maximum=1024,
+                    value=32,
+                    step=64,
+                    label="Anchura",
+                    info="Anchura del video en píxeles"
+                )
+                num_inference_steps = gr.Slider(
+                    minimum=10,
+                    maximum=100,
+                    value=10,
+                    step=1,
+                    label="Pasos de Inferencia",
+                    info="Mayor número = mejor calidad pero más lento"
+                )
+            
             generate_btn = gr.Button("Generar Video")
         
         with gr.Column():
@@ -74,7 +109,15 @@ with gr.Blocks() as demo:
     # Conectar la función
     generate_btn.click(
         fn=generate_video,
-        inputs=[model_dropdown, prompt_input, negative_prompt_input],
+        inputs=[
+            model_dropdown, 
+            prompt_input, 
+            negative_prompt_input,
+            num_frames,
+            height,
+            width,
+            num_inference_steps
+        ],
         outputs=video_output
     )