Update tools/plot_generator.py

tools/plot_generator.py

@@ -1,25 +1,28 @@
 import pandas as pd
 import plotly.graph_objects as go
 
-def plot_sales_tool(file_path: str, date_col: str = "Month") -> str:
+def plot_metric_tool(file_path: str, date_col: str, value_col: str):
     """
-    Builds an interactive Plotly line chart for Sales.
-    Saves a PNG fallback (sales_plot.png) for Streamlit thumbnail.
+    Generic interactive line chart for any numeric metric.
+    Returns a Plotly Figure or error message string.
     """
     df = pd.read_csv(file_path)
 
-    if date_col not in df.columns or "Sales" not in df.columns:
-        return f"❌ '{date_col}' or 'Sales' column missing."
+    if date_col not in df.columns or value_col not in df.columns:
+        return f"❌ '{date_col}' or '{value_col}' column missing."
+
+    df[date_col] = pd.to_datetime(df[date_col], errors="coerce")
+    if df[date_col].isna().all():
+        return f"❌ '{date_col}' could not be parsed as dates."
 
     fig = go.Figure(
         go.Scatter(
             x=df[date_col],
-            y=df["Sales"],
+            y=df[value_col],
             mode="lines+markers",
             line=dict(width=2),
         )
     )
-    fig.update_layout(title="Sales Trend", template="plotly_dark")
-    fig.write_image("sales_plot.png")  # still save for PNG fallback
-
+    fig.update_layout(title=f"{value_col} Trend", template="plotly_dark")
+    fig.write_image("metric_plot.png")
     return fig