| |
| """Chart Generation Plugin""" |
| import plotly.express as px |
| import plotly.graph_objects as go |
| from typing import Dict, Any |
| import pandas as pd |
| import json |
|
|
| class ChartGenerator: |
| """Generate interactive charts.""" |
| def create_chart_html(self, data: pd.DataFrame, chart_type: str, x: str, y: str = None, |
| title: str = "Chart", color: str = None, **kwargs) -> str: |
| |
| if data.empty or x not in data.columns: |
| return json.dumps({"error": "Invalid data"}) |
| |
| try: |
| if chart_type == "bar": |
| fig = px.bar(data, x=x, y=y, title=title, color=color, **kwargs) |
| elif chart_type == "line": |
| fig = px.line(data, x=x, y=y, title=title, color=color, **kwargs) |
| elif chart_type == "scatter": |
| fig = px.scatter(data, x=x, y=y, title=title, color=color, **kwargs) |
| elif chart_type == "pie": |
| fig = px.pie(data, names=x, values=y, title=title, **kwargs) |
| else: |
| fig = px.bar(data, x=x, y=y, title=title) |
| |
| fig.update_layout(template="plotly_white", height=400) |
| return fig.to_json() |
| except Exception as e: |
| return json.dumps({"error": str(e)}) |
|
|
