Update app.py

app.py

@@ -42,20 +42,41 @@ def create_video_interface():
                 custom_elements = gr.Textbox(label="Custom Technical Elements", 
                                            placeholder="e.g., Infrared hybrid, Datamosh transitions")
                 
+        with gr.Row():
+            custom_elements = gr.Textbox(label="Custom Elements", scale=3)
+            with gr.Column(scale=1):
                 provider = gr.Dropdown(
                     choices=["Hugging Face", "Groq", "SambaNova"],
                     value="Hugging Face",
                     label="LLM Provider"
                 )
-                model = gr.Dropdown(label="Model", value="meta-llama/Meta-Llama-3.1-70B-Instruct")
+                model = gr.Dropdown(label="Model", value="Qwen/Qwen2.5-72B-Instruct")
+
+        generate_btn = gr.Button("Generate Video Prompt", variant="primary")
+        output = gr.Textbox(label="Generated Prompt", lines=12, show_copy_button=True)
 
-        generate_btn = gr.Button("Generate Video Blueprint", variant="primary")
-        output = gr.Textbox(label="Video Production Prompt", lines=12, interactive=False)
+        def update_models(provider):
+            models = {
+                "Hugging Face": [
+                    "Qwen/Qwen2.5-72B-Instruct",
+                    "meta-llama/Meta-Llama-3.1-70B-Instruct",
+                    "mistralai/Mixtral-8x7B-Instruct-v0.1",
+                    "mistralai/Mistral-7B-Instruct-v0.3"
+                ],
+                "Groq": ["llama-3.1-70b-versatile"],
+                "SambaNova": [
+                    "Meta-Llama-3.1-70B-Instruct",
+                    "Meta-Llama-3.1-405B-Instruct",
+                    "Meta-Llama-3.1-8B-Instruct"
+                ]
+            }
+            return gr.Dropdown(choices=models[provider], value=models[provider][0])
+
+        provider.change(update_models, inputs=provider, outputs=model)
 
         generate_btn.click(
             llm_node.generate_video_prompt,
-            inputs=[input_concept, duration, style, camera_style, pacing, special_effects, 
-                   custom_elements, provider, model],
+            inputs=[input_concept, duration, style, camera_style, pacing, special_effects, custom_elements, provider, model],
             outputs=output
         )
 
@@ -63,4 +84,6 @@ def create_video_interface():
 
 if __name__ == "__main__":
     demo = create_video_interface()
-    demo.launch(share=True)
+    demo.launch(share=True)
+
+    return demo