|
|
|
|
|
"""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)}) |
|
|
|
