Update app.py

app.py

@@ -54,11 +54,20 @@ def infinite_scroll_programs(q: int = Query(0, ge=0)):
     start = q * 10
     return scroll_data["infiniteScrollPrograms"][start : start + 10]
 
-@app.get("/searchSomething/")
-def search_something(q: str):
-    key = "packages" if "package" in q.lower() else "programs"
-    if key not in indices:
-        return {"error": "Invalid category"}
+@app.get("/searchPackages/")
+def search_packages(q: str):
+    key = "packages"
+    index, dataset = indices[key]
+    query_embedding = model.encode([q])
+    distances, indices_ = index.search(np.array(query_embedding), len(dataset))
+    min_distance = distances[0][0]
+    threshold = min_distance * 1.5
+    results = [dataset[int(i)] for d, i in zip(distances[0], indices_[0]) if d <= threshold]
+    return results[:280] if len(results) > 280 else results
+
+@app.get("/searchPrograms/")
+def search_programs(q: str):
+    key = "programs"
     index, dataset = indices[key]
     query_embedding = model.encode([q])
     distances, indices_ = index.search(np.array(query_embedding), len(dataset))