Update app.py

app.py

@@ -1,14 +1,13 @@
 import subprocess
 import sys
 
-# Force reinstall pydantic v1 before anything else loads
+# Force reinstall pydantic v1 before any other imports to ensure compatibility with Chromadb.
 subprocess.run([sys.executable, "-m", "pip", "install", "--force-reinstall", "pydantic==1.10.7"])
 
 import os
-# Then proceed with normal imports
+# Set HF Hub timeout to 60 seconds.
 os.environ["HF_HUB_TIMEOUT"] = "60"
 
-
 import streamlit as st
 import streamlit.components.v1 as components
 from backend import process_medical_query, docs_cache
@@ -20,21 +19,14 @@ def main():
         "**Using BioGPT-Large-PubMedQA + PubMed + Chroma** for advanced retrieval-augmented generation."
     )
 
+    # Clinical query input – designed for clarity and ease-of-use.
     user_query = st.text_input("Enter biomedical/medical query", "Malaria and cough treatment")
     if st.button("Submit"):
         with st.spinner("Generating answer..."):
+            # Process the query using the streamlined backend.
             final_answer, sub_questions, initial_answer, critique = process_medical_query(user_query)
 
-        st.subheader("Sub-Question Decomposition")
-        st.write(sub_questions)
-
-        st.subheader("Initial AI Answer")
-        st.write(initial_answer)
-
-        st.subheader("Self-Critique")
-        st.write(critique)
-
-        st.subheader("Refined AI Answer")
+        st.subheader("AI Answer")
         st.write(final_answer)
 
         st.subheader("Knowledge Graph")