Update app.py

app.py

@@ -154,18 +154,38 @@ if query:
                     with st.expander("🔍 View Source Data used for this summary"):
                         st.text(full_doc_text[:2000] + "...")
 
-        # --- SEARCH RESULTS SECTION ---
-        st.subheader("Reference Documents")
-        for res in results:
-            score = res['score']
-            # Dynamic color coding based on relevance
-            color = "#09ab3b" if score > 2 else "#ffbd45" if score > 0 else "#ff4b4b"
+        # --- SEARCH RESULTS SECTION (Updated) ---
+        # We wrap the results in an expander to keep the UI clean
+        with st.expander("📚 Reference Documents (Click to view)", expanded=False):
             
-            with st.container():
+            if not results:
+                st.info("No matching documents found.")
+            
+            for res in results:
+                score = res['score']
+                # Dynamic color coding based on relevance
+                color = "#09ab3b" if score > 2 else "#ffbd45" if score > 0 else "#ff4b4b"
+                
+                # CSS FIX:
+                # 1. We kept 'background-color: #f0f2f6' (Light Gray)
+                # 2. We ADDED 'color: #1f1f1f' (Dark Gray) to force readability 
+                #    regardless of whether the user is in Dark Mode or Light Mode.
+                
                 st.markdown(f"""
-                <div style="border-left: 5px solid {color}; padding: 10px; background-color: #f0f2f6; margin-bottom: 10px; border-radius: 5px;">
-                    <h4 style="margin:0;">📄 {res['source']}</h4>
-                    <p style="margin:0; font-style: italic; font-size: 0.9em;">"...{res['snippet']}..."</p>
-                    <small>Relevance Score: {score:.2f}</small>
+                <div style="
+                    border-left: 5px solid {color}; 
+                    padding: 15px; 
+                    background-color: #f0f2f6; 
+                    margin-bottom: 10px; 
+                    border-radius: 5px;
+                    color: #1f1f1f;
+                ">
+                    <div style="display: flex; justify-content: space-between; align-items: center;">
+                        <h4 style="margin:0; color: #0e1117;">📄 {res['source']}</h4>
+                        <span style="font-size: 0.8em; color: #555; background: #ddd; padding: 2px 6px; border-radius: 4px;">Score: {score:.2f}</span>
+                    </div>
+                    <p style="margin: 8px 0; font-style: italic; font-size: 0.95em; color: #333; line-height: 1.5;">
+                        "...{res['snippet']}..."
+                    </p>
                 </div>
                 """, unsafe_allow_html=True)