seperate mode 1 2 cache dir

app.py

@@ -10,14 +10,13 @@ from modelscope import snapshot_download
 from urllib.parse import urlparse
 from convert import midi2xml, xml2abc, xml2mxl, xml2jpg
 
-CACHE_DIR = "./flagged"
 WEIGHTS_PATH = (
     snapshot_download("Genius-Society/piano_trans", cache_dir="./__pycache__")
     + "/CRNN_note_F1=0.9677_pedal_F1=0.9186.pth"
 )
 
 
-def clean_cache(cache_dir=CACHE_DIR):
+def clean_cache(cache_dir):
     if os.path.exists(cache_dir):
         shutil.rmtree(cache_dir)
 
@@ -41,7 +40,7 @@ def is_url(s: str):
         return False
 
 
-def audio2midi(audio_path: str):
+def audio2midi(audio_path: str, cache_dir: str):
     # Load audio
     audio, _ = load_audio(audio_path, sr=sample_rate, mono=True)
     # Transcriptor
@@ -50,16 +49,16 @@ def audio2midi(audio_path: str):
         checkpoint_path=WEIGHTS_PATH,
     )
     # device: 'cuda' | 'cpu' Transcribe and write out to MIDI file
-    midi_path = f"{CACHE_DIR}/output.mid"
+    midi_path = f"{cache_dir}/output.mid"
     # midi_path = audio_path.replace(audio_path.split(".")[-1], "mid")
     transcriptor.transcribe(audio, midi_path)
     return midi_path, os.path.basename(audio_path).split(".")[-2].capitalize()
 
 
-def upl_infer(audio_path: str):
-    clean_cache()
+def upl_infer(audio_path: str, cache_dir="./__pycache__/mode1"):
+    clean_cache(cache_dir)
     try:
-        midi, title = audio2midi(audio_path)
+        midi, title = audio2midi(audio_path, cache_dir)
         xml = midi2xml(midi, title)
         abc = xml2abc(xml)
         mxl = xml2mxl(xml)
@@ -101,10 +100,10 @@ def music163_song_info(id: str):
     return song_name, free
 
 
-def url_infer(song: str):
-    clean_cache()
+def url_infer(song: str, cache_dir="./__pycache__/mode2"):
     song_name = ""
-    download_path = f"{CACHE_DIR}/output.mp3"
+    clean_cache(cache_dir)
+    download_path = f"{cache_dir}/output.mp3"
     try:
         if is_url(song):
             if "163" in song and "?id=" in song:
@@ -125,7 +124,7 @@ def url_infer(song: str):
 
             download_audio(song, download_path)
 
-        midi, title = audio2midi(download_path)
+        midi, title = audio2midi(download_path, cache_dir)
         if song_name:
             title = song_name