Update app.py

app.py

@@ -118,15 +118,15 @@ def generate_answer(question,openAI_key):
         prompt += c + '\n\n'
 
     prompt += "Instructions: Compose a comprehensive reply to the query using the search results given. "\
-              "Cite each reference using [ Page Number] notation (every result has this number at the beginning). "\
+              "Cite each reference using [Page Number] notation (every result has this number at the beginning). "\
               "Citation should be done at the end of each sentence. If the search results mention multiple subjects "\
               "with the same name, create separate answers for each. Only include information found in the results and "\
-              "don't add any additional information. Make sure the answer is correct and don't output false content. "\
+              "don't add any additional information. Make sure the answer is correct, and don't output false content. "\
               "If the text does not relate to the query, simply state 'Found Nothing'. Ignore outlier "\
-              "search results which has nothing to do with the question. Only answer what is asked. The "\
+              "search results that have nothing to do with the question. Only answer what is asked. The "\
               "answer should be short and concise. \n\nQuery: {question}\nAnswer: "
 
-    prompt += f"Query: {question}\nAnswer:"
+    prompt += f"Query: {question}\nAnswer: "
     answer = generate_text(openAI_key, prompt,"text-davinci-003")
     return answer
 
@@ -264,9 +264,15 @@ if generate_all:
     for worker in workers:
         worker.join()
 
+    # Collect all results first
+    answers = {}
     while not results.empty():
         i, answer = results.get()
         if isinstance(answer, str) and 'Error' in answer:
             st.error(f"Error on row {i}: {answer}")
         else:
-            st.session_state[f'session_answer{i}'] = answer
+            answers[i] = answer
+
+    # Update session state in the main thread
+    for i, answer in answers.items():
+        st.session_state[f'session_answer{i}'] = answer