shortening the name of files

app.py

@@ -1,11 +1,11 @@
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, Request, Query
 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
 
-
+path_prefix = "/Users/wes/Google Drive/Shared drives/datalab/projects/2025_coul_aisearch/data/original_box_download/"
 
 app = FastAPI()
 
@@ -36,13 +36,16 @@ def square(x: int):
 
 
 @app.get("/search", response_class=JSONResponse)
-async def search(q: str):
+def search(q: str = Query(..., description="Search query")):
     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")
+    res_combined = joined.with_columns(
+        (0.7 * pl.col("rank-sbert") + 0.3 * pl.col("rank-tfidf")).alias("rank-combined"),
+        pl.col("file").removeprefix(path_prefix).alias("file")
+    ).sort("rank-combined")
 
     #return {"request": request, "results": str(res_combined)}
     #return {"request": request, "results": res_combined.to_dicts()}