Update app.py

app.py

@@ -104,38 +104,31 @@ def load_recommender(paths, start_page=1):
     return 'Corpus Loaded.'
 
 def generate_text(messages, engine='gpt-3.5-turbo', max_tokens=2048, temperature=0.8):
-    response = openai.Completion.create(
+   response = openai.ChatCompletion.create(
         model=engine,
-        messages=messages,
+        messages=[{"role": "system", "content": "You are a research assistant"},
+             {"role": "user", "content": prompt}],
         max_tokens=max_tokens,
+        n=1,
         temperature=temperature
     )
-    return response.choices[0].message['content']
+        return response.choices[0].message['content']
 
 
 def generate_answer(question):
     topn_chunks = recommender(question)
-    
-    messages = [
-        {
-            'role': 'system',
-            'content': 'You are a helpful assistant.'
-        },
-        {
-            'role': 'user',
-            'content': question
-        },
-    ]
-    
+
+    prompt = "You are a helpful assistant.\n"
+    prompt += "User: " + question + "\n"
+
     for c in topn_chunks:
-        messages.append({
-            'role': 'assistant',
-            'content': c
-        })
-    
-    answer = generate_text(messages)
+        prompt += "Assistant: " + c + "\n"
+
+    answer = generate_text(prompt)
     return answer
 
+
+
 def question_answer(urls, file, question):
     if urls.strip() == '' and file is None:
         return '[ERROR]: Both URLs and PDFs are empty. Provide at least one.'