Update main.py

main.py

@@ -160,6 +160,7 @@ def forecast(monthly_sales):
 
 def process_product(product_name, data):
     full_trend = ""
+    sub_sku = data.loc[0, 'sub_sku']
     try:
         # Get today's date
         today = datetime.today().strftime('%Y-%m-%d %H:%M:%S')
@@ -175,8 +176,6 @@ def process_product(product_name, data):
         # Concatenate the original DataFrame with the new fake transaction DataFrame
         data = pd.concat([data, fake_transaction_df], ignore_index=True)
         
-        # Extract the 'sub_sku' column
-        sub_sku = df['sub_sku']
         
         # Summarize the sales count per month
         data['transaction_date'] = pd.to_datetime(data['transaction_date'])