Update app.py

app.py

@@ -13,6 +13,11 @@ current_id = 0
 chat_history = []
 chat_metadata = []
 history_ids = []
+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"}]
 
 def generate_response(messages):
     model_name = os.getenv("MODEL_NAME")
@@ -24,11 +29,7 @@ def generate_response(messages):
 
 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]:
         messages.append({"role": "user", "content": f"previous chat: {res}"})