Spaces:
Running
Running
File size: 2,134 Bytes
653dc95 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
import os
import sys
import subprocess
from PIL import Image
from pdf2image import convert_from_path
from utils import download
def add_path():
# """
# 将指定目录添加到 LD_LIBRARY_PATH 环境变量中，并在当前 Python 进程中生效。
# Parameters:
# - directory_path (str): 要添加的目录路径。
# """
# dir_path = os.path.join(os.getcwd(), 'lib')
# # 获取当前环境变量的值
# current_path = os.environ.get("LD_LIBRARY_PATH", "")
# # 将目录路径添加到 LD_LIBRARY_PATH 中
# new_path = f"{current_path}:{dir_path}"
# # 设置 LD_LIBRARY_PATH 环境变量，以便在当前 Python 进程中生效
# os.environ["LD_LIBRARY_PATH"] = new_path
os.environ['QT_QPA_PLATFORM'] = 'offscreen'
if sys.platform.startswith('linux'):
apkname = 'MuseScore.AppImage'
extra_dir = 'squashfs-root'
download(
filename=apkname,
url='https://cdn.jsdelivr.net/musescore/v4.2.0/MuseScore-4.2.0.233521125-x86_64.AppImage'
)
if not os.path.exists(extra_dir):
subprocess.run(['chmod', '+x', f'./{apkname}'])
subprocess.run([f'./{apkname}', '--appimage-extract'])
mscore = f'./{extra_dir}/AppRun'
else:
mscore = "D:/Program Files/MuseScore 3/bin/MuseScore3.exe"
def midi2wav(mid_file: str):
wav_file = mid_file.replace('.mid', '.wav')
command = [mscore, "-o", wav_file, mid_file]
result = subprocess.run(command)
print(result)
return wav_file
def pdf_to_img(pdf_path: str):
output_path = pdf_path.replace('.pdf', '.jpg')
images = convert_from_path(pdf_path)
combined_image = Image.new(
'RGB', (images[0].width, sum(image.height for image in images))
)
y_offset = 0
for image in images:
combined_image.paste(image, (0, y_offset))
y_offset += image.height
combined_image.save(output_path)
return output_path
def midi2png(mid_file: str):
pdf_score = mid_file.replace('.mid', '.pdf')
command = [mscore, "-o", pdf_score, mid_file]
result = subprocess.run(command)
print(result)
return pdf_to_img(pdf_score)
|