Update app.py

app.py

@@ -104,12 +104,20 @@ def roleChoice(role):
 
 
 
-def textGPT(text):
+def talk2file(index_name, text):
     global messages
+    
+    #same as filesearch
+    init_pinecone()
+    embeddings = OpenAIEmbeddings(openai_api_key=os.environ['OPENAI_API_KEY'])
+    docsearch = Pinecone.from_existing_index(index_name, embeddings)
+    docs = docsearch.similarity_search(prompt)
 
-    messages.append({"role": "user", "content": text})
+    
+    prompt = text + ", based on the following text: \n\n" + docs[0].page_content 
+    messages.append({"role": "user", "content": prompt})
 
-    response = openai.ChatCompletion.create(model="gpt-4", messages=messages)
+    response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
 
     system_message = response["choices"][0]["message"]
     messages.append(system_message)
@@ -122,6 +130,9 @@ def textGPT(text):
     return chats
 
 
+
+
+
 def fileSearch(index_name, prompt):
     global messages