Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from groq import Groq
 
 # -----------------------------
-# 1. Initialize Groq client
+# Initialize Groq
 # -----------------------------
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
@@ -13,13 +13,12 @@ SYSTEM_PROMPT = (
 )
 
 # -----------------------------
-# 2. Chat function
+# Chat function
 # -----------------------------
 def chat_with_bot(user_input, history):
     if history is None:
         history = []
 
-    # Add system message once
     if not history:
         history.append({"role": "system", "content": SYSTEM_PROMPT})
 
@@ -28,103 +27,67 @@ def chat_with_bot(user_input, history):
     completion = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
         messages=history,
+        max_tokens=512,
         temperature=1,
-        max_tokens=1024,
     )
 
-    assistant_reply = completion.choices[0].message.content
-    history.append({"role": "assistant", "content": assistant_reply})
+    reply = completion.choices[0].message.content
+    history.append({"role": "assistant", "content": reply})
 
     return history, history
 
 
 # -----------------------------
-# 3. Storyboard generator
+# Storyboard generator
 # -----------------------------
 def generate_storyboard(scenario):
     if not scenario.strip():
-        return "❗ Please enter a scenario."
-
-    messages = [
-        {
-            "role": "system",
-            "content": "Generate a 6-scene storyboard in a table format."
-        },
-        {
-            "role": "user",
-            "content": scenario
-        }
-    ]
+        return "Please enter a scenario."
 
     completion = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
-        messages=messages,
-        temperature=1,
-        max_tokens=1024,
+        messages=[
+            {"role": "system", "content": "Generate a 6-scene storyboard in table format."},
+            {"role": "user", "content": scenario},
+        ],
+        max_tokens=700,
     )
 
     return completion.choices[0].message.content
 
 
 # -----------------------------
-# 4. Gradio UI
+# UI
 # -----------------------------
-with gr.Blocks(
-    theme=gr.themes.Glass(),
-    title="Storyboard Assistant"
-) as demo:
-
-    gr.Markdown(
-        "<h1 style='text-align:center;'>📖 Storyboard Assistant</h1>"
-    )
+with gr.Blocks(title="Storyboard Assistant") as demo:
+    gr.Markdown("## 📖 Storyboard Assistant")
 
     with gr.Tabs():
-
-        # -------- Chat Tab --------
         with gr.Tab("💬 Chat"):
-            chatbot = gr.Chatbot(
-                label="Storyboard Chatbot",
-                type="messages"
-            )
-
+            chatbot = gr.Chatbot(type="messages")
             state = gr.State([])
 
-            user_input = gr.Textbox(
-                placeholder="Ask about storyboards, scenes, pacing...",
-                label="Your Message"
-            )
-
-            send_btn = gr.Button("Ask")
+            user_input = gr.Textbox(label="Your Message")
+            send = gr.Button("Ask")
 
-            send_btn.click(
-                fn=chat_with_bot,
+            send.click(
+                chat_with_bot,
                 inputs=[user_input, state],
                 outputs=[chatbot, state],
             )
 
-        # -------- Storyboard Tab --------
         with gr.Tab("📖 Generate Storyboard"):
-            scenario_input = gr.Textbox(
-                label="Scenario",
-                placeholder="A child discovers a hidden robot in the attic..."
-            )
+            scenario = gr.Textbox(label="Scenario")
+            generate = gr.Button("Generate")
+            output = gr.Textbox(lines=12)
 
-            generate_btn = gr.Button("Generate Storyboard")
-
-            storyboard_output = gr.Textbox(
-                label="Generated Storyboard",
-                lines=14
-            )
-
-            generate_btn.click(
-                fn=generate_storyboard,
-                inputs=scenario_input,
-                outputs=storyboard_output
-            )
+            generate.click(generate_storyboard, scenario, output)
 
 
 # -----------------------------
-# 5. Launch app
+# 🚨 CRITICAL LAUNCH FIX
 # -----------------------------
-if __name__ == "__main__":
-    demo.launch()
+demo.launch(
+    ssr=False,        # ✅ disables schema bug
+    show_api=False    # ✅ prevents /api/info crash
+)