Update main.py

main.py

@@ -110,7 +110,7 @@ def predict_revenue():
     for doc in query:
         transaction = doc.to_dict()
         data.append({
-            "date": transaction["date"],  # Assuming transaction["date"] is already a datetime object
+            "date": transaction["date"], 
             "amountDue": transaction["amountDue"]
         })
 
@@ -143,10 +143,11 @@ def predict_revenue():
 
     # Extract the forecast for the requested interval
     forecast_data = forecast[['ds', 'yhat']].tail(interval)  
-    predictions = forecast_data['yhat'].tolist()
+    predictions = [{"date": row['ds'].strftime('%Y-%m-%d'), "value": row['yhat']} for _, row in forecast_data.iterrows()]
 
     # Return predictions in JSON format
     return jsonify({"predictedData": predictions})
 
+
 if __name__ == "__main__":
     app.run(debug=True, host="0.0.0.0", port=7860)