Update app.py

app.py

@@ -103,31 +103,38 @@ def load_recommender(paths, start_page=1):
     recommender.fit(chunks)
     return 'Corpus Loaded.'
 
-def generate_text(prompt, engine="gpt-3.5-turbo"):
-    completions = openai.Completion.create(
-        engine=engine,
-        prompt=prompt,
+def generate_text(messages, engine="gpt-3.5-turbo"):
+    response = openai.ChatCompletion.create(
+        model=engine,
+        messages=messages,
         max_tokens=512,
-        n=1,
-        stop=None,
         temperature=0.7,
     )
-    message = completions.choices[0].text
+    message = response['choices'][0]['message']['content']
     return message
 
 def generate_answer(question):
     topn_chunks = recommender(question)
-    prompt = ""
-    prompt += 'search results:\n\n'
+    
+    system_message = {
+        'role': 'system',
+        'content': 'Compose a comprehensive reply to the query using the search results given. Only include information found in the results. If the text does not relate to the query, simply state "Text Not Found in Body of Knowledge".'
+    }
+    user_message = {
+        'role': 'user',
+        'content': 'Query: ' + question
+    }
+    
     for c in topn_chunks:
-        prompt += c + '\n\n'
-
-    prompt += "Compose a comprehensive reply to the query using the search results given. "\
-              "Only include information found in the results. "\
-              "If the text does not relate to the query, simply state 'Text Not Found in Body of Knowledge'. "\
-              "\n\nQuery: {question}\n Answer: "
-
-    answer = generate_text(prompt, "gpt-3.5-turbo")
+        search_result_message = {
+            'role': 'assistant',
+            'content': c
+        }
+        system_message['content'] += '\n\n' + search_result_message['content']
+    
+    messages = [system_message, user_message]
+    
+    answer = generate_text(messages, "gpt-3.5-turbo")
     return answer
 
 def question_answer(urls, file, question):