Upload main.py

main.py

@@ -46,7 +46,12 @@ def predict_sales():
 
     sequence_len = 60
     if len(sales_data) <= sequence_len:
-        return jsonify({'code': 400, 'message': 'Insufficient data points for training after adding missing dates.'}), 400
+        num_missing_dates = sequence_len - len(sales_data)
+        missing_dates_df = pd.DataFrame({
+            'Date': pd.date_range(end=min_date, periods=num_missing_dates, freq='D'),
+            'Daily Sales': [0] * num_missing_dates
+        })
+        sales_data = pd.concat([missing_dates_df, sales_data], ignore_index=True)
 
     daily_sales = sales_data['Daily Sales']
     values = daily_sales.values