Update main.py

main.py

@@ -55,18 +55,15 @@ def find_similar_courses(query_text):
 @app.get("/health")
 async def health():
     return {"status": "OK"}
-
 @app.post("/search")
-async def search(request: Request):
-    body = await request.json()  # Get the JSON body of the request
-    query_text = body.get("query")  # Extract the 'query' field from the body
-    
-    if not query_text:
+async def search(query: str):  # Accept `query` directly as a parameter
+    if not query:
         return {"error": "Query parameter is required."}
     
-    top_courses = find_similar_courses(query_text)
+    top_courses = find_similar_courses(query)
     return [course["course_id"] for course in top_courses]
 
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)  # Hugging Face requires port 7860