Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from groq import Groq
 
 # -----------------------------
-# Initialize Groq
+# Groq Client
 # -----------------------------
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
@@ -15,24 +15,24 @@ SYSTEM_PROMPT = (
 # -----------------------------
 # Chat function
 # -----------------------------
-def chat_with_bot(user_input, history):
+def chat(user_message, history):
     if history is None:
         history = []
 
     if not history:
         history.append({"role": "system", "content": SYSTEM_PROMPT})
 
-    history.append({"role": "user", "content": user_input})
+    history.append({"role": "user", "content": user_message})
 
-    completion = client.chat.completions.create(
+    response = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
         messages=history,
-        max_tokens=512,
         temperature=1,
+        max_tokens=512,
     )
 
-    reply = completion.choices[0].message.content
-    history.append({"role": "assistant", "content": reply})
+    assistant_reply = response.choices[0].message.content
+    history.append({"role": "assistant", "content": assistant_reply})
 
     return history, history
 
@@ -44,7 +44,7 @@ def generate_storyboard(scenario):
     if not scenario.strip():
         return "Please enter a scenario."
 
-    completion = client.chat.completions.create(
+    response = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
         messages=[
             {"role": "system", "content": "Generate a 6-scene storyboard in table format."},
@@ -53,11 +53,11 @@ def generate_storyboard(scenario):
         max_tokens=700,
     )
 
-    return completion.choices[0].message.content
+    return response.choices[0].message.content
 
 
 # -----------------------------
-# UI
+# Gradio UI
 # -----------------------------
 with gr.Blocks(title="Storyboard Assistant") as demo:
     gr.Markdown("## 📖 Storyboard Assistant")
@@ -68,26 +68,28 @@ with gr.Blocks(title="Storyboard Assistant") as demo:
             state = gr.State([])
 
             user_input = gr.Textbox(label="Your Message")
-            send = gr.Button("Ask")
+            send_btn = gr.Button("Ask")
 
-            send.click(
-                chat_with_bot,
+            send_btn.click(
+                chat,
                 inputs=[user_input, state],
                 outputs=[chatbot, state],
             )
 
         with gr.Tab("📖 Generate Storyboard"):
             scenario = gr.Textbox(label="Scenario")
-            generate = gr.Button("Generate")
+            generate_btn = gr.Button("Generate")
             output = gr.Textbox(lines=12)
 
-            generate.click(generate_storyboard, scenario, output)
+            generate_btn.click(
+                generate_storyboard,
+                inputs=scenario,
+                outputs=output,
+            )
 
 
 # -----------------------------
-# 🚨 CRITICAL LAUNCH FIX
+# Launch (CRITICAL FIX)
 # -----------------------------
-demo.launch(
-    ssr=False,        # ✅ disables schema bug
-    show_api=False    # ✅ prevents /api/info crash
-)
+if __name__ == "__main__":
+    demo.launch()