visualizer.py

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