Update app.py

app.py

@@ -118,20 +118,35 @@ def load_recommender(path, start_page=1):
 
 
 
-def generate_text(openAI_key,prompt, engine="text-davinci-003"):
+#def generate_text(openAI_key,prompt, engine="text-davinci-003"):
+ #   openai.api_key = openAI_key
+  #  completions = openai.Completion.create(
+   #     engine=engine,
+    #    prompt=prompt,
+     #   max_tokens=512,
+      #  n=1,
+       # stop=None,
+        #temperature=0.7,
+    #)
+    #message = completions.choices[0].text
+    #return message
+    
+def generate_text(openAI_key, prompt, engine="gpt-3.5-turbo"):
     openai.api_key = openAI_key
-    completions = openai.Completion.create(
-        engine=engine,
-        prompt=prompt,
+    messages = [{'role': 'system', 'content': 'You are a helpful assistant.'},
+                {'role': 'user', 'content': prompt}]
+    
+    completions = openai.ChatCompletion.create(
+        model=engine,
+        messages=messages,
         max_tokens=512,
         n=1,
         stop=None,
         temperature=0.7,
     )
-    message = completions.choices[0].text
+    message = completions.choices[0].message['content']
     return message
 
-
 def generate_answer(question,openAI_key):
     topn_chunks = recommender(question)
     prompt = ""