Update app.py

app.py

@@ -1,20 +1,28 @@
-import gradio as gr
 import os
+import gradio as gr
 from groq import Groq
 
-# Initialize Groq client
+# -----------------------------
+# 1. Initialize Groq client
+# -----------------------------
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
-# Conversation memory
-conversation_history = [
-    {
-        "role": "system",
-        "content": "You are an expert in storyboarding. Provide structured and insightful responses."
-    }
-]
+SYSTEM_PROMPT = (
+    "You are an expert in storyboarding. "
+    "Provide clear, structured, and creative responses."
+)
 
+# -----------------------------
+# 2. Chat function
+# -----------------------------
 def chat_with_bot(user_input, history):
-    history = history or conversation_history
+    if history is None:
+        history = []
+
+    # Add system message once
+    if not history:
+        history.append({"role": "system", "content": SYSTEM_PROMPT})
+
     history.append({"role": "user", "content": user_input})
 
     completion = client.chat.completions.create(
@@ -22,22 +30,30 @@ def chat_with_bot(user_input, history):
         messages=history,
         temperature=1,
         max_tokens=1024,
-        top_p=1,
-        stream=False,
     )
 
-    assistant_msg = completion.choices[0].message.content
-    history.append({"role": "assistant", "content": assistant_msg})
+    assistant_reply = completion.choices[0].message.content
+    history.append({"role": "assistant", "content": assistant_reply})
 
     return history, history
 
+
+# -----------------------------
+# 3. Storyboard generator
+# -----------------------------
 def generate_storyboard(scenario):
     if not scenario.strip():
-        return "Please provide a scenario."
+        return "❗ Please enter a scenario."
 
     messages = [
-        {"role": "system", "content": "Generate a 6-scene storyboard in table form."},
-        {"role": "user", "content": scenario}
+        {
+            "role": "system",
+            "content": "Generate a 6-scene storyboard in a table format."
+        },
+        {
+            "role": "user",
+            "content": scenario
+        }
     ]
 
     completion = client.chat.completions.create(
@@ -50,35 +66,65 @@ def generate_storyboard(scenario):
     return completion.choices[0].message.content
 
 
-TITLE = """
-<h1 style="text-align:center;">📖 Storyboard Assistant</h1>
-"""
+# -----------------------------
+# 4. Gradio UI
+# -----------------------------
+with gr.Blocks(
+    theme=gr.themes.Glass(),
+    title="Storyboard Assistant"
+) as demo:
 
-with gr.Blocks(theme=gr.themes.Glass()) as demo:
-    gr.HTML(TITLE)
+    gr.Markdown(
+        "<h1 style='text-align:center;'>📖 Storyboard Assistant</h1>"
+    )
 
     with gr.Tabs():
+
+        # -------- Chat Tab --------
         with gr.Tab("💬 Chat"):
-            chatbot = gr.Chatbot(type="messages", label="Storyboard Chatbot")
+            chatbot = gr.Chatbot(
+                label="Storyboard Chatbot",
+                type="messages"
+            )
+
             state = gr.State([])
 
             user_input = gr.Textbox(
-                placeholder="Ask about storyboards...",
+                placeholder="Ask about storyboards, scenes, pacing...",
                 label="Your Message"
             )
-            send = gr.Button("Ask")
 
-            send.click(
-                chat_with_bot,
+            send_btn = gr.Button("Ask")
+
+            send_btn.click(
+                fn=chat_with_bot,
                 inputs=[user_input, state],
-                outputs=[chatbot, state]
+                outputs=[chatbot, state],
             )
 
+        # -------- Storyboard Tab --------
         with gr.Tab("📖 Generate Storyboard"):
-            scenario = gr.Textbox(label="Scenario")
-            generate = gr.Button("Generate")
-            output = gr.Textbox(label="Storyboard", lines=12)
+            scenario_input = gr.Textbox(
+                label="Scenario",
+                placeholder="A child discovers a hidden robot in the attic..."
+            )
+
+            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)
 
-demo.launch()
+# -----------------------------
+# 5. Launch app
+# -----------------------------
+if __name__ == "__main__":
+    demo.launch()