Update app.py

app.py

@@ -1,65 +1,71 @@
 import streamlit as st
 from search_utils import SemanticSearch
 
-@st.cache_resource
-def init_search_system():
-    search_system = SemanticSearch()
-    search_system.initialize_system()
-    return search_system
+def main():
+    st.set_page_config(
+        page_title="Semantic Search Engine",
+        page_icon="🔍",
+        layout="wide"
+    )
 
-st.set_page_config(
-    page_title="Semantic Search Engine",
-    page_icon="🔍",
-    layout="wide"
-)
+    # Initialize search system first
+    @st.cache_resource
+    def init_search_system():
+        system = SemanticSearch()
+        system.initialize_system()
+        return system
 
-# 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: #1e88e5 !important;
-    text-decoration: none !important;
-    border-bottom: 1px dotted #1e88e5;
-}
-a.source-link:hover {
-    opacity: 0.8;
-    border-bottom-style: solid;
-}
-</style>
-""", unsafe_allow_html=True)
+    # Custom CSS moved outside cached function
+    st.markdown("""
+    <style>
+    div[data-testid="stExpander"] div[role="button"] p {
+        font-size: 1.2rem;
+        font-weight: bold;
+    }
+    a.source-link {
+        color: #1e88e5 !important;
+        text-decoration: none !important;
+        border-bottom: 1px dotted #1e88e5;
+    }
+    a.source-link:hover {
+        opacity: 0.8;
+        border-bottom-style: solid;
+    }
+    </style>
+    """, unsafe_allow_html=True)
 
-search_system = init_search_system()
+    search_system = init_search_system()
 
-# Main UI
-st.title("🔍 Semantic Search Engine")
-query = st.text_input("Enter your search query:", placeholder="Search documents...")
+    # Main UI components
+    st.title("🔍 Semantic Search Engine")
+    query = st.text_input("Enter your search query:", placeholder="Search documents...")
 
-if query:
-    with st.spinner("🔍 Searching through documents..."):
-        results = search_system.search(query, 5)
-        
-        # 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")
+    if query:
+        with st.spinner("🔍 Searching through documents..."):
+            results = search_system.search(query, 5)
+            
+            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.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))
+        # System status sidebar
+        with st.sidebar:
+            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))
 
+    # Sidebar controls outside main query block
+    with st.sidebar:
         if st.button("Clear Cache"):
             st.cache_resource.clear()
-            st.rerun()
+            st.rerun()
+
+if __name__ == "__main__":
+    main()