Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import streamlit as st
 from search_utils import SemanticSearch
 
@@ -8,57 +7,59 @@ def init_search_system():
     search_system.initialize_system()
     return search_system
 
-def format_source(url):
-    """Convert URL to clickable link"""
-    return f'<a href="{url}" target="_blank" style="text-decoration: none;">🌐 Source</a>'
-
 st.set_page_config(
     page_title="Semantic Search Engine",
     page_icon="🔍",
     layout="wide"
 )
 
-search_system = init_search_system()
-
-# Custom CSS for better link display
+# Custom CSS for better presentation
 st.markdown("""
 <style>
+div[data-testid="stExpander"] div[role="button"] p {
+    font-size: 1.2rem;
+    font-weight: bold;
+}
 a.source-link {
-    color: #1a73e8 !important;
+    color: #1e88e5 !important;
     text-decoration: none !important;
-    transition: opacity 0.2s;
+    border-bottom: 1px dotted #1e88e5;
 }
 a.source-link:hover {
-    opacity: 0.7;
-    text-decoration: none !important;
+    opacity: 0.8;
+    border-bottom-style: solid;
 }
 </style>
 """, unsafe_allow_html=True)
 
-# Search interface
+search_system = init_search_system()
+
+# Main UI
 st.title("🔍 Semantic Search Engine")
-query = st.text_input("Search knowledge base:", placeholder="Enter your question...")
+query = st.text_input("Enter your search query:", placeholder="Search documents...")
 
 if query:
-    with st.spinner("Searching through documents..."):
+    with st.spinner("🔍 Searching through documents..."):
         results = search_system.search(query, 5)
         
-        # Format sources as links
-        results['source'] = results['source'].apply(
-            lambda x: f'<a class="source-link" href="{x}" target="_blank">🌐 View Source</a>'
-        )
-        
-        # Display results with HTML
-        st.markdown("### Search Results")
-        for _, row in results.iterrows():
-            with st.container(border=True):
-                st.markdown(f"**{row['title']}**")
-                st.markdown(row['summary'])
-                st.markdown(row['source'], unsafe_allow_html=True)
-                st.progress(float(row['similarity']))
-                
-    # Sidebar information
+        # Display results
+        if not results.empty:
+            st.subheader("Top Results")
+            for _, row in results.iterrows():
+                with st.expander(f"{row['title']} (Similarity: {row['similarity']:.1%})"):
+                    st.markdown(f"**Summary**: {row['summary']}")
+                    st.markdown(f"<a class='source-link' href='{row['source']}' target='_blank'>View Source</a>", 
+                              unsafe_allow_html=True)
+        else:
+            st.warning("No matching documents found")
+
+    # System status sidebar
     with st.sidebar:
-        st.markdown("### System Status")
-        st.metric("Loaded Documents", f"{len(search_system.metadata_mgr.shard_map):,}")
-        st.metric("Index Shards", len(search_system.index_shards))
+        st.subheader("System Status")
+        st.metric("Total Documents", f"{search_system.metadata_mgr.total_docs:,}")
+        st.metric("FAISS Shards", len(search_system.index_shards))
+        st.metric("Metadata Shards", len(search_system.metadata_mgr.shard_map))
+
+        if st.button("Clear Cache"):
+            st.cache_resource.clear()
+            st.rerun()