Update main.py

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException ,Query
 from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 import data_collector as dc
@@ -271,7 +271,11 @@ def find_related_products(sub_sku=None, product_name=None,forecast_data=None):
     return filtered_products
 
 @app.post("/search_results_from_DB")
-async def get_product_count_prediction_from_DB(b_id: int,search_sku:str,product_name:str):
+async def get_product_count_prediction_from_DB(
+                                               b_id: int,
+                                               search_sku: str = Query(None, description="The SKU of the product to search"),
+                                               product_name: str = Query(None, description="The name of the product to search"):
+    
         related_products = find_related_products(sub_sku=search_sku, product_name=product_name,forecast_data = get_data(b_id,1,2,True))
         response_content = {
             "status": "done",