Update src/streamlit_app.py

src/streamlit_app.py

@@ -23,7 +23,7 @@ index = pc.Index("legal-bot")
 model = SentenceTransformer('all-mpnet-base-v2')
 chat_history = deque(maxlen=10)  # last 5 pairs = 10 messages
 
-st.title("AI Legal Assistant ⚖️👨🏻‍⚖️")
+st.title("AI Legal Assistant ⚖️")
 
 if "history" not in st.session_state:
     st.session_state.history = deque(maxlen=10)
@@ -49,7 +49,7 @@ def get_rewritten_query(user_query):
     except Exception as e:
         st.error(f"Rewrite error: {e}")
         rewritten = user_query
-    st.session_state.history.append({"role": "assistant", "content": f"🔁 Rewritten query: {rewritten}"})
+    # st.session_state.history.append({"role": "assistant", "content": f"🔁 Rewritten query: {rewritten}"})
     return rewritten
 
 def retrieve_documents(query, top_k=10):
@@ -79,17 +79,17 @@ def generate_response(user_query, docs):
         reply = "Sorry, I encountered an error generating the answer."
 
 
-    unique_sources = sorted({d['metadata']['source'] for d in docs})
-    if unique_sources:
-        reply += "\n\n---\n\n**Sources used:**\n"
-        for src in unique_sources:
-            reply += f"> “{src}”\n"
+    # unique_sources = sorted({d['metadata']['source'] for d in docs})
+    # if unique_sources:
+    #     reply += "\n\n---\n\n**Sources used:**\n"
+    #     for src in unique_sources:
+    #         reply += f"> “{src}”\n"
 
-    unique_sources_txt = sorted({d['metadata']['text'] for d in docs})
-    if unique_sources_txt:
-        reply += "\n\n---\n\n**Document(s) retrieved:**\n"
-        for src in unique_sources_txt:
-            reply += f"> “{src}”\n"
+    # unique_sources_txt = sorted({d['metadata']['text'] for d in docs})
+    # if unique_sources_txt:
+    #     reply += "\n\n---\n\n**Document(s) retrieved:**\n"
+    #     for src in unique_sources_txt:
+    #         reply += f"> “{src}”\n"
 
     st.session_state.history.append({"role": "assistant", "content": reply})
     return reply
@@ -107,8 +107,9 @@ if submit and user_input:
     assistant_reply = generate_response(rewritten, docs)
 
 # Display history
-for msg in st.session_state.history:
+st.markdown("---")
+for msg in reversed(st.session_state.history):
     if msg["role"] == "user":
         st.markdown(f"**You:** {msg['content']}")
     else:
-        st.markdown(f"**Assistant:** {msg['content']}")
+        st.markdown(f"**Legal Assistant:** {msg['content']}")