Add missing wrapper functions for visualizations

visualizations.py

@@ -170,3 +170,39 @@ def figure_to_png_bytes(fig: go.Figure) -> BytesIO:
     except ValueError as exc:  # pragma: no cover - fallback for environments without kaleido
         raise ValueError("PNG export requires the 'kaleido' package. Please install it to enable downloads.") from exc
     return BytesIO(image_bytes)
+
+
+def create_time_series_plot(df: pd.DataFrame, date_column: str, value_column: str, aggregation: str = "sum") -> go.Figure:
+    """Generate a time-series plot using the TimeSeriesStrategy."""
+    strategy = TimeSeriesStrategy()
+    return strategy.generate(df, date_column=date_column, value_column=value_column, aggregation=aggregation)
+
+
+def create_distribution_plot(df: pd.DataFrame, column: str, plot_type: str = "histogram") -> go.Figure:
+    """Generate a distribution plot using the DistributionStrategy."""
+    strategy = DistributionStrategy()
+    return strategy.generate(df, column=column, plot_type=plot_type)
+
+
+def create_category_plot(
+    df: pd.DataFrame, category_column: str, value_column: str, aggregation: str = "sum", chart_type: str = "bar"
+) -> go.Figure:
+    """Generate a category plot using the CategoryStrategy."""
+    strategy = CategoryStrategy()
+    return strategy.generate(
+        df, category_column=category_column, value_column=value_column, aggregation=aggregation, chart_type=chart_type
+    )
+
+
+def create_scatter_plot(
+    df: pd.DataFrame, x_column: str, y_column: str, color_column: Optional[str] = None
+) -> go.Figure:
+    """Generate a scatter plot using the ScatterStrategy."""
+    strategy = ScatterStrategy()
+    return strategy.generate(df, x_column=x_column, y_column=y_column, color_column=color_column)
+
+
+def create_correlation_heatmap(df: pd.DataFrame) -> go.Figure:
+    """Generate a correlation heatmap using the CorrelationHeatmapStrategy."""
+    strategy = CorrelationHeatmapStrategy()
+    return strategy.generate(df)