Update app.py

app.py

@@ -3,100 +3,82 @@ import os
 from groq import Groq
 
 # Initialize Groq client
-api_key = os.getenv("GROQ_API_KEY")
-client = Groq(api_key=api_key)
-
-# Initialize conversation history
-conversation_history = []
-
-def chat_with_bot_stream(user_input):
-    global conversation_history
-    conversation_history.append({"role": "user", "content": user_input})
-    
-    if len(conversation_history) == 1:
-        conversation_history.insert(0, {
-            "role": "system",
-            "content": "You are an expert in storyboarding. Provide structured and insightful responses to queries about creating and refining storyboards."
-        })
-    
+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."
+    }
+]
+
+def chat_with_bot(user_input, history):
+    history = history or conversation_history
+    history.append({"role": "user", "content": user_input})
+
     completion = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
-        messages=conversation_history,
+        messages=history,
         temperature=1,
         max_tokens=1024,
         top_p=1,
-        stream=True,
-        stop=None,
+        stream=False,
     )
-    
-    response_content = ""
-    for chunk in completion:
-        response_content += chunk.choices[0].delta.content or ""
-    
-    conversation_history.append({"role": "assistant", "content": response_content})
-    
-    return [(msg["content"] if msg["role"] == "user" else None, 
-             msg["content"] if msg["role"] == "assistant" else None) 
-            for msg in conversation_history]
-
-# Function to generate a storyboard
+
+    assistant_msg = completion.choices[0].message.content
+    history.append({"role": "assistant", "content": assistant_msg})
+
+    return history, history
+
 def generate_storyboard(scenario):
     if not scenario.strip():
-        return "Please provide a scenario to generate the storyboard."
-    
+        return "Please provide a scenario."
+
     messages = [
-        {"role": "system", "content": "You are an AI storyteller. Generate a storyboard in a structured table with six scenes."},
-        {"role": "user", "content": f"Generate a 6-scene storyboard for: {scenario}"}
+        {"role": "system", "content": "Generate a 6-scene storyboard in table form."},
+        {"role": "user", "content": scenario}
     ]
-    
+
     completion = client.chat.completions.create(
         model="llama-3.3-70b-versatile",
         messages=messages,
         temperature=1,
         max_tokens=1024,
-        top_p=1,
-        stream=False,
-        stop=None,
     )
+
     return completion.choices[0].message.content
 
+
 TITLE = """
-<style>
-h1 { text-align: center; font-size: 24px; margin-bottom: 10px; }
-</style>
-<h1>📖 Storyboard Assistant</h1>
+<h1 style="text-align:center;">📖 Storyboard Assistant</h1>
 """
 
-with gr.Blocks(theme=gr.themes.Glass(primary_hue="violet", secondary_hue="emerald", neutral_hue="stone")) as demo:
+with gr.Blocks(theme=gr.themes.Glass()) as demo:
+    gr.HTML(TITLE)
+
     with gr.Tabs():
-        with gr.TabItem("💬Chat"):
-            gr.HTML(TITLE)
-            chatbot = gr.Chatbot(label="Storyboard Chatbot")
-            with gr.Row():
-                user_input = gr.Textbox(
-                    label="Your Message",
-                    placeholder="Type your question here...",
-                    lines=1
-                )
-                send_button = gr.Button("✋Ask Question")
-            
-            # Chatbot functionality
-            send_button.click(
-                fn=chat_with_bot_stream,
-                inputs=user_input,
-                outputs=chatbot,
-                queue=True
-            ).then(
-                fn=lambda: "",
-                inputs=None,
-                outputs=user_input
+        with gr.Tab("💬 Chat"):
+            chatbot = gr.Chatbot(type="messages", label="Storyboard Chatbot")
+            state = gr.State([])
+
+            user_input = gr.Textbox(
+                placeholder="Ask about storyboards...",
+                label="Your Message"
             )
-        
-        with gr.TabItem("📖 Generate Storyboard"):
-            gr.Markdown("## Generate a Storyboard")
-            scenario_input = gr.Textbox(label="Enter your scenario")
-            generate_btn = gr.Button("Generate Storyboard")
-            storyboard_output = gr.Textbox(label="Generated Storyboard", interactive=False)
-            generate_btn.click(generate_storyboard, inputs=scenario_input, outputs=storyboard_output)
+            send = gr.Button("Ask")
+
+            send.click(
+                chat_with_bot,
+                inputs=[user_input, state],
+                outputs=[chatbot, state]
+            )
+
+        with gr.Tab("📖 Generate Storyboard"):
+            scenario = gr.Textbox(label="Scenario")
+            generate = gr.Button("Generate")
+            output = gr.Textbox(label="Storyboard", lines=12)
+
+            generate.click(generate_storyboard, scenario, output)
 
 demo.launch()