Update app.py

app.py

@@ -71,16 +71,26 @@ def get_index(index_file_path):
         print(f"Error: '{index_file_path}' does not exist.")
         sys.exit()
 
-# passes the prompt to the chatbot
-def chatbot(input_text, mentioned_person='Mediator John Haynes', confidence_threshold=0.5):
-    index = get_index(INDEX_FILE)
-    prompt = f"You are {mentioned_person}. Answer this: {input_text}. Only reply from the contextual data, or say you don't know. At the end of your answer ask an insightful question."
-    response = index.query(prompt, response_mode="compact", temperature=0.8)
+openai = OpenAI()
+index = get_index(INDEX_FILE)
 
-    store_message(input_text,response)
+# passes the prompt to the chatbot
+#def chatbot(input_text, mentioned_person='Mediator John Haynes', confidence_threshold=0.5):
+#    
+#    prompt = f"You are {mentioned_person}. Answer this: {input_text}. Only reply from the contextual data, or say you don't know. At the end of your answer ask an insightful question."
+#    response = index.query(prompt, response_mode="compact")
+#
+#    store_message(input_text,response)
     
     # return the response
-    return response.response
+#    return response.response
+
+def chatbot(input_text, mentioned_person='Mediator John Haynes', temperature=0.8):
+    prompt = f"You are {mentioned_person}. Answer this: {input_text}. Only reply from the contextual data, or say you don't know. At the end of your answer ask an insightful question."
+    context = index.get_context(prompt)
+    response = openai.complete(prompt, context=context, temperature=temperature)['choices'][0]['text']
+    store_message(input_text,response)
+    return response
 
 with open('about.txt', 'r') as file:
     about = file.read()