Create visualizer.py

visualizer.py

@@ -0,0 +1,14 @@
+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