Update app.py

app.py

@@ -111,10 +111,15 @@ def text_lookup(data, sentence_ids):
     return context
 
 
-def gpt3_summary(text):
+def gpt3_summary(context_text, query):
+    prompt = f"""Context information is below. \n"
+    "---------------------\n {context_text}"
+    "\n---------------------\n"
+    "Given the context information and  prior knowledge, "
+    "answer the question: {query}\n" """
     response = openai.Completion.create(
-        model="text-davinci-003",
-        prompt=text + "\n\nTl;dr",
+        model="text-ada-001",
+        prompt=  prompt,
         temperature=0.1,
         max_tokens=512,
         top_p=1.0,
@@ -237,11 +242,11 @@ if decoder_model == "GPT3 (summary_davinci)":
     )
     api_key = save_key(openai_key)
     openai.api_key = api_key
-    output_text = []
-    for context_text in context_list:
-        output_text.append(gpt3_summary(context_text))
-    generated_text = ". ".join(output_text)
-    st.write(gpt3_summary(generated_text))
+    #output_text = []
+    #for context_text in context_list:
+    #    output_text.append(gpt3_summary(context_text))
+    #generated_text = ". ".join(output_text)
+    st.write(gpt3_summary(context_text,query_text))
 
 elif decoder_model == "GPT3 (QA_davinci)":
     openai_key = st.text_input(