Update app.py

app.py

@@ -103,13 +103,10 @@ def load_recommender(paths, start_page=1):
     recommender.fit(chunks)
     return 'Corpus Loaded.'
 
-def generate_text(prompt, engine='gpt-3.5-turbo', max_tokens=2048, temperature=0.8):
+def generate_text(messages, engine='gpt-3.5-turbo', max_tokens=2048, temperature=0.8):
     response = openai.ChatCompletion.create(
         model=engine,
-        messages=[
-            {"role": "system", "content": "You are a research assistant"},
-            {"role": "user", "content": prompt}
-        ],
+        messages=messages,
         max_tokens=max_tokens,
         temperature=temperature
     )
@@ -122,11 +119,11 @@ def generate_answer(question):
     messages = [
         {
             'role': 'system',
-            'content': 'You are a research assistant. 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".'
+            'content': 'You are a helpful assistant.'
         },
         {
             'role': 'user',
-            'content': 'Query: ' + question
+            'content': question
         },
     ]