Update main.py

main.py

@@ -84,7 +84,7 @@ def delete_json(b_id):
     cursor.close()
     connection.close()
 
-def get_data(b_id,page_number,page_size):
+def get_data(b_id,page_number,page_size,search):
     start = (page_number - 1) * page_size
     end = start + page_size
 
@@ -116,15 +116,20 @@ def get_data(b_id,page_number,page_size):
         bid = row[0]
         forecast_data_json = row[1]
         created_at = row[2]
-        
+            
         # Convert JSON string back to Python dictionary
         forecast_data = json.loads(forecast_data_json)
-        result = {
-            "BID":bid,
-            "created_at":created_at,
-            "forecast_data":forecast_data[start:end]
-        }
-        return result
+        
+        if search==False:
+            result = {
+                "BID":bid,
+                "created_at":created_at,
+                "forecast_data":forecast_data[start:end]
+            }
+            return result
+        else:
+            return forecast_data
+            
         
     else:
         return f"plesae genertae forcast for this business id:{b_id}"
@@ -248,7 +253,30 @@ async def get_product_count_prediction_from_DB(b_id: int,page_number :int,page_s
         response_content = {
             "status": "done",
             "message": "data from DB",
-            "data": get_data(b_id,page_number,page_size),
+            "data": get_data(b_id,page_number,page_size,False),
+            "status_code":200
+        }
+        return response_content
+
+
+
+def find_related_products(sub_sku=None, product_name=None,forecast_data=None):
+    print(type(forecast_data))
+    # Filter products based on the given sub_sku or Product Name
+    filtered_products = [
+        item for item in forecast_data
+        if (sub_sku and item['sub_sku'] == sub_sku) or
+           (product_name and item['Product Name'].lower() == product_name.lower())
+    ]
+    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):
+        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",
+            "message": "data from DB",
+            "data": related_products,
             "status_code":200
         }
         return response_content