Update app.py

app.py

@@ -1,20 +1,12 @@
 import faiss
 import numpy as np
 from fastapi import FastAPI, Query
-from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
 from datasets import load_dataset
 from sentence_transformers import SentenceTransformer
 
 app = FastAPI()
 
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],  # Change this to specific origins if needed for security
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
 FIELDS = (
     "full_name", "description", "watchers_count", "forks_count", "license",
     "default_branch", "has_build_zig", "has_build_zig_zon", "fork",
@@ -56,12 +48,16 @@ scroll_data = {
 @app.get("/infiniteScrollPackages/")
 def infinite_scroll_packages(q: int = Query(0, ge=0)):
     start = q * 10
-    return scroll_data["infiniteScrollPackages"][start : start + 10]
+    content = scroll_data["infiniteScrollPackages"][start : start + 10]
+    headers = {"Access-Control-Allow-Origin": "*", "Content-Type": "application/json"}
+    return JSONResponse(content=content, headers=headers)
 
 @app.get("/infiniteScrollPrograms/")
 def infinite_scroll_programs(q: int = Query(0, ge=0)):
     start = q * 10
-    return scroll_data["infiniteScrollPrograms"][start : start + 10]
+    content = scroll_data["infiniteScrollPrograms"][start : start + 10]
+    headers = {"Access-Control-Allow-Origin": "*", "Content-Type": "application/json"}
+    return JSONResponse(content=content, headers=headers)
 
 @app.get("/searchPackages/")
 def search_packages(q: str):
@@ -72,10 +68,9 @@ def search_packages(q: str):
     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]
-    final_thing = results[:280] if len(results) > 280 else results
-    for i in final_thing:
-        i.pop("readme_content", None)
-    return final_thing
+    content = results[:280] if len(results) > 280 else results
+    headers = {"Access-Control-Allow-Origin": "*", "Content-Type": "application/json"}
+    return JSONResponse(content=content, headers=headers)
 
 @app.get("/searchPrograms/")
 def search_programs(q: str):
@@ -86,7 +81,6 @@ def search_programs(q: str):
     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]
-    final_thing = results[:280] if len(results) > 280 else results
-    for i in final_thing:
-        i.pop("readme_content", None)
-    return final_thing
+    content = results[:280] if len(results) > 280 else results
+    headers = {"Access-Control-Allow-Origin": "*", "Content-Type": "application/json"}
+    return JSONResponse(content=content, headers=headers)