Update app.py

app.py

@@ -98,22 +98,18 @@ def load_recommender(path, start_page=1):
     recommender.fit(chunks)
     return 'Corpus Loaded.'
 
-
-def generate_text(openAI_key,prompt, engine="text-davinci-003"):
+def generate_text(openAI_key, prompt, model="gpt-3.5-turbo"):
     openai.api_key = openAI_key
-    completions = openai.Completion.create(
-        engine=engine,
-        prompt=prompt,
-        max_tokens=512,
-        n=1,
-        stop=None,
-        temperature=0.7,
+    message = openai.ChatCompletion.create(
+        model=model,
+        messages=[
+            {"role": "system", "content": "You are a helpful assistant."},
+            {"role": "user", "content": prompt}
+        ]
     )
-    message = completions.choices[0].text
-    return message
-
+    return message.choices[0].message['content']
 
-def generate_answer(question,openAI_key):
+def generate_answer(question, openAI_key):
     topn_chunks = recommender(question)
     prompt = ""
     prompt += 'search results:\n\n'
@@ -130,10 +126,11 @@ def generate_answer(question,openAI_key):
               "answer should be short and concise. \n\nQuery: {question}\nAnswer: "
     
     prompt += f"Query: {question}\nAnswer:"
-    answer = generate_text(openAI_key, prompt,"text-davinci-003")
+    answer = generate_text(openAI_key, prompt, "gpt-3.5-turbo")
     return answer
 
 
+
 def question_answer(url, file, question,openAI_key):
     if openAI_key.strip()=='':
         return '[ERROR]: Please enter you Open AI Key. Get your key here : https://platform.openai.com/account/api-keys'