ui improved

app.py

@@ -67,11 +67,12 @@ vectorstore = load_vectorstore()
 qa = RetrievalQA.from_chain_type(
     llm=ChatGroq(
         groq_api_key=GROQ_API_KEY,
-        model_name="meta-llama/llama-3.1-8b-instruct",
+        model_name="llama-3.1-8b-instant",
         temperature=0
     ),
     retriever=vectorstore.as_retriever(search_kwargs={"k": 3}),
-    chain_type_kwargs={"prompt": PROMPT}
+    chain_type_kwargs={"prompt": PROMPT},
+    return_source_documents=True
 )
 
 question = st.chat_input("Ask your medical question")
@@ -79,4 +80,7 @@ question = st.chat_input("Ask your medical question")
 if question:
     st.chat_message("user").write(question)
     result = qa.invoke({"query": question})
+    st.markdown("#### Answer 😎")
     st.chat_message("assistant").write(result["result"])
+    st.markdown("#### Source Docs 📑")
+    st.chat_message("assistant").write(result["source_documents"])