Update main.py

main.py

@@ -3,6 +3,8 @@ from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 import data_collector as dc
 import pandas as pd
+from prophet import Prophet
+import math
 
 app = FastAPI()
 app.add_middleware(
@@ -13,6 +15,32 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+
+def forecast(monthly_sales):
+  # Prepare the data for Prophet
+  monthly_sales.rename(columns={'transaction_date': 'ds', 'sell_qty': 'y'}, inplace=True)
+
+  # Initialize and fit the Prophet model
+  model = Prophet()
+  model.fit(monthly_sales)
+
+  # Make a future dataframe for the next month
+  future = model.make_future_dataframe(periods=1, freq='M')
+  forecast = model.predict(future)
+
+  # Extract the forecasted sales for the next month
+  forecasted_sales = forecast[['ds', 'yhat']].tail(2)
+
+  # Combine historical and forecasted data
+  combined_sales = pd.concat([monthly_sales, forecasted_sales[-1:]], ignore_index=True)
+  original_forecasted_value = combined_sales.tail(1)
+  rounded_value = combined_sales.tail(1)
+
+  rounded_value['yhat'] = rounded_value['yhat'].apply(lambda x: max(0, math.ceil(x)))
+
+  return combined_sales,original_forecasted_value,rounded_value
+
+
 @app.post("/get_product_count_prediction")
 async def get_product_count_prediction(b_id: int):
     try: