Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 def group_names(file, members_per_group, fixed_groups_input):
     # Read the CSV file
     df = pd.read_csv(file.name, encoding='utf-8')
-
+    
     # Parse fixed groups input
     fixed_groups = [group.strip() for group in fixed_groups_input.split(';') if group.strip()]
     fixed_groups_df_list = []
@@ -21,7 +21,7 @@ def group_names(file, members_per_group, fixed_groups_input):
 
     # Shuffle the remaining DataFrame
     remaining_df = remaining_df.sample(frac=1).reset_index(drop=True)
-
+    
     # Adjusting fixed groups to include additional members if they're under the specified group size
     for i, group_df in enumerate(fixed_groups_df_list):
         while len(group_df) < members_per_group and not remaining_df.empty:
@@ -34,14 +34,17 @@ def group_names(file, members_per_group, fixed_groups_input):
     for i in range(0, len(remaining_df), members_per_group):
         groups.append(remaining_df[i:i + members_per_group])
 
-    # Creating a new DataFrame for grouped data
-    grouped_data = []
-    for i, group in enumerate(groups):
-        group_name = f'Group {i+1}'
-        names = ', '.join(group['Names'].tolist())
-        grouped_data.append({'Group': group_name, 'Members': names})
-    grouped_df = pd.DataFrame(grouped_data)
+    # Determine the maximum group size
+    max_group_size = max(len(group) for group in groups)
+    
+    # Creating a new DataFrame for grouped data with separate columns for each member
+    grouped_data = {'Group': [f'Group {i+1}' for i in range(len(groups))]}
+    # Add columns for each member
+    for i in range(max_group_size):
+        grouped_data[f'Member{i+1}'] = [group['Names'].tolist()[i] if i < len(group) else "" for group in groups]
 
+    grouped_df = pd.DataFrame(grouped_data)
+    
     return grouped_df
 
 def main_interface(file, members_per_group, fixed_groups_input):
@@ -55,11 +58,11 @@ iface = gr.Interface(
     fn=main_interface,
     inputs=[
         gr.File(label="Upload CSV File"),
-        gr.Number(label="Members per Group", value=4),
-        gr.Textbox(label="Fixed Groups (separated by semicolon)", placeholder="Name1, Name2; Name3, Name4", value=""),
+        gr.Number(label="Members per Group", value=5),
+        gr.Textbox(label="Fixed Groups (separated by semicolon)", placeholder="Group1Name1, Group1Name2; Group2Name1, Group2Name2", value=""),
     ],
     outputs=[gr.Dataframe(label="Grouped Names"), gr.File(label="Download Grouped Names CSV")],
     allow_flagging="never"
 )
 
-iface.launch()
+iface.launch()