return results as JSON

app.py

@@ -1,11 +1,15 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
 from src.do_pca_on_tfidf import query_docs
 from src.search_embeddings import sbert_query_docs
 import polars as pl
+#from jinja2 import Template
+
 
 
 app = FastAPI()
 
+
 @app.get("/")
 def default():
 
@@ -30,19 +34,17 @@ def tf(query: str):
 def square(x: int):
     return {"result": x * x}
 
-@app.get("/search")
-def greet_json(query: str):
-    res_tfidf = query_docs(query)
-    res_sbert = sbert_query_docs(query)
+
+@app.get("/search", response_class=JSONResponse)
+async def search(request: Request, q: str):
+    res_tfidf = query_docs(q)
+    res_sbert = sbert_query_docs(q)
 
     joined = res_sbert.join(res_tfidf, on='file', how = 'inner')
 
     res_combined = joined.with_columns((0.7 * pl.col("rank-sbert") + 0.3 * pl.col("rank-tfidf")).alias("rank-combined")).sort("rank-combined")
 
-    
-
-
-    return {"result": res_combined.to_pandas().to_html(), "shape": res_combined.shape}
+    return {"request": request, "results": res_combined.to_dicts()}