Update app.py

app.py

@@ -44,20 +44,23 @@ scroll_data = {
     "infiniteScrollPrograms": load_dataset_with_fields("zigistry/programs", include_readme=False)[0],
 }
 
-@app.get("/fetch_data/")
-def fetch_data(category: str, page_number: int = Query(0, ge=0)):
-    if category not in scroll_data:
-        return {"error": "Invalid category"}
-    start = page_number * 10
-    return scroll_data[category][start : start + 10]
+@app.get("/infiniteScrollPackages/")
+def infinite_scroll_packages(q: int = Query(0, ge=0)):
+    start = q * 10
+    return scroll_data["infiniteScrollPackages"][start : start + 10]
+
+@app.get("/infiniteScrollPrograms/")
+def infinite_scroll_programs(q: int = Query(0, ge=0)):
+    start = q * 10
+    return scroll_data["infiniteScrollPrograms"][start : start + 10]
 
-@app.get("/search_repositories/")
-def search_repositories(category: str, query: str):
-    key = "packages" if category == "SearchPackages" else "programs"
+@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"}
     index, dataset = indices[key]
-    query_embedding = model.encode([query])
+    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