Update app.py

app.py

@@ -134,7 +134,7 @@ def respond(
 ):
     new_query = ra(message)
     print("old: ",new_query)
-    new_query = str(new_query.split("Output: ", 1)[-1] if "Output: " in new_query else new_query)
+    #new_query = str(new_query.split("Output: ", 1)[-1] if "Output: " in new_query else new_query)
     if new_query[-1] == "(":
         new_query = new_query[:-1] 
     print("new: ",new_query)
@@ -164,8 +164,8 @@ Instructions:
             messages.append({"role": "user", "content": val[0]})
         if val[1]:
             messages.append({"role": "assistant", "content": val[1]})
-    messages.append({"role": "user", "content": "Context: "+str(retriever(message))+"\n\nQuestion: "+message +"\n\nAnswer: "})
-    print("Context: "+str(retriever(message))+"\n\nQuestion: "+message +"\n\nAnswer: ")
+    messages.append({"role": "user", "content": "Context: "+str(retriever(new_query))+"\n\nQuestion: "+new_query +"\n\nAnswer: "})
+    print("Context: "+str(retriever(new_query))+"\n\nQuestion: "+new_query +"\n\nAnswer: ")
     response = ""
 
     for message in client.chat_completion(