Update app.py

app.py

@@ -2,7 +2,6 @@ import os
 import torch
 import shutil
 import gradio as gr
-import traceback
 from mutagen.mp3 import MP3
 from mutagen.flac import FLAC
 from piano_transcription_inference import PianoTranscription, load_audio, sample_rate
@@ -58,6 +57,7 @@ def extract_meta(audio_path: str):
         raise ValueError("文件路径为空!")
 
     artist = None
+    title = None
     name, ext = os.path.splitext(os.path.basename(audio_path))
     ext == ext.lower()
     if ext == ".mp3":
@@ -80,16 +80,14 @@ def extract_meta(audio_path: str):
 
     if not title:
         title = name.strip().capitalize()
+    if not artist:
+        artist = "Unknown artist"
 
     return title, artist
 
 
 def audio2midi(audio_path: str, cache_dir: str):
     title, artist = extract_meta(audio_path)
-    if not title:
-        title = "Unknown title"
-    if not artist:
-        artist = "Unknown artist"
     audio, _ = load_audio(audio_path, sr=sample_rate, mono=True)
     transcriptor = PianoTranscription(
         device="cuda" if torch.cuda.is_available() else "cpu",
@@ -112,8 +110,7 @@ def upl_infer(audio_path: str, cache_dir=f"{TMP_DIR}/cache"):
         pdf, jpg = xml2jpg(xml)
 
     except Exception as e:
-        tb = repr(traceback.format_exception(e))
-        status = f"{tb} (audio_path = {audio_path}, cache_dir={cache_dir})"
+        status = f"{e}"
 
     return status, midi, pdf, xml, mxl, abc, jpg