recover

app.py

@@ -28,57 +28,52 @@ def clean_cache(cache_dir=CACHE_DIR):
 
 def get_audio_file_type(file_path: str):
     try:
-        # 获取媒体信息
         info = mediainfo(file_path)
-        # 返回文件格式
         return "." + info["format_name"]
 
     except Exception as e:
-        print(f"出错了: {e}")
+        print(f"Error occurred: {e}")
         return None
 
 
 def download_audio(url: str, save_path: str):
     with NamedTemporaryFile(delete=False, suffix="_temp") as tmp_file:
         temp_file_path = tmp_file.name
-        # 发送HTTP GET请求并下载内容
         response = requests.get(url, stream=True)
-        # 检查请求是否成功
         if response.status_code == 200:
-            # 将音频内容写入临时文件
             for chunk in response.iter_content(chunk_size=8192):
                 tmp_file.write(chunk)
 
         else:
-            print(f"下载文件失败: HTTP {response.status_code}")
+            print(f"Failed to download file: HTTP {response.status_code}")
             return ""
 
     ext = get_audio_file_type(temp_file_path)
     full_path = f"{save_path}{ext}"
-    # 重命名临时文件以包含正确的扩展名
     shutil.move(temp_file_path, full_path)
     return full_path
 
 
 def is_url(s: str):
     try:
-        # 解析字符串
         result = urlparse(s)
-        # 检查scheme（如http, https）和netloc（域名）
         return all([result.scheme, result.netloc])
 
     except:
-        # 如果解析过程中发生异常，则返回False
         return False
 
 
 def audio2midi(audio_path: str):
+    # Load audio
     audio, _ = load_audio(audio_path, sr=sample_rate, mono=True)
+    # Transcriptor
     transcriptor = PianoTranscription(
         device="cuda" if torch.cuda.is_available() else "cpu",
         checkpoint_path=WEIGHTS_PATH,
     )
+    # device: 'cuda' | 'cpu' Transcribe and write out to MIDI file
     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()
 
@@ -110,11 +105,9 @@ def music163_song_info(id: str):
     detail_api = "https://music.163.com/api/v3/song/detail"
     parm_dict = {"id": id, "c": str([{"id": id}]), "csrf_token": ""}
     free = False
-    song_name = "获取歌曲失败"
+    song_name = "Failed to get the song"
     response = requests.get(detail_api, params=parm_dict)
-    # 检查请求是否成功
     if response.status_code == 200:
-        # 处理成功响应
         data = json.loads(response.text)
         if data and "songs" in data and data["songs"]:
             fee = int(data["songs"][0]["fee"])
@@ -122,10 +115,10 @@ def music163_song_info(id: str):
             song_name = str(data["songs"][0]["name"])
 
         else:
-            song_name = "歌曲不存在"
+            song_name = "The song does not exist"
 
     else:
-        raise ConnectionError(f"错误: {response.status_code}, {response.text}")
+        raise ConnectionError(f"Error: {response.status_code}, {response.text}")
 
     return song_name, free
 
@@ -141,16 +134,16 @@ def url_infer(song: str):
                 song = f"https://music.163.com/song/media/outer/url?id={song_id}.mp3"
                 song_name, free = music163_song_info(song_id)
                 if not free:
-                    raise AttributeError("付费歌曲无法解析")
+                    raise AttributeError("Unable to parse VIP songs")
 
-                download_path = download_audio(song, download_path)
+            download_path = download_audio(song, download_path)
 
         elif song.isdigit():
             song_id = song
             song = f"https://music.163.com/song/media/outer/url?id={song_id}.mp3"
             song_name, free = music163_song_info(song_id)
             if not free:
-                raise AttributeError("付费歌曲无法解析")
+                raise AttributeError("Unable to parse VIP songs")
 
             download_path = download_audio(song, download_path)
 
@@ -170,42 +163,42 @@ def url_infer(song: str):
 
 if __name__ == "__main__":
     with gr.Blocks() as iface:
-        with gr.Tab("上传模式"):
+        with gr.Tab("Uploading Mode"):
             gr.Interface(
                 fn=upl_infer,
                 inputs=gr.Audio(
-                    label="上传音频",
+                    label="Upload an audio",
                     type="filepath",
                 ),
                 outputs=[
-                    gr.File(label="下载 MIDI"),
-                    gr.File(label="下载 PDF 乐谱"),
-                    gr.File(label="下载 MusicXML"),
-                    gr.File(label="下载 MXL"),
-                    gr.Textbox(label="ABC 记谱", show_copy_button=True),
-                    gr.Image(label="五线谱", type="filepath"),
+                    gr.File(label="Download MIDI"),
+                    gr.File(label="Download PDF score"),
+                    gr.File(label="Download MusicXML"),
+                    gr.File(label="Download MXL"),
+                    gr.Textbox(label="ABC notation", show_copy_button=True),
+                    gr.Image(label="Staff", type="filepath"),
                 ],
-                title="请上传音频 100% 后再点提交",
+                title="Please make sure the audio is completely uploaded before clicking Submit",
                 flagging_mode="never",
             )
 
-        with gr.Tab("直链模式"):
+        with gr.Tab("Direct Link Mode"):
             gr.Interface(
                 fn=url_infer,
                 inputs=gr.Textbox(
-                    label="输入音频 URL 直链",
+                    label="Input audio direct link",
                     placeholder="https://music.163.com/#/song?id=",
                 ),
                 outputs=[
-                    gr.Audio(label="下载音频", type="filepath"),
-                    gr.File(label="下载 MIDI"),
-                    gr.File(label="下载 PDF 乐谱"),
-                    gr.File(label="下载 MusicXML"),
-                    gr.File(label="下载 MXL"),
-                    gr.Textbox(label="ABC 记谱", show_copy_button=True),
-                    gr.Image(label="五线谱", type="filepath"),
+                    gr.Audio(label="Download audio", type="filepath"),
+                    gr.File(label="Download MIDI"),
+                    gr.File(label="Download PDF score"),
+                    gr.File(label="Download MusicXML"),
+                    gr.File(label="Download MXL"),
+                    gr.Textbox(label="ABC notation", show_copy_button=True),
+                    gr.Image(label="Staff", type="filepath"),
                 ],
-                title="网易云音乐可直接输入非 VIP 歌曲页面链接自动解析",
+                title="For Netease Cloud music, you can directly input the non-VIP song page link",
                 examples=["1945798894", "1945798973", "1946098771"],
                 flagging_mode="never",
                 cache_examples=False,