fix models

app.py

@@ -59,6 +59,11 @@ def create_video_interface():
                     value="Meta-Llama-3.1-70B-Instruct",
                     label="Model"
                 )
+                prompt_length = gr.Dropdown(
+                    choices=["Short", "Medium", "Long"],
+                    value="Medium",
+                    label="Prompt Length"
+                )
 
         generate_btn = gr.Button("Generate Video Prompt", variant="primary")
         output = gr.Textbox(label="Generated Prompt", lines=12, show_copy_button=True)
@@ -78,7 +83,8 @@ def create_video_interface():
 
         generate_btn.click(
             llm_node.generate_video_prompt,
-            inputs=[input_concept, style, camera_style, pacing, special_effects, custom_elements, provider, model],
+            inputs=[input_concept, style, camera_style, pacing, special_effects, 
+                   custom_elements, provider, model, prompt_length],
             outputs=output
         )
 

llm_inference_video.py

@@ -24,7 +24,8 @@ class VideoLLMInferenceNode:
         special_effects,
         custom_elements,
         provider="SambaNova",
-        model=None
+        model=None,
+        prompt_length="Medium"
     ):
         try:
             # Video prompt templates
@@ -42,13 +43,42 @@ class VideoLLMInferenceNode:
             }
 
             base_prompt = prompt_templates.get(style.lower(), prompt_templates["cinematic"])
-            system_message = """You are a visionary video director and creative storyteller. Create a single, richly detailed paragraph that paints a complete picture of the video concept. Focus on:
-1. Overall visual atmosphere and mood
-2. Narrative flow and story progression
-3. Distinctive visual style and aesthetic choices
-4. Key moments and visual highlights
-5. Emotional impact and audience experience
-Avoid technical specifications or shot-by-shot breakdowns. Instead, create a flowing, descriptive narrative that captures the essence of the video."""
+
+            # Configure length requirements
+            length_config = {
+                "Short": {
+                    "guidance": "Create exactly ONE impactful sentence that captures the essence of the video. Be concise but descriptive.",
+                    "structure": "Combine all elements into a single, powerful sentence."
+                },
+                "Medium": {
+                    "guidance": "Create 2-3 flowing sentences that paint a picture of the video.",
+                    "structure": "First sentence should set the scene, followed by 1-2 sentences developing the concept."
+                },
+                "Long": {
+                    "guidance": "Create 4-5 detailed sentences that thoroughly describe the video.",
+                    "structure": "Begin with the setting, develop the action/movement, and conclude with impact."
+                }
+            }
+            
+            config = length_config[prompt_length]
+            
+            system_message = f"""You are a visionary video director and creative storyteller. {config['guidance']}
+
+Structure: {config['structure']}
+
+Focus on these elements while maintaining the specified sentence count:
+1. Visual atmosphere and mood
+2. Narrative flow
+3. Style and aesthetic choices
+4. Key moments
+5. Emotional impact
+
+Important: Deliver exactly the specified number of sentences:
+- Short: ONE sentence
+- Medium: TWO to THREE sentences
+- Long: FOUR to FIVE sentences
+
+Keep everything in a single paragraph format. Avoid technical specifications or shot lists."""
 
             # Select provider
             if provider == "Groq":
@@ -65,7 +95,6 @@ Avoid technical specifications or shot-by-shot breakdowns. Instead, create a flo
                     {"role": "user", "content": f"{base_prompt}\nCore Concept: {input_concept}"}
                 ],
                 temperature=1.2,
-                max_tokens=1500,
                 top_p=0.95,
                 seed=random.randint(0, 10000)
             )