Update app.py

app.py

@@ -132,27 +132,22 @@ def generate_text(openAI_key, prompt, model="gpt-3.5-turbo"):
         ).choices[0].message['content']
     return message
 
-
-def generate_answer(question, openAI_key,model):
+  
+def generate_answer(question, openAI_key, model):
     topn_chunks = recommender(question)
-    prompt = ""
-    prompt += 'search results:\n\n'
+    prompt = 'search results:\n\n'
     for c in topn_chunks:
         prompt += c + '\n\n'
         
     prompt += "Instructions: Compose a comprehensive reply to the query using the search results given. "\
-              "Cite each reference using [ Page Number] notation (every result has this number at the beginning). "\
-              "Citation should be done at the end of each sentence. If the search results mention multiple subjects "\
-              "with the same name, create separate answers for each. Only include information found in the results and "\
-              "don't add any additional information. Make sure the answer is correct and don't output false content. "\
-              "If the text does not relate to the query, simply state 'Found Nothing.Please select text-davinci-003 model.'. Ignore outlier "\
-              "search results which has nothing to do with the question. Only answer what is asked. The "\
-              "answer should be short and concise. \n\nQuery: {question}\nAnswer: "
+              "Cite each reference using [ Page Number] notation. "\
+              "Only answer what is asked. The answer should be short and concise. \n\nQuery: "
     
-    prompt += f"Query: {question}\nAnswer:"
+    prompt += f"{question}\nAnswer:"
     answer = generate_text(openAI_key, prompt, model)
     return answer
 
+
 def question_answer(chat_history, url, file, question, openAI_key, model):
     try:
         if openAI_key.strip()=='':
@@ -175,7 +170,7 @@ def question_answer(chat_history, url, file, question, openAI_key, model):
             load_recommender(file_name)
         if question.strip() == '':
             return '[ERROR]: Question field is empty'
-        if model == "text-davinci-003":
+        if model == "text-davinci-003" or model == "gpt-4" or model == "gpt-4-32k":
             answer = generate_answer_text_davinci_003(question, openAI_key)
         else:
             answer = generate_answer(question, openAI_key, model)