Update sozo_gen.py

sozo_gen.py

@@ -172,7 +172,32 @@ class ChartGenerator:
             # In the new architecture, we cannot fall back to a placeholder. We must fail.
             raise ValueError(f"Failed to generate a valid chart specification for '{description}'.") from e
 
+# PASTE THIS CODE INTO YOUR sozo_gen.py FILE
 
+def execute_chart_spec(spec: ChartSpecification, df: pd.DataFrame, output_path: Path) -> bool:
+    try:
+        plot_data = prepare_plot_data(spec, df)
+        fig, ax = plt.subplots(figsize=(12, 8)); plt.style.use('default')
+        if spec.chart_type == "bar": ax.bar(plot_data.index.astype(str), plot_data.values, color='#2E86AB', alpha=0.8); ax.set_xlabel(spec.x_col); ax.set_ylabel(spec.y_col); ax.tick_params(axis='x', rotation=45)
+        elif spec.chart_type == "pie": ax.pie(plot_data.values, labels=plot_data.index, autopct='%1.1f%%', startangle=90); ax.axis('equal')
+        elif spec.chart_type == "line": ax.plot(plot_data.index, plot_data.values, marker='o', linewidth=2, color='#A23B72'); ax.set_xlabel(spec.x_col); ax.set_ylabel(spec.y_col); ax.grid(True, alpha=0.3)
+        elif spec.chart_type == "scatter": ax.scatter(plot_data.iloc[:, 0], plot_data.iloc[:, 1], alpha=0.6, color='#F18F01'); ax.set_xlabel(spec.x_col); ax.set_ylabel(spec.y_col); ax.grid(True, alpha=0.3)
+        elif spec.chart_type == "hist": ax.hist(plot_data.values, bins=20, color='#C73E1D', alpha=0.7, edgecolor='black'); ax.set_xlabel(spec.x_col); ax.set_ylabel('Frequency'); ax.grid(True, alpha=0.3)
+        ax.set_title(spec.title, fontsize=14, fontweight='bold', pad=20); plt.tight_layout()
+        plt.savefig(output_path, dpi=300, bbox_inches='tight', facecolor='white'); plt.close()
+        return True
+    except Exception as e: logging.error(f"Static chart generation failed for '{spec.title}': {e}"); return False
+
+def prepare_plot_data(spec: ChartSpecification, df: pd.DataFrame) -> pd.Series:
+    if spec.x_col not in df.columns or (spec.y_col and spec.y_col not in df.columns): raise ValueError(f"Invalid columns in chart spec: {spec.x_col}, {spec.y_col}")
+    if spec.chart_type in ["bar", "pie"]:
+        if not spec.y_col: return df[spec.x_col].value_counts().nlargest(spec.top_n or 10)
+        grouped = df.groupby(spec.x_col)[spec.y_col].agg(spec.agg_method or 'sum')
+        return grouped.nlargest(spec.top_n or 10)
+    elif spec.chart_type == "line": return df.set_index(spec.x_col)[spec.y_col].sort_index()
+    elif spec.chart_type == "scatter": return df[[spec.x_col, spec.y_col]].dropna()
+    elif spec.chart_type == "hist": return df[spec.x_col].dropna()
+    return df[spec.x_col]
 # --- Main Business Logic Functions ---
 
 # In sozo_gen.py