Update app.py

app.py

@@ -6,31 +6,47 @@ from groq import Groq
 api_key = os.getenv("GROQ_API_KEY")
 client = Groq(api_key=api_key)
 
-# Function to generate a storyboard using LLM
-def generate_storyboard(scenario):
-    if not scenario.strip():
-        return "Please provide a scenario to generate the storyboard."
+# Initialize conversation history
+conversation_history = []
+
+def chat_with_bot_stream(user_input):
+    global conversation_history
+    conversation_history.append({"role": "user", "content": user_input})
     
-    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}"}
-    ]
+    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."
+        })
     
     completion = client.chat.completions.create(
         model="llama3-70b-8192",
-        messages=messages,
+        messages=conversation_history,
         temperature=1,
         max_tokens=1024,
         top_p=1,
-        stream=False,
+        stream=True,
         stop=None,
     )
-    return completion.choices[0].message.content
+    
+    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]
 
-# Interactive chatbot with streaming response
-def chatbot_stream(user_input, chat_history):
-    messages = chat_history + [
-        {"role": "user", "content": user_input}
+# Function to generate a storyboard
+def generate_storyboard(scenario):
+    if not scenario.strip():
+        return "Please provide a scenario to generate the storyboard."
+    
+    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}"}
     ]
     
     completion = client.chat.completions.create(
@@ -39,60 +55,48 @@ def chatbot_stream(user_input, chat_history):
         temperature=1,
         max_tokens=1024,
         top_p=1,
-        stream=True,
+        stream=False,
         stop=None,
     )
-    
-    response = ""
-    for chunk in completion:
-        text_chunk = chunk.choices[0].delta.content or ""
-        response += text_chunk
-        yield chat_history + [{"role": "assistant", "content": response}]
+    return completion.choices[0].message.content
 
-# Gradio UI with enhanced chat interface
-def ui():
-    with gr.Blocks(theme=gr.themes.Glass(primary_hue="violet", secondary_hue="emerald", neutral_hue="stone")) as app:
-        with gr.Tabs():
-            with gr.TabItem("💬Chat"):
-                gr.Markdown("# AI Storyboard & Chatbot")
-                chatbot = gr.Chatbot(label="Storyboard Chatbot", type="messages")
-                with gr.Row():
-                    chat_input = gr.Textbox(
-                        label="Your Message",
-                        placeholder="Type your question here...",
-                        lines=1
-                    )
-                    send_button = gr.Button("✋Ask Question")
-                
-                # Quick question examples
-                quick_questions = ["How do I structure a storyboard?", "What makes a good visual scene?", "How to add depth to a story?"]
-                with gr.Row():
-                    for question in quick_questions:
-                        gr.Button(question, variant="secondary").click(
-                            fn=lambda q, h=[]: chatbot_stream(q, h),
-                            inputs=[gr.State(), chatbot],
-                            outputs=chatbot
-                        )
-                
-                # Chatbot functionality
-                send_button.click(
-                    fn=chatbot_stream,
-                    inputs=[chat_input, chatbot],
-                    outputs=chatbot,
-                    queue=True
-                ).then(
-                    fn=lambda: "",  # Clear the input box after sending
-                    inputs=None,
-                    outputs=chat_input
-                )
+TITLE = """
+<style>
+h1 { text-align: center; font-size: 24px; margin-bottom: 10px; }
+</style>
+<h1>📖 Storyboard Assistant</h1>
+"""
 
-            with gr.TabItem("📖 Storyboard Generator"):
-                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)
-    
-    app.launch()
+with gr.Blocks(theme=gr.themes.Glass(primary_hue="violet", secondary_hue="emerald", neutral_hue="stone")) as demo:
+    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.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)
 
-ui()
+demo.launch()