Omkar008 commited on
Commit
cf53afd
1 Parent(s): ad8a5dc

Update routers/queryfilter_router.py

Browse files
Files changed (1) hide show
  1. routers/queryfilter_router.py +6 -5
routers/queryfilter_router.py CHANGED
@@ -7,11 +7,12 @@ router = APIRouter(prefix="/queryfilter")
7
 
8
  @router.post("/gemini")
9
  async def get_data(request:Request):
10
- user_query = await request.json().get('query')
11
-
12
 
13
- user_query = await request.json().get('query')
14
-
15
- return {"brand_name":ai.get_brand_from_query(user_query)}
 
16
 
17
 
 
7
 
8
  @router.post("/gemini")
9
  async def get_data(request:Request):
10
+ body = await request.json().get('query')
11
+ # user_query = body.get('query')
12
 
13
+ user_query = body.get('query', '') if body else ''
14
+
15
+ return {"brand_name": None} if not user_query.strip() else {"brand_name": ai.get_brand_from_query(user_query)}
16
+ # return {"brand_name":ai.get_brand_from_query(user_query)}
17
 
18