Initial commit - semantic search engine

Readme.md

@@ -0,0 +1,7 @@
+# Semantic Search Engine
+
+A simple semantic search app using Sentence Transformers and ChromaDB.
+
+## Run locally
+pip install -r requirements.txt
+python app.py

app.py

@@ -0,0 +1,28 @@
+from sentence_transformers import SentenceTransformer
+from chromadb import PersistentClient
+import gradio as gr
+
+# Load ChromaDB
+client = PersistentClient(path="chromadb")
+collection = client.get_collection("my_collection")
+
+# Load embedding model
+model = SentenceTransformer("all-MiniLM-L6-v2")
+
+# Define search
+def semantic_search(query):
+    query_embedding = model.encode([query])
+    results = collection.query(query_embeddings=query_embedding.tolist(), n_results=3)
+    return "\n\n".join(results["documents"][0])
+
+# Launch Gradio interface
+demo = gr.Interface(
+    fn=semantic_search,
+    inputs=gr.Textbox(label="Enter your search query"),
+    outputs=gr.Textbox(label="Top Matches"),
+    title="Semantic Search Engine",
+    description="Search over your custom dataset using semantic similarity."
+)
+
+if __name__ == "__main__":
+    demo.launch()