Update app.py

app.py

@@ -3,84 +3,72 @@ import gradio as gr
 from groq import Groq
 
 # -------------------------------
-# Initialize Groq Client
+# Groq Client (HF-safe)
 # -------------------------------
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
 # -------------------------------
-# Conversation Memory
+# In-memory chat history
 # -------------------------------
-conversation_history = [
-    {
-        "role": "system",
-        "content": (
-            "You are an expert in storyboarding. "
-            "Provide structured, visual, and insightful guidance "
-            "for creating and refining storyboards."
-        )
-    }
-]
+chat_history = []
 
 # -------------------------------
-# Chatbot (Streaming)
+# Chat function (NON-streaming)
 # -------------------------------
-def chat_with_bot(user_input):
-    global conversation_history
+def chat_with_bot(user_input, history):
+    messages = [
+        {
+            "role": "system",
+            "content": (
+                "You are an expert in storyboarding. "
+                "Provide clear, structured, and visual guidance."
+            )
+        }
+    ]
 
-    conversation_history.append(
-        {"role": "user", "content": user_input}
-    )
+    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_input})
 
     completion = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
-        messages=conversation_history,
+        messages=messages,
         temperature=1,
-        max_completion_tokens=1024,
+        max_completion_tokens=512,
         top_p=1,
-        stream=True,
+        stream=False,
     )
 
-    response = ""
-    for chunk in completion:
-        delta = chunk.choices[0].delta.content
-        if delta:
-            response += delta
-
-    conversation_history.append(
-        {"role": "assistant", "content": response}
-    )
-
-    return [
-        (
-            msg["content"] if msg["role"] == "user" else None,
-            msg["content"] if msg["role"] == "assistant" else None
-        )
-        for msg in conversation_history
-        if msg["role"] != "system"
-    ]
+    reply = completion.choices[0].message.content
+    history.append((user_input, reply))
+    return history, ""
 
 # -------------------------------
 # Storyboard Generator
 # -------------------------------
 def generate_storyboard(scenario):
     if not scenario.strip():
-        return "Please provide a scenario."
+        return "Please enter a scenario."
 
     messages = [
         {
             "role": "system",
             "content": (
-                "You are an AI storyteller. Generate a storyboard "
-                "as a table with exactly six scenes.\n\n"
-                "Each scene must include:\n"
-                "1) Scenario: the user problem and product/feature used\n"
-                "2) Storyline: visual description and purpose\n\n"
-                "Output ONLY a markdown table."
+                "Generate a storyboard as a markdown table with EXACTLY six scenes.\n\n"
+                "Columns:\n"
+                "- Scene\n"
+                "- Scenario (problem + product/feature)\n"
+                "- Storyline (visual description + purpose)\n\n"
+                "Return ONLY the table."
             )
         },
         {
             "role": "user",
-            "content": f"Generate a 6-scene storyboard for: {scenario}"
+            "content": scenario
         }
     ]
 
@@ -88,7 +76,7 @@ def generate_storyboard(scenario):
         model="llama-3.3-70b-versatile",
         messages=messages,
         temperature=1,
-        max_completion_tokens=1024,
+        max_completion_tokens=700,
         top_p=1,
         stream=False,
     )
@@ -98,37 +86,38 @@ def generate_storyboard(scenario):
 # -------------------------------
 # UI
 # -------------------------------
-with gr.Blocks(theme=gr.themes.Glass()) as demo:
+with gr.Blocks() as demo:
     gr.Markdown("# 📖 Storyboard Assistant")
 
     with gr.Tabs():
+
+        # ---- Chat Tab ----
         with gr.TabItem("💬 Chat"):
             chatbot = gr.Chatbot(height=400)
-            with gr.Row():
-                user_input = gr.Textbox(
-                    placeholder="Ask about storyboarding...",
-                    scale=4
-                )
-                send_btn = gr.Button("Ask")
+            user_input = gr.Textbox(
+                placeholder="Ask a storyboarding question..."
+            )
+            send_btn = gr.Button("Send")
 
             send_btn.click(
-                fn=chat_with_bot,
-                inputs=user_input,
-                outputs=chatbot,
-            ).then(lambda: "", outputs=user_input)
+                chat_with_bot,
+                inputs=[user_input, chatbot],
+                outputs=[chatbot, user_input]
+            )
 
+        # ---- Storyboard Tab ----
         with gr.TabItem("📖 Generate Storyboard"):
             scenario_input = gr.Textbox(
                 label="Scenario",
-                placeholder="e.g., A student using an AI planner app"
+                placeholder="A student using an AI planner to manage exams"
             )
-            generate_btn = gr.Button("Generate Storyboard")
-            storyboard_output = gr.Markdown()
+            generate_btn = gr.Button("Generate")
+            output = gr.Markdown()
 
             generate_btn.click(
                 generate_storyboard,
                 inputs=scenario_input,
-                outputs=storyboard_output
+                outputs=output
             )
 
 demo.launch()