Update app.py

app.py

@@ -76,7 +76,7 @@ def retrieve(query):
 
     # get relevant contexts
     res = index.query(xq, top_k=3, include_metadata=True)
-    contexts = [x["metadata"]["text"] for x in res["matches"]]
+    contexts = [x["metadata"].get("text", "") for x in res["matches"]]
 
     # build our prompt with the retrieved contexts included
     prompt_start = "Answer the question based on the context below.\n\n" + "Context:\n"
@@ -325,7 +325,7 @@ if st.session_state.results:
             #  )
             all_contexts = '\n'.join(contexts)
             st.write(
-                markdown(f"Answer: {result} \n Extracted from context {all_contexts}"),
+                markdown(f"# Answer: {result} \n ## Extracted from context: \n {all_contexts}"),
                 unsafe_allow_html=True,
             )
         except: