Update app.py

app.py

@@ -22,6 +22,17 @@ 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 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()