Update main.py

main.py

@@ -165,7 +165,8 @@ def forecast(monthly_sales):
 
 def process_product(product_name, data):
     full_trend = ""
-
+    brand_name = data['brand_name'].tolist()[0]
+    product_image =  data['product_image'].tolist()[0]
     sub_sku = data['sub_sku'].tolist()[0]
     try:
         # Get today's date
@@ -196,6 +197,8 @@ def process_product(product_name, data):
         return {
             "sub_sku": sub_sku,
             "Product Name": product_name,
+            "brand_name" : brand_name,
+            "product_image" : f"http://app.storematepro.lk/img/{product_image}",
             "next_month": str(result_dict["next_month"]),
             "predicted_count": result_dict["predicted_count"]
             #"full_trend" : str(full_trend)
@@ -204,6 +207,8 @@ def process_product(product_name, data):
         return {
             "sub_sku": sub_sku,
             "Product Name": product_name,
+            "brand_name" : brand_name,
+            "product_image" : f"http://app.storematepro.lk/img/{product_image}",
             "next_month": str(e),
             "predicted_count": "not predicted"
             #"full_trend" : str(full_trend)