Update app.py

app.py

@@ -15,6 +15,11 @@ import uuid
 from qdrant_client import models, QdrantClient
 from itertools import islice
 
+# The file where NeuralSearcher is stored
+from neural_searcher import NeuralSearcher
+# The file where HybridSearcher is stored
+from hybrid_searcher import HybridSearcher
+
 app = FastAPI()
 
 
@@ -144,6 +149,40 @@ async def download_database():
     
     # Return the zip file as a response for download
     return FileResponse(zip_path, media_type='application/zip', filename='database.zip')
+
+@app.get("/neural_search")
+def neural_search(q: str, city: str, collection_name: str):
+    import time
+
+    start_time = time.time()
+    
+    # Create a neural searcher instance
+    neural_searcher = NeuralSearcher(collection_name=collection_name)
+
+    end_time = time.time()
+
+    elapsed_time = end_time - start_time
+    
+    print(f"Execution time: {elapsed_time:.6f} seconds")
+    
+    return {"result": neural_searcher.search(text=q, city=city)}
+
+@app.get("/hybrid_search")
+def hybrid_search(q: str, city: str, collection_name: str):
+    import time
+
+    start_time = time.time()
+    
+    # Create a hybrid searcher instance
+    hybrid_searcher = HybridSearcher(collection_name=collection_name)
+    
+    end_time = time.time()
+
+    elapsed_time = end_time - start_time
+    
+    print(f"Execution time: {elapsed_time:.6f} seconds")
+    
+    return {"result": hybrid_searcher.search(text=q, city=city)}
     
 @app.get("/")
 def api_home():