Update app.py

app.py

@@ -28,9 +28,8 @@ def generate_storyboard(scenario):
     return completion.choices[0].message.content
 
 # Interactive chatbot with streaming response
-def chatbot_stream(user_input):
-    messages = [
-        {"role": "system", "content": "You are an expert in storyboarding. Answer questions interactively."},
+def chatbot_stream(user_input, chat_history):
+    messages = chat_history + [
         {"role": "user", "content": user_input}
     ]
     
@@ -48,7 +47,7 @@ def chatbot_stream(user_input):
     for chunk in completion:
         text_chunk = chunk.choices[0].delta.content or ""
         response += text_chunk
-        yield {"role": "assistant", "content": response}
+        yield chat_history + [{"role": "assistant", "content": response}]
 
 # Gradio UI with enhanced chat interface
 def ui():
@@ -69,12 +68,16 @@ def ui():
                 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(lambda q=question: chatbot_stream(q), inputs=None, outputs=chatbot)
+                        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,
+                    inputs=[chat_input, chatbot],
                     outputs=chatbot,
                     queue=True
                 ).then(