Update app.py

app.py

@@ -7,7 +7,27 @@ from scipy.stats import norm
 
 def figo(plot_type, df, title, xlabel=None, ylabel=None, legend_title=None, colorscale='Plotly3'):
 
-    if plot_type == "Heatmap":
+    if plot_type == "Scatter":
+        fig = go.Figure()
+
+        for column in df.columns[0:]:
+            fig.add_trace(go.Scatter(
+                y=df['categories'],
+                x=df[column],
+                mode='lines+markers+text',
+                name=column,
+                text=df[column],
+                textposition="middle right"
+            ))
+        
+        fig.update_layout(
+            title=title,
+            xaxis_title="percentage",
+            yaxis_title="category",
+            yaxis={'categoryorder': 'array', 'categoryarray': data['categories']}
+        )
+    
+    elif plot_type == "Heatmap":
         df = df.apply(pd.to_numeric, errors='coerce')
 
         fig = go.Figure(data=go.Heatmap(
@@ -18,6 +38,20 @@ def figo(plot_type, df, title, xlabel=None, ylabel=None, legend_title=None, colo
             colorscale=colorscale
         ))
 
+        fig.update_layout(
+            title={
+                'text': title,
+                'y':0.95,
+                'x':0.5,
+                'xanchor': 'center',
+                'yanchor': 'top'
+            },
+            xaxis_title=xlabel,
+            yaxis_title=ylabel,
+            legend_title=legend_title,
+            template="plotly_white"
+        )
+
     elif plot_type == "Bar":
         fig = go.Figure()
         col = df.name
@@ -28,23 +62,24 @@ def figo(plot_type, df, title, xlabel=None, ylabel=None, legend_title=None, colo
         ))
         
         fig.update_layout(barmode='group')
+        
+        fig.update_layout(
+            title={
+                'text': title,
+                'y':0.95,
+                'x':0.5,
+                'xanchor': 'center',
+                'yanchor': 'top'
+            },
+            xaxis_title=xlabel,
+            yaxis_title=ylabel,
+            legend_title=legend_title,
+            template="plotly_white"
+        )
 
     else:
         raise ValueError("Invalid plot_type. Supported types are 'Heatmap' and 'Bar'.")
 
-    fig.update_layout(
-        title={
-            'text': title,
-            'y':0.95,
-            'x':0.5,
-            'xanchor': 'center',
-            'yanchor': 'top'
-        },
-        xaxis_title=xlabel,
-        yaxis_title=ylabel,
-        legend_title=legend_title,
-        template="plotly_white"
-    )
 
     return fig