Update app.py

app.py

@@ -1,9 +1,14 @@
+from fastapi import FastAPI, Query
+from pydantic import BaseModel
+from typing import Optional, Dict, Any
 import spacy
-from typing import Dict, Any, Optional
 
 # Load spaCy language model
 nlp = spacy.load("en_core_web_sm")
 
+# FastAPI instance
+app = FastAPI()
+
 # Define mappings for categories and filters
 KEYWORD_MAPPINGS = {
     "categories": {
@@ -27,6 +32,13 @@ KEYWORD_MAPPINGS = {
     },
 }
 
+# Pydantic schema for the input query
+class QueryRequest(BaseModel):
+    sentence: str
+    latitude: Optional[float] = None
+    longitude: Optional[float] = None
+
+# Function to parse the sentence using spaCy
 def parse_sentence_to_query_ner(sentence: str, lat: Optional[float] = None, lng: Optional[float] = None) -> Dict[str, Any]:
     """
     Parse a sentence using spaCy NER and build a search query.
@@ -73,11 +85,21 @@ def parse_sentence_to_query_ner(sentence: str, lat: Optional[float] = None, lng:
 
     return query
 
-# Example usage
-if __name__ == "__main__":
-    sentence = "Top-rated salons near me"
-    latitude = 13.0827
-    longitude = 80.2707
+# FastAPI route
+@app.post("/parse-query/")
+async def parse_query(request: QueryRequest):
+    """
+    API endpoint to parse a query and return a structured response.
+
+    Args:
+        request (QueryRequest): Input sentence and optional location data.
 
-    query = parse_sentence_to_query_ner(sentence, lat=latitude, lng=longitude)
-    print(query)
+    Returns:
+        Dict[str, Any]: Parsed search query.
+    """
+    parsed_query = parse_sentence_to_query_ner(
+        sentence=request.sentence,
+        lat=request.latitude,
+        lng=request.longitude
+    )
+    return {"query": parsed_query}