Update app.py

app.py

@@ -310,13 +310,54 @@ if __name__ == "__main__":
 
     soundfont = "SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2"
 
-    print('Loading Annotated MIDI Dataset processed scores...')
-    processed_scores = TMIDIX.Tegridy_Any_Pickle_File_Reader('processed_scores')
-    processed_scores_labels = [l[0] for l in processed_scores]
+    #===============================================================================
+    # Helper functions
+    #===============================================================================
+    
+    def str_strip_song(string):
+      if string is not None:
+        string = string.replace('-', ' ').replace('_', ' ').replace('=', ' ')
+        str1 = re.compile('[^a-zA-Z ]').sub('', string)
+        return re.sub(' +', ' ', str1).strip().title()
+      else:
+        return ''
+    
+    def str_strip_artist(string):
+      if string is not None:
+        string = string.replace('-', ' ').replace('_', ' ').replace('=', ' ')
+        str1 = re.compile('[^0-9a-zA-Z ]').sub('', string)
+        return re.sub(' +', ' ', str1).strip().title()
+      else:
+        return ''
+    
+    def song_artist_to_song_artist_tokens(file_name):
+        idx = classifier_labels.index(file_name)
+    
+        tok1 = idx // 424
+        tok2 = idx % 424
+    
+        return [tok1, tok2]
+    
+    def song_artist_tokens_to_song_artist(file_name_tokens):
+    
+        tok1 = file_name_tokens[0]
+        tok2 = file_name_tokens[1]
+    
+        idx = (tok1 * 424) + tok2
+    
+        return classifier_labels[idx]
+    
+    #===============================================================================
+    
     print('=' * 70)
-
-    print('Loading Annotated MIDI Dataset Classifier Songs Artists Labels...')
-    classifier_labels = TMIDIX.Tegridy_Any_Pickle_File_Reader('Annotated_MIDI_Dataset_Classifier_Songs_Artists_Labels')
+    print('Loading Ultimate MIDI Classifier labels...')
+    print('=' * 70)
+    classifier_labels = TMIDIX.Tegridy_Any_Pickle_File_Reader('Ultimate-MIDI-Classifier/Data/Ultimate_MIDI_Classifier_Song_Artist_Labels')
+    print('=' * 70)
+    genre_labels = TMIDIX.Tegridy_Any_Pickle_File_Reader('Ultimate-MIDI-Classifier/Data/Ultimate_MIDI_Classifier_Music_Genre_Labels')
+    genre_labels_fnames = [f[0] for f in genre_labels]
+    print('=' * 70)
+    print('Done!')
     print('=' * 70)
 
     app = gr.Blocks()