Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from groq import Groq
 
 # -----------------------------
-# Groq Client
+# Initialize Groq
 # -----------------------------
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
@@ -13,26 +13,28 @@ SYSTEM_PROMPT = (
 )
 
 # -----------------------------
-# Chat function
+# Chat logic (NO message schema)
 # -----------------------------
-def chat(user_message, history):
+def chat(user_input, history):
     if history is None:
         history = []
 
-    if not history:
-        history.append({"role": "system", "content": SYSTEM_PROMPT})
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+    for u, a in history:
+        messages.append({"role": "user", "content": u})
+        messages.append({"role": "assistant", "content": a})
 
-    history.append({"role": "user", "content": user_message})
+    messages.append({"role": "user", "content": user_input})
 
-    response = client.chat.completions.create(
+    completion = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
-        messages=history,
-        temperature=1,
+        messages=messages,
         max_tokens=512,
+        temperature=1,
     )
 
-    assistant_reply = response.choices[0].message.content
-    history.append({"role": "assistant", "content": assistant_reply})
+    reply = completion.choices[0].message.content
+    history.append((user_input, reply))
 
     return history, history
 
@@ -44,7 +46,7 @@ def generate_storyboard(scenario):
     if not scenario.strip():
         return "Please enter a scenario."
 
-    response = client.chat.completions.create(
+    completion = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
         messages=[
             {"role": "system", "content": "Generate a 6-scene storyboard in table format."},
@@ -53,43 +55,39 @@ def generate_storyboard(scenario):
         max_tokens=700,
     )
 
-    return response.choices[0].message.content
+    return completion.choices[0].message.content
 
 
 # -----------------------------
-# Gradio UI
+# UI (NO Blocks, NO schema)
 # -----------------------------
-with gr.Blocks(title="Storyboard Assistant") as demo:
-    gr.Markdown("## 📖 Storyboard Assistant")
-
-    with gr.Tabs():
-        with gr.Tab("💬 Chat"):
-            chatbot = gr.Chatbot(type="messages")
-            state = gr.State([])
-
-            user_input = gr.Textbox(label="Your Message")
-            send_btn = gr.Button("Ask")
-
-            send_btn.click(
-                chat,
-                inputs=[user_input, state],
-                outputs=[chatbot, state],
-            )
-
-        with gr.Tab("📖 Generate Storyboard"):
-            scenario = gr.Textbox(label="Scenario")
-            generate_btn = gr.Button("Generate")
-            output = gr.Textbox(lines=12)
-
-            generate_btn.click(
-                generate_storyboard,
-                inputs=scenario,
-                outputs=output,
-            )
+chat_ui = gr.Interface(
+    fn=chat,
+    inputs=[
+        gr.Textbox(label="Your Message"),
+        gr.State([])
+    ],
+    outputs=[
+        gr.Chatbot(label="Storyboard Chat"),
+        gr.State([])
+    ],
+    title="📖 Storyboard Assistant",
+)
 
+storyboard_ui = gr.Interface(
+    fn=generate_storyboard,
+    inputs=gr.Textbox(label="Scenario"),
+    outputs=gr.Textbox(lines=12),
+    title="📖 Storyboard Generator",
+)
 
 # -----------------------------
-# Launch (CRITICAL FIX)
+# App
 # -----------------------------
+demo = gr.TabbedInterface(
+    [chat_ui, storyboard_ui],
+    ["💬 Chat", "📖 Storyboard"]
+)
+
 if __name__ == "__main__":
     demo.launch()