Update app.py

app.py

@@ -5,8 +5,7 @@ import plotly.express as px
 import plotly.graph_objects as go
 from scipy.stats import norm
 
-def figo(plot_type, df, title, xlabel=None, ylabel=None, legend_title=None, colorscale='Plotly3'):
-
+def figo(plot_type, df, title, xlabel=None, ylabel=None, legend_title=None, colorscale='Plotly3', width=800, height=600):
     if plot_type == "Scatter":
         fig = go.Figure()
 
@@ -26,7 +25,9 @@ def figo(plot_type, df, title, xlabel=None, ylabel=None, legend_title=None, colo
             title=title,
             xaxis_title="Percentage",
             yaxis_title="Category",
-            yaxis={'categoryorder': 'array', 'categoryarray': sorted_data.index}
+            yaxis={'categoryorder': 'array', 'categoryarray': sorted_data.index},
+            width=width,
+            height=height
         )
     
     elif plot_type == "Heatmap":
@@ -51,7 +52,9 @@ def figo(plot_type, df, title, xlabel=None, ylabel=None, legend_title=None, colo
             xaxis_title=xlabel,
             yaxis_title=ylabel,
             legend_title=legend_title,
-            template="plotly_white"
+            template="plotly_white",
+            width=width,
+            height=height
         )
 
     elif plot_type == "Bar":
@@ -63,8 +66,6 @@ def figo(plot_type, df, title, xlabel=None, ylabel=None, legend_title=None, colo
             name=col
         ))
         
-        fig.update_layout(barmode='group')
-        
         fig.update_layout(
             title={
                 'text': title,
@@ -76,17 +77,17 @@ def figo(plot_type, df, title, xlabel=None, ylabel=None, legend_title=None, colo
             xaxis_title=xlabel,
             yaxis_title=ylabel,
             legend_title=legend_title,
-            template="plotly_white"
+            template="plotly_white",
+            barmode='group',
+            width=width,
+            height=height
         )
 
     else:
-        raise ValueError("Invalid plot_type. Supported types are 'Heatmap' and 'Bar'.")
-
+        raise ValueError("Invalid plot_type. Supported types are 'Scatter', 'Heatmap', and 'Bar'.")
 
     return fig
 
-
-
 def is_matching_pattern(column, prefix):
     if not column.startswith(prefix + '_'):
         return False
@@ -378,7 +379,8 @@ if main_option == "Tabulation":
                         st.subheader("Frequency Table")
                         st.dataframe(frequency_df)
 
-                        fig = figo('Scatter', percentile_df.iloc[:-1,:], title='Percentage Scatter plot')
+                        row, col = df.shape
+                        fig = figo('Scatter', percentile_df.iloc[:-1,:], title='Percentage Scatter plot', width=len(col)*50, height=len(row)*50)
                         st.plotly_chart(fig, use_container_width=True)
 
                     elif type1 == "Single answer" and type2 == "Multiple answer":
@@ -391,7 +393,7 @@ if main_option == "Tabulation":
                             st.subheader("Frequency Table")
                             st.dataframe(frequency_df)
                             
-                            fig = figo('Scatter', percentile_df.iloc[:-1,:], title='Percentage Scatter plot')
+                            fig = figo('Scatter', percentile_df.iloc[:-1,:], title='Percentage Scatter plot', width=len(col)*50, height=len(row)*50)
                             st.plotly_chart(fig, use_container_width=True)
                             
                         else:
@@ -408,7 +410,7 @@ if main_option == "Tabulation":
                             st.subheader("Frequency Table")
                             st.dataframe(frequency_df)
                             
-                            fig = figo('Scatter', percentile_df.iloc[:-1,:], title='Percentage Scatter plot')
+                            fig = figo('Scatter', percentile_df.iloc[:-1,:], title='Percentage Scatter plot', width=len(col)*50, height=len(row)*50)
                             st.plotly_chart(fig, use_container_width=True)
                             
                         else: