Update main.py

main.py

@@ -158,9 +158,9 @@ def forecast(monthly_sales):
 
     return combined_sales, original_forecasted_value, rounded_value
 
-def process_product(product_name, data,sub_sku):
+def process_product(product_name, data):
     full_trend = ""
-
+    print(data)
     try:
         # Get today's date
         today = datetime.today().strftime('%Y-%m-%d %H:%M:%S')
@@ -211,13 +211,11 @@ async def generate_product_count_prediction(b_id: int):
 
         if message == "done":
 
-            # Group the DataFrame by both 'product_name' and 'sub_sku'
-            grouped_df = data.groupby(['product_name', 'sub_sku'])
-    
+            grouped_df = data.groupby('product_name')
+
             results = []
             with ThreadPoolExecutor() as executor:
-                # Submit jobs to the executor, passing both product_name and sub_sku
-                futures = [executor.submit(process_product, product_name, sub_sku, product_df.copy()) for (product_name, sub_sku), product_df in grouped_df]
+                futures = [executor.submit(process_product, product_name, product_df.copy()) for product_name, product_df in grouped_df]
                 for future in as_completed(futures):
                     results.append(future.result())