Update app.py

app.py

@@ -88,7 +88,7 @@ def retrieve(query):
             break
         elif i == len(contexts) - 1:
             prompt = prompt_start + "\n\n---\n\n".join(contexts) + prompt_end
-    return prompt, contexts
+    return prompt
 
 
 # first let's make it simpler to get answers
@@ -109,8 +109,8 @@ def complete(prompt):
 
 def query(question, top_k_reader, top_k_retriever):
     # first we retrieve relevant items from Pinecone
-    query_with_contexts, contexts = retrieve(question)
-    return complete(query_with_contexts), contexts
+    query_with_contexts = retrieve(question)
+    return complete(query_with_contexts)
 
 
 indexing_pipeline_with_classification = Pipeline()
@@ -312,7 +312,7 @@ if st.session_state.results:
 
     st.write("## Results:")
 
-    for (result, contexts) in st.session_state.results:
+    for result in st.session_state.results:
         # answer, context = result.answer, result.context
         # start_idx = context.find(answer)
         # end_idx = start_idx + len(answer)
@@ -323,9 +323,9 @@ if st.session_state.results:
             # markdown(f'**Source:** {source} \n {context[:start_idx] } {str(annotation(answer, "ANSWER", "#8ef"))} {context[end_idx:]} \n '),
             # unsafe_allow_html=True,
             #  )
-            all_contexts = '\n'.join(contexts)
+            # all_contexts = '\n'.join(contexts)
             st.write(
-                markdown(f"# Answer: {result} \n ## Extracted from context: \n {all_contexts}"),
+                markdown(f"# Answer: {result}")#"\n ## Extracted from context: \n {all_contexts}"),
                 unsafe_allow_html=True,
             )
         except: