Update app.py

app.py

@@ -65,9 +65,8 @@ def search_packages(q: str):
     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]
+    sorted_results = sorted(zip(distances[0], indices_[0]), key=lambda x: x[0])
+    results = [dataset[int(i)] for d, i in sorted_results]
     content = results[:280] if len(results) > 280 else results
     headers = {"Access-Control-Allow-Origin": "*", "Content-Type": "application/json"}
     return JSONResponse(content=content, headers=headers)
@@ -78,9 +77,8 @@ def search_programs(q: str):
     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]
+    sorted_results = sorted(zip(distances[0], indices_[0]), key=lambda x: x[0])
+    results = [dataset[int(i)] for d, i in sorted_results]
     content = results[:280] if len(results) > 280 else results
     headers = {"Access-Control-Allow-Origin": "*", "Content-Type": "application/json"}
     return JSONResponse(content=content, headers=headers)