not use pdf2image

app.py

@@ -172,7 +172,6 @@ def generate_abc(args, region):
     create_dir('./tmp')
     timestamp = time.strftime("%a_%d_%b_%Y_%H_%M_%S", time.localtime())
     out_midi = abc_to_midi(tunes, f'./tmp/[{region}]{timestamp}.mid')
-    add_path()
     png_file = midi2png(out_midi)
     wav_file = midi2wav(out_midi)
 

render.py

@@ -1,30 +1,10 @@
 import os
 import sys
+import fitz
 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'
@@ -37,6 +17,7 @@ if sys.platform.startswith('linux'):
         subprocess.run([f'./{apkname}', '--appimage-extract'])
 
     mscore = f'./{extra_dir}/AppRun'
+    os.environ['QT_QPA_PLATFORM'] = 'offscreen'
 
 else:
     mscore = "D:/Program Files/MuseScore 3/bin/MuseScore3.exe"
@@ -52,16 +33,32 @@ def midi2wav(mid_file: str):
 
 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))
-    )
+    doc = fitz.open(pdf_path)
+
+    # 创建一个图像列表
+    images = []
+
+    for page_number in range(doc.page_count):
+        page = doc[page_number]
+        # 将页面渲染为图像
+        image = page.get_pixmap()
+        # 将图像添加到列表
+        images.append(Image.frombytes(
+            "RGB", [image.width, image.height], image.samples))
+
+    # 竖向合并图像
+    merged_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))
+        merged_image.paste(image, (0, y_offset))
         y_offset += image.height
 
-    combined_image.save(output_path)
+    # 保存合并后的图像为JPG
+    merged_image.save(output_path, "JPEG")
+
+    # 关闭PDF文档
+    doc.close()
     return output_path
 
 

requirements.txt

@@ -5,5 +5,5 @@ music21
 autopep8
 pillow==9.4.0
 gradio
-pdf2image
+pymupdf
 torch