Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,6 +4,7 @@ import sys
 import subprocess
 from pathlib import Path
 from typing import List
+import re
 
 import streamlit as st
 from sentence_transformers import SentenceTransformer
@@ -47,6 +48,27 @@ def get_query_embedding(query: str) -> List[float]:
     return model.encode(query).tolist()
 
 
+# ================= HTML → MARKDOWN RENDER =================
+def render_answer(answer: str):
+    """
+    Converts StackOverflow-style HTML into clean Streamlit output
+    """
+
+    # Split text and code blocks
+    parts = re.split(r"<pre><code>|</code></pre>", answer)
+
+    for i, part in enumerate(parts):
+        if i % 2 == 0:
+            # Normal text → remove simple HTML tags
+            clean_text = re.sub(r"<.*?>", "", part)
+            if clean_text.strip():
+                st.markdown(clean_text)
+        else:
+            # Code block
+            code = part.strip()
+            st.code(code, language="python")  # default language
+
+
 # ================= CHAT STATE =================
 def init_chat():
     if "messages" not in st.session_state:
@@ -68,7 +90,9 @@ def main():
     with st.sidebar:
         st.title("⚙️ Settings")
         top_k = st.slider("Number of results", 1, 10, 5)
+
         st.markdown("---")
+
         if st.button("🗑️ Clear Chat"):
             st.session_state.messages = []
             st.rerun()
@@ -80,13 +104,16 @@ def main():
     # ================= DISPLAY CHAT =================
     for msg in st.session_state.messages:
         with st.chat_message(msg["role"]):
-            st.markdown(msg["content"])
+            if msg["role"] == "assistant":
+                render_answer(msg["content"])
+            else:
+                st.markdown(msg["content"])
 
     # ================= INPUT =================
     user_input = st.chat_input("Ask a programming question...")
 
     if user_input:
-        # Show user message
+        # Store user message
         st.session_state.messages.append({
             "role": "user",
             "content": user_input
@@ -103,26 +130,30 @@ def main():
                     query_embedding = get_query_embedding(user_input.strip())
                     results = engine.search(query_embedding, top_k=top_k)
 
-                # Format response
-                response = ""
+                # Save raw response for rendering
+                full_response = ""
+
                 for i, item in enumerate(results, start=1):
-                    response += f"### 🔹 Result {i}\n"
-                    response += f"**{item['question']}**\n\n"
-                    response += f"{item['answer']}\n\n"
-                    response += f"`Score: {item['score']:.4f}`\n\n"
-                    response += "---\n\n"
+                    st.markdown(f"### 🔹 Result {i}")
+                    st.markdown(f"**{item['question']}**")
+
+                    render_answer(item["answer"])
+
+                    st.caption(f"Score: {item['score']:.4f}")
+                    st.divider()
 
-                st.markdown(response)
+                    # Save for history (raw)
+                    full_response += f"{item['question']}\n{item['answer']}\n"
 
         except Exception as e:
-            response = f"❌ Error: {e}"
+            full_response = f"Error: {e}"
             with st.chat_message("assistant"):
-                st.error(response)
+                st.error(full_response)
 
-        # Save assistant response
+        # Store assistant message
         st.session_state.messages.append({
             "role": "assistant",
-            "content": response
+            "content": full_response
         })