merge try

app.py

@@ -46,81 +46,66 @@ def most_common_element(input_list):
 
 
 def mp3_to_mel(audio_path: str, width=11.4):
-    os.makedirs(CACHE_DIR, exist_ok=True)
-    try:
-        y, sr = librosa.load(audio_path)
-        mel_spec = librosa.feature.melspectrogram(y=y, sr=sr)
-        log_mel_spec = librosa.power_to_db(mel_spec, ref=np.max)
-        dur = librosa.get_duration(y=y, sr=sr)
-        total_frames = log_mel_spec.shape[1]
-        step = int(width * total_frames / dur)
-        count = int(total_frames / step)
-        begin = int(0.5 * (total_frames - count * step))
-        end = begin + step * count
-        for i in range(begin, end, step):
-            librosa.display.specshow(log_mel_spec[:, i : i + step])
-            plt.axis("off")
-            plt.savefig(
-                f"{CACHE_DIR}/mel_{round(dur, 2)}_{i}.jpg",
-                bbox_inches="tight",
-                pad_inches=0.0,
-            )
-            plt.close()
-
-    except Exception as e:
-        print(f"Error converting {audio_path} : {e}")
+    y, sr = librosa.load(audio_path)
+    mel_spec = librosa.feature.melspectrogram(y=y, sr=sr)
+    log_mel_spec = librosa.power_to_db(mel_spec, ref=np.max)
+    dur = librosa.get_duration(y=y, sr=sr)
+    total_frames = log_mel_spec.shape[1]
+    step = int(width * total_frames / dur)
+    count = int(total_frames / step)
+    begin = int(0.5 * (total_frames - count * step))
+    end = begin + step * count
+    for i in range(begin, end, step):
+        librosa.display.specshow(log_mel_spec[:, i : i + step])
+        plt.axis("off")
+        plt.savefig(
+            f"{CACHE_DIR}/mel_{round(dur, 2)}_{i}.jpg",
+            bbox_inches="tight",
+            pad_inches=0.0,
+        )
+        plt.close()
 
 
 def mp3_to_cqt(audio_path: str, width=11.4):
-    os.makedirs(CACHE_DIR, exist_ok=True)
-    try:
-        y, sr = librosa.load(audio_path)
-        cqt_spec = librosa.cqt(y=y, sr=sr)
-        log_cqt_spec = librosa.power_to_db(np.abs(cqt_spec) ** 2, ref=np.max)
-        dur = librosa.get_duration(y=y, sr=sr)
-        total_frames = log_cqt_spec.shape[1]
-        step = int(width * total_frames / dur)
-        count = int(total_frames / step)
-        begin = int(0.5 * (total_frames - count * step))
-        end = begin + step * count
-        for i in range(begin, end, step):
-            librosa.display.specshow(log_cqt_spec[:, i : i + step])
-            plt.axis("off")
-            plt.savefig(
-                f"{CACHE_DIR}/cqt_{round(dur, 2)}_{i}.jpg",
-                bbox_inches="tight",
-                pad_inches=0.0,
-            )
-            plt.close()
-
-    except Exception as e:
-        print(f"Error converting {audio_path} : {e}")
+    y, sr = librosa.load(audio_path)
+    cqt_spec = librosa.cqt(y=y, sr=sr)
+    log_cqt_spec = librosa.power_to_db(np.abs(cqt_spec) ** 2, ref=np.max)
+    dur = librosa.get_duration(y=y, sr=sr)
+    total_frames = log_cqt_spec.shape[1]
+    step = int(width * total_frames / dur)
+    count = int(total_frames / step)
+    begin = int(0.5 * (total_frames - count * step))
+    end = begin + step * count
+    for i in range(begin, end, step):
+        librosa.display.specshow(log_cqt_spec[:, i : i + step])
+        plt.axis("off")
+        plt.savefig(
+            f"{CACHE_DIR}/cqt_{round(dur, 2)}_{i}.jpg",
+            bbox_inches="tight",
+            pad_inches=0.0,
+        )
+        plt.close()
 
 
 def mp3_to_chroma(audio_path: str, width=11.4):
-    os.makedirs(CACHE_DIR, exist_ok=True)
-    try:
-        y, sr = librosa.load(audio_path)
-        chroma_spec = librosa.feature.chroma_stft(y=y, sr=sr)
-        log_chroma_spec = librosa.power_to_db(np.abs(chroma_spec) ** 2, ref=np.max)
-        dur = librosa.get_duration(y=y, sr=sr)
-        total_frames = log_chroma_spec.shape[1]
-        step = int(width * total_frames / dur)
-        count = int(total_frames / step)
-        begin = int(0.5 * (total_frames - count * step))
-        end = begin + step * count
-        for i in range(begin, end, step):
-            librosa.display.specshow(log_chroma_spec[:, i : i + step])
-            plt.axis("off")
-            plt.savefig(
-                f"{CACHE_DIR}/chroma_{round(dur, 2)}_{i}.jpg",
-                bbox_inches="tight",
-                pad_inches=0.0,
-            )
-            plt.close()
-
-    except Exception as e:
-        print(f"Error converting {audio_path} : {e}")
+    y, sr = librosa.load(audio_path)
+    chroma_spec = librosa.feature.chroma_stft(y=y, sr=sr)
+    log_chroma_spec = librosa.power_to_db(np.abs(chroma_spec) ** 2, ref=np.max)
+    dur = librosa.get_duration(y=y, sr=sr)
+    total_frames = log_chroma_spec.shape[1]
+    step = int(width * total_frames / dur)
+    count = int(total_frames / step)
+    begin = int(0.5 * (total_frames - count * step))
+    end = begin + step * count
+    for i in range(begin, end, step):
+        librosa.display.specshow(log_chroma_spec[:, i : i + step])
+        plt.axis("off")
+        plt.savefig(
+            f"{CACHE_DIR}/chroma_{round(dur, 2)}_{i}.jpg",
+            bbox_inches="tight",
+            pad_inches=0.0,
+        )
+        plt.close()
 
 
 def embed_img(img_path, input_size=224):
@@ -142,9 +127,15 @@ def inference(mp3_path, log_name: str, folder_path=CACHE_DIR):
     if not mp3_path:
         return None, "Please input an audio!"
 
-    network = EvalNet(log_name)
     spec = log_name.split("_")[-1]
-    eval("mp3_to_%s" % spec)(mp3_path)
+    os.makedirs(folder_path, exist_ok=True)
+    try:
+        network = EvalNet(log_name)
+        eval("mp3_to_%s" % spec)(mp3_path)
+
+    except Exception as e:
+        print(f"Error converting {mp3_path} : {e}")
+
     outputs = []
     all_files = os.listdir(folder_path)
     for file_name in all_files: