| import plotly.express as px | |
| class Visualizer: | |
| def create_plot(self, df, x_axis, y_axis, chart_type): | |
| if chart_type == "Scatter": | |
| fig = px.scatter(df, x=x_axis, y=y_axis) | |
| elif chart_type == "Line": | |
| fig = px.line(df, x=x_axis, y=y_axis) | |
| elif chart_type == "Bar": | |
| fig = px.bar(df, x=x_axis, y=y_axis) | |
| else: | |
| raise ValueError("Invalid chart type") | |
| return fig |