Update app.py

app.py

@@ -9,6 +9,11 @@ from chromadb.utils.embedding_functions import OpenAIEmbeddingFunction
 client = OpenAI(api_key=os.getenv("OPENAI_KEY"))
 pp = pprint.PrettyPrinter(indent=4)
 
+current_id = 0
+chat_history = []
+chat_metadata = []
+history_ids = []
+
 def generate_response(messages):
     model_name = os.getenv("MODEL_NAME")
     response = client.chat.completions.create(model=model_name, messages=messages, temperature=0.5, max_tokens=250)
@@ -18,10 +23,11 @@ def generate_response(messages):
     return response.choices[0].message
 
 def chat_interface(user_input):
+    global current_id
     chroma_client = chromadb.Client()
     embedding_function = OpenAIEmbeddingFunction(api_key=os.getenv("OPENAI_KEY"), model_name=os.getenv("EMBEDDING_MODEL"))
     collection = chroma_client.create_collection(name="conversations", embedding_function=embedding_function)
-
+    
     messages = [{"role": "system", "content": "You are a kind and friendly chatbot"}]
     results = collection.query(query_texts=[user_input], n_results=2)
     for res in results['documents'][0]:
@@ -29,6 +35,19 @@ def chat_interface(user_input):
     messages.append({"role": "user", "content": user_input})
 
     response = generate_response(messages)
+    chat_metadata.append({"role":"user"})
+    chat_history.append(input_text)
+    chat_metadata.append({"role":"assistant"})
+    chat_history.append(response.content)
+    current_id += 1
+    history_ids.append(f"id_{current_id}")
+    current_id += 1
+    history_ids.append(f"id_{current_id}")
+    collection.add(
+        documents=chat_history,
+        metadatas=chat_metadata,
+        ids=history_ids
+    )
     return response
 
 def main():