Update app.py

app.py

@@ -22,16 +22,16 @@ def edit_strings(string_list):
       edited_list.append(string)
   return edited_list
 
-def edit_strings(string_list):
-  edited_list = []
-  for string in string_list:
-    if "_" in string:
-      last_underscore_index = string.rfind("_")
-      edited_string = string[:last_underscore_index]
-      edited_list.append(edited_string)
-    else:
-      edited_list.append(string)
-  return edited_list
+def equalize_list_lengths(input_dict):
+  max_len = 0
+  for key in input_dict:
+    max_len = max(max_len, len(input_dict[key]))
+  
+  for key in input_dict:
+    while len(input_dict[key]) < max_len:
+      input_dict[key].append(None)
+      
+  return pd.DataFrame(input_dict)
 
 def figo(plot_type, df, title, xlabel=None, ylabel=None, legend_title=None, colorscale='Plotly3', width=800, height=600):
     if plot_type == "Scatter":