Update app.py

app.py

@@ -2,55 +2,48 @@ import os
 import gradio as gr
 from groq import Groq
 
-# -------------------------------
-# Groq Client (HF-safe)
-# -------------------------------
+# -------------------------------------------------
+# Groq Client
+# -------------------------------------------------
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
-# -------------------------------
-# In-memory chat history
-# -------------------------------
-chat_history = []
-
-# -------------------------------
-# Chat function (NON-streaming)
-# -------------------------------
-def chat_with_bot(user_input, history):
+# -------------------------------------------------
+# Chat Function (Groq mode, non-streaming)
+# -------------------------------------------------
+def chat_fn(message, history):
     messages = [
         {
             "role": "system",
             "content": (
                 "You are an expert in storyboarding. "
-                "Provide clear, structured, and visual guidance."
+                "Give clear, structured, and visual guidance."
             )
         }
     ]
 
+    # Rebuild conversation from Gradio history
     for user, assistant in history:
-        if user:
-            messages.append({"role": "user", "content": user})
-        if assistant:
-            messages.append({"role": "assistant", "content": assistant})
+        messages.append({"role": "user", "content": user})
+        messages.append({"role": "assistant", "content": assistant})
 
-    messages.append({"role": "user", "content": user_input})
+    messages.append({"role": "user", "content": message})
 
-    completion = client.chat.completions.create(
+    response = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
         messages=messages,
         temperature=1,
         max_completion_tokens=512,
         top_p=1,
-        stream=False,
     )
 
-    reply = completion.choices[0].message.content
-    history.append((user_input, reply))
-    return history, ""
+    assistant_reply = response.choices[0].message.content
+    history.append((message, assistant_reply))
+    return history
 
-# -------------------------------
+# -------------------------------------------------
 # Storyboard Generator
-# -------------------------------
-def generate_storyboard(scenario):
+# -------------------------------------------------
+def storyboard_fn(scenario):
     if not scenario.strip():
         return "Please enter a scenario."
 
@@ -58,11 +51,9 @@ def generate_storyboard(scenario):
         {
             "role": "system",
             "content": (
-                "Generate a storyboard as a markdown table with EXACTLY six scenes.\n\n"
+                "Create a storyboard as a MARKDOWN TABLE with EXACTLY 6 scenes.\n\n"
                 "Columns:\n"
-                "- Scene\n"
-                "- Scenario (problem + product/feature)\n"
-                "- Storyline (visual description + purpose)\n\n"
+                "Scene | Scenario (problem + product/feature) | Storyline (visual + purpose)\n\n"
                 "Return ONLY the table."
             )
         },
@@ -72,51 +63,48 @@ def generate_storyboard(scenario):
         }
     ]
 
-    completion = client.chat.completions.create(
+    response = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
         messages=messages,
         temperature=1,
         max_completion_tokens=700,
         top_p=1,
-        stream=False,
     )
 
-    return completion.choices[0].message.content
+    return response.choices[0].message.content
 
-# -------------------------------
+# -------------------------------------------------
 # UI
-# -------------------------------
+# -------------------------------------------------
 with gr.Blocks() as demo:
-    gr.Markdown("# 📖 Storyboard Assistant")
+    gr.Markdown("# 📖 Storyboard Assistant (Groq)")
 
     with gr.Tabs():
 
-        # ---- Chat Tab ----
         with gr.TabItem("💬 Chat"):
             chatbot = gr.Chatbot(height=400)
             user_input = gr.Textbox(
-                placeholder="Ask a storyboarding question..."
+                placeholder="Ask about storyboarding..."
             )
-            send_btn = gr.Button("Send")
+            send = gr.Button("Send")
 
-            send_btn.click(
-                chat_with_bot,
+            send.click(
+                chat_fn,
                 inputs=[user_input, chatbot],
-                outputs=[chatbot, user_input]
+                outputs=chatbot
             )
 
-        # ---- Storyboard Tab ----
         with gr.TabItem("📖 Generate Storyboard"):
-            scenario_input = gr.Textbox(
+            scenario = gr.Textbox(
                 label="Scenario",
                 placeholder="A student using an AI planner to manage exams"
             )
-            generate_btn = gr.Button("Generate")
+            generate = gr.Button("Generate")
             output = gr.Markdown()
 
-            generate_btn.click(
-                generate_storyboard,
-                inputs=scenario_input,
+            generate.click(
+                storyboard_fn,
+                inputs=scenario,
                 outputs=output
             )