Upload app.py

app.py

@@ -1,6 +1,5 @@
 """
-Semantic Quote Search Engine
-Deploy this to Hugging Face Spaces!
+Semantic Quote Search Engine, 510, Jaideep, week 13
 """
 
 import gradio as gr
@@ -10,13 +9,13 @@ from datasets import load_dataset
 import pandas as pd
 import os
 
-# ============== INITIALIZATION ==============
-print("🚀 Starting Semantic Search Engine...")
+# INITIALIZATIO
+print(" Starting Semantic Search Engine...")
 
 # Load embedding model
-print("📦 Loading embedding model...")
+print(" Loading embedding model...")
 model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
-print("✅ Model loaded!")
+print("Model loaded!")
 
 # Initialize ChromaDB
 chroma_path = "./chroma_db"
@@ -26,9 +25,9 @@ client = chromadb.PersistentClient(path=chroma_path)
 # Check if collection exists, otherwise create it
 try:
     collection = client.get_collection("quotes_collection")
-    print(f"✅ Loaded existing collection with {collection.count()} documents")
+    print(f"Loaded existing collection with {collection.count()} documents")
 except:
-    print("📊 Creating new collection from dataset...")
+    print("Creating new collection from dataset...")
     
     # Load dataset
     dataset = load_dataset("Abirate/english_quotes", split="train")
@@ -74,10 +73,10 @@ except:
             metadatas=metadata[i:end_idx]
         )
     
-    print(f"✅ Collection created with {collection.count()} documents!")
+    print(f" Collection created with {collection.count()} documents!")
 
 
-# ============== SEARCH FUNCTION ==============
+# SEARCH FUNCTION
 def semantic_search(query, n_results=5):
     """
     Perform semantic search over the quotes collection.
@@ -116,11 +115,11 @@ def semantic_search(query, n_results=5):
 def search_quotes(query, num_results):
     """Wrapper for Gradio interface"""
     if not query.strip():
-        return "⚠️ Please enter a search query!"
+        return "⚠ Please enter a search query!"
     return semantic_search(query, n_results=int(num_results))
 
 
-# ============== GRADIO INTERFACE ==============
+# GRADIO INTERFACE 
 demo = gr.Interface(
     fn=search_quotes,
     inputs=[
@@ -134,11 +133,11 @@ demo = gr.Interface(
             maximum=10,
             value=5,
             step=1,
-            label="📊 Number of Results"
+            label=" Number of Results"
         )
     ],
-    outputs=gr.Markdown(label="📚 Search Results"),
-    title="📚 Semantic Quote Search Engine",
+    outputs=gr.Markdown(label="Search Results"),
+    title=" Semantic Quote Search Engine",
     description="""
     ## Search through famous quotes using AI-powered semantic similarity!