Update app.py

app.py

@@ -13,6 +13,7 @@ import copy
 from collections import Counter
 import random
 import statistics
+import re
 
 import gradio as gr
 
@@ -61,6 +62,25 @@ def format_table_data(data_string):
 
 #==========================================================================================================
 
+def clean_string(original_string,
+                 regex=r'[^a-zA-Z0-9 \n]',
+                 remove_duplicate_spaces=True,
+                 title=False
+                ):
+    
+    cstr1 = re.sub(regex, '', original_string)
+
+    if title:
+        cstr1 = cstr1.title()
+
+    if remove_duplicate_spaces:
+        return re.sub(r'[ ]+', ' ', cstr1).strip()
+
+    else:
+        return cstr1
+
+#==========================================================================================================
+
 def ID_MIDI(input_midi):
     
     print('*' * 70)
@@ -133,7 +153,7 @@ def ID_MIDI(input_midi):
 
         output_midi_src_dataset = output_entry[0]
 
-        output_midi_path_str = TMIDIX.clean_string(output_entry[1], regex=r'[^a-zA-Z0-9 .()\n]+')
+        output_midi_path_str = clean_string(output_entry[1])
 
     client = InferenceClient(api_key=HF_TOKEN)