Update app.py

app.py

@@ -4,12 +4,10 @@ import os
 
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-SYSTEM_PROMPT = """You are an expert storyboard generator. Create detailed storyboards with:
-- Shot numbers, visual descriptions, camera angles
-- Actions, dialogue, sound effects, timing
-Format as markdown tables. Be creative and cinematic."""
+SYSTEM_PROMPT = """You are an expert in storyboarding. Provide structured and insightful responses to queries 
+about creating and refining storyboards"""
 
-def respond(message, history):
+def respond(message, history, model, temperature, max_tokens):
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
     
     for h in history:
@@ -21,16 +19,16 @@ def respond(message, history):
     
     try:
         response = client.chat.completions.create(
-            model="llama-3.3-70b-versatile",
+            model=model,
             messages=messages,
-            temperature=0.9,
-            max_completion_tokens=2048,
+            temperature=temperature,
+            max_completion_tokens=max_tokens,
         )
         return response.choices[0].message.content
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Simple ChatInterface - most compatible
+# ChatInterface with additional inputs for parameters
 demo = gr.ChatInterface(
     fn=respond,
     title="🎬 Storyboard Generator AI",
@@ -40,6 +38,36 @@ demo = gr.ChatInterface(
         "Generate a horror movie opening scene storyboard",
         "Design a storyboard for a romantic comedy meet-cute at a bookstore",
     ],
+    additional_inputs=[
+        gr.Dropdown(
+            choices=[
+                "llama-3.3-70b-versatile",
+                "llama-3.1-70b-versatile",
+                "llama-3.1-8b-instant",
+                "mixtral-8x7b-32768",
+                "gemma2-9b-it"
+            ],
+            value="llama-3.3-70b-versatile",
+            label="Model",
+            info="Select the AI model to use"
+        ),
+        gr.Slider(
+            minimum=0,
+            maximum=2,
+            value=0.9,
+            step=0.1,
+            label="Temperature",
+            info="Controls randomness. Lower = more focused, Higher = more creative"
+        ),
+        gr.Slider(
+            minimum=256,
+            maximum=8192,
+            value=2048,
+            step=256,
+            label="Max Tokens",
+            info="Maximum length of the response"
+        ),
+    ],
     theme="soft",
 )