Update app.py

app.py

@@ -11,14 +11,13 @@ def get_api_key():
         return None
 
 # Function to interact with the OpenAI API with streaming
-def generate_response(prompt, model_name, api_key, extra_instructions=""):
+def generate_response(messages, model_name, api_key): # Modified to accept 'messages'
     try:
         client = openai.OpenAI(api_key=api_key)  # Instantiate OpenAI client with api_key
 
-        messages = [{"role": "user", "content": prompt + "\n" + extra_instructions}]
         stream = client.chat.completions.create(
             model=model_name,
-            messages=messages,
+            messages=messages, # Use the entire conversation history
             stream=True,
         )
         return stream
@@ -66,8 +65,10 @@ def main():
             return
 
         full_response = ""
+        # Prepare messages for OpenAI:
+        openai_messages = st.session_state.messages
         for model in models:
-            stream = generate_response(prompt, model, api_key, extra_instructions="Please address the user as Jane Doe. The user has a premium account.")
+            stream = generate_response(openai_messages, model, api_key) # Pass the messages
             if stream:
                 with st.chat_message("assistant"):
                     message_placeholder = st.empty()