fix problems

app.py

@@ -3,9 +3,9 @@ import chromadb
 from sentence_transformers import SentenceTransformer
 import uuid
 
-# --------------------------------------------------
-# Page Config
-# --------------------------------------------------
+# ==========================================
+# PAGE CONFIG
+# ==========================================
 
 st.set_page_config(
     page_title="Semantic Search Engine",
@@ -13,9 +13,9 @@ st.set_page_config(
     layout="wide"
 )
 
-# --------------------------------------------------
-# CSS
-# --------------------------------------------------
+# ==========================================
+# CUSTOM CSS
+# ==========================================
 
 st.markdown("""
 <style>
@@ -24,26 +24,32 @@ st.markdown("""
     padding-top: 1rem;
 }
 
-.result-card {
+.block-container {
+    padding-top: 2rem;
+}
+
+.result-box {
     padding: 1rem;
     border-radius: 12px;
-    border: 1px solid #444;
+    border: 1px solid #333;
     margin-bottom: 10px;
 }
 
 </style>
 """, unsafe_allow_html=True)
 
-# --------------------------------------------------
-# Title
-# --------------------------------------------------
+# ==========================================
+# TITLE
+# ==========================================
 
 st.title("🔍 Semantic Search Engine")
-st.caption("Search documents using meaning instead of keywords.")
+st.caption(
+    "Search documents using semantic similarity powered by Hugging Face embeddings."
+)
 
-# --------------------------------------------------
-# Load Model
-# --------------------------------------------------
+# ==========================================
+# LOAD MODEL
+# ==========================================
 
 @st.cache_resource
 def load_model():
@@ -53,9 +59,9 @@ def load_model():
 
 model = load_model()
 
-# --------------------------------------------------
-# ChromaDB
-# --------------------------------------------------
+# ==========================================
+# CHROMADB
+# ==========================================
 
 client = chromadb.PersistentClient(
     path="./chroma_db"
@@ -65,9 +71,9 @@ collection = client.get_or_create_collection(
     name="documents"
 )
 
-# --------------------------------------------------
-# Sidebar
-# --------------------------------------------------
+# ==========================================
+# SIDEBAR
+# ==========================================
 
 with st.sidebar:
 
@@ -83,12 +89,12 @@ with st.sidebar:
     st.markdown("---")
 
     st.info(
-        "Semantic Search compares meanings using embeddings."
+        "Semantic Search compares meanings instead of matching exact keywords."
     )
 
-# --------------------------------------------------
-# Database Stats
-# --------------------------------------------------
+# ==========================================
+# DATABASE STATS
+# ==========================================
 
 st.markdown("## 📊 Database Statistics")
 
@@ -106,16 +112,16 @@ with col2:
         "MiniLM-L6-v2"
     )
 
-# --------------------------------------------------
-# Add Documents
-# --------------------------------------------------
+# ==========================================
+# DOCUMENT INPUT
+# ==========================================
 
 st.markdown("---")
 st.markdown("## 📥 Add Documents")
 
 documents = st.text_area(
-    "Enter Documents (one per line)",
-    height=250,
+    "Enter documents (one document per line)",
+    height=220,
     placeholder="""
 Python is a programming language.
 FastAPI is used to build APIs.
@@ -153,20 +159,20 @@ if st.button("💾 Store Documents"):
             )
 
         st.success(
-            f"{len(docs)} documents stored successfully!"
+            f"{len(docs)} document(s) stored successfully."
         )
 
         st.rerun()
 
-# --------------------------------------------------
-# Search
-# --------------------------------------------------
+# ==========================================
+# SEARCH SECTION
+# ==========================================
 
 st.markdown("---")
-st.markdown("## 🔎 Search Documents")
+st.markdown("## 🔎 Search")
 
 query = st.text_input(
-    "Enter Search Query",
+    "Enter your search query",
     placeholder="How can I build an API?"
 )
 
@@ -178,7 +184,7 @@ if st.button(
     if collection.count() == 0:
 
         st.error(
-            "No documents found. Add documents first."
+            "No documents available. Add documents first."
         )
 
     elif not query.strip():
@@ -190,7 +196,7 @@ if st.button(
     else:
 
         with st.spinner(
-            "Searching relevant documents..."
+            "Searching similar documents..."
         ):
 
             query_embedding = model.encode(
@@ -210,25 +216,33 @@ if st.button(
         docs = results["documents"][0]
         distances = results["distances"][0]
 
-        st.markdown("## 📄 Results")
+        st.markdown("---")
+        st.markdown("## 📄 Search Results")
 
         for rank, (doc, distance) in enumerate(
             zip(docs, distances),
             start=1
         ):
 
-            similarity = max(
-                0,
-                (1 - distance) * 100
-            )
+            # Relevance Label
+            if distance < 0.7:
+                relevance = "🟢 Highly Relevant"
+            elif distance < 1.2:
+                relevance = "🟡 Relevant"
+            else:
+                relevance = "🔴 Weak Match"
 
             with st.expander(
-                f"#{rank} | Similarity Score: {similarity:.2f}%"
+                f"#{rank} | {relevance}"
             ):
                 st.write(doc)
 
-# --------------------------------------------------
-# Footer
-# --------------------------------------------------
+                st.caption(
+                    f"Distance Score: {distance:.4f}"
+                )
+
+# ==========================================
+# FOOTER
+# ==========================================
 
 st.markdown("---")