Forecast_Agent / src /summarizer.py
ashantharosary's picture
Upload 6 files
3a83363 verified
Raw
History Blame Contribute Delete
562 Bytes
class Summarizer:
def summarize(self, df, fc, query):
last_actual = df.iloc[-1].to_dict()
forecast_tail = fc.tail(5)[["ds", "yhat"]].to_string(index=False)
summary = (
f"πŸ” **User asked:** {query}\n\n"
f"πŸ“Š **Last actual data:**\n{last_actual}\n\n"
f"πŸ“ˆ **Forecast for next steps:**\n{forecast_tail}\n\n"
f"πŸ“Œ Summary: Based on the recent trends, the forecast suggests continued fluctuation or growth depending on recent values."
)
return summary