Update main.py

main.py

@@ -17,20 +17,19 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# main
-data,message = dc.get_data(b_id=637,product_name = "RS 10 TOFFEE")
-print(message)
 
 
 @app.post("/get_image_for_text")
 async def get_image_for_text(b_id:int,product_name:str):
     # main
     data,message = dc.get_data(b_id = b_id , product_name = product_name)
+    
     if message=="done":
       # Summarize the sales count per month
       data['transaction_date'] = pd.to_datetime(data['transaction_date'])
       data.set_index('transaction_date', inplace=True)
       monthly_sales = data['sell_qty'].resample('M').sum().reset_index()
-      full_trend,forecasted_value,rounded_value = forecast(monthly_sales)
+        
+      full_trend,forecasted_value,rounded_value = dc.forecast(monthly_sales)
       print(full_trend,forecasted_value,rounded_value)
       return rounded_value