Update app.py

app.py

@@ -117,6 +117,27 @@ def two_variable_sm(df, var1, var2):
     percentage_dataframe.loc['Sample_size'] = list(single_answer(df[var1]).iloc[:,1])
     
 
+    return percentage_dataframe, friquency_dataframe
+
+def two_variable_mm(df, var1, var2):
+    friquency_dataframe, percentage_dataframe = {}, {}
+    value = multi_answer(df[var2]).iloc[:-1,0]
+
+    for i in var1:
+        unique_values = list(set(df[i].dropna()))[0]
+        dataframe = multi_answer(df[df[i] == unique_values][var2]).iloc[:-1,:]
+        friquency_dataframe[i], percentage_dataframe[i] = dataframe['Friquency'], dataframe['Percentage']
+
+    friquency_dataframe = pd.DataFrame(friquency_dataframe)
+    percentage_dataframe = pd.DataFrame(percentage_dataframe)
+
+    friquency_dataframe.index, percentage_dataframe.index = value, value
+
+    friquency_dataframe['Total'] = list(multi_answer(df[var2]).iloc[:,1])[:-1]
+    friquency_dataframe.loc['Sample_size'] = list(multi_answer(df[var1]).iloc[:,1])
+    percentage_dataframe['Total'] = list(multi_answer(df[var2]).iloc[:,2])[:-1]
+    percentage_dataframe.loc['Sample_size'] = list(multi_answer(df[var1]).iloc[:,1])
+
     return percentage_dataframe, friquency_dataframe
 
 # Functions related to Z-Test
@@ -298,6 +319,24 @@ if main_option == "Tabulation":
                             
                         else:
                             st.error("No columns matching the entered pattern were found.")
+
+                    elif type1 == "Multiple answer" and type2 == "Multiple answer":
+                        matching_cols1 = [col for col in df.columns if is_matching_pattern(col, var1)]
+                        matching_cols2 = [col for col in df.columns if is_matching_pattern(col, var2)]
+                        if matching_cols1 and matching_cols2:
+                            percentile_df, frequency_df = two_variable_mm(df[matching_cols1 + matching_cols2], matching_cols1, matching_cols2)
+                            st.subheader("Percentage Table")
+                            st.dataframe(percentile_df)
+
+                            st.subheader("Frequency Table")
+                            st.dataframe(frequency_df)
+                            
+                            fig = figo('Heatmap', percentile_df.iloc[:-1,:-1], title='Percentage Histogram', xlabel=var1, ylabel=var2, colorscale='Plotly3')
+                            st.plotly_chart(fig, use_container_width=True)
+                            
+                        else:
+                            st.error("No columns matching the entered pattern were found.")
+                    
                     else:
                         st.info("This section of the program is under development.")