Update src/pyscripts/file_process.py

src/pyscripts/file_process.py

@@ -1,69 +1,63 @@
-from fastapi import FastAPI, Path, Query, Response, BackgroundTasks
-from fastapi.responses import FileResponse, HTMLResponse
-import gradio as gr
-from src.pyscripts import file_process, oss, img_process
-from datetime import datetime
-from collections import namedtuple
-from pydantic import BaseModel
-
-Person = namedtuple('Person', ['name', 'title', 'category', 'conference', 'respond_person'])
-
-CUSTOM_PATH = "/gradio"
-
-app = FastAPI()
-
-pdf_path = "src/assets/InvitationTemplate.pdf"
-output_path = 'output/output.pdf'
-person = Person(name="雷军", title="先生", category="观礼", conference="世界教育者大会", respond_person="Eazy")
-
-# data = {
-#     'name': '雷军',
-#     'title': '先生',
-#     "conference": "世界教育者大会"
-# }
-
-
-@app.get("/")
-def read_main():
-    return {"message": "This is your main app"}
-
-# @app.post("/getInvitationPDF/{name}/{title}/{conference}")
-@app.get("/getInvitationPDF")
-def getInvitationPDF(name, title, conference, category, respond_person):
-    person = Person(name=name, title=title, category=category, conference=conference, respond_person=respond_person)
-    formatted_date = datetime.now().strftime('%Y-%m-%d')
-    file_name = f"致{name+title}的WWEC2024邀请函.pdf"
-    file_dir = f"invitation/{category}"
-    object_name = file_dir + "/" + file_name
-    pdfStream = file_process.writeInvitationPDF(person)
-    # with open(f"output/{file_name}", 'wb') as f:
+import io
+import fitz  # PyMuPDF
+
+pdf_path_dict = {
+    "old": "src/assets/InvitationTemplateCompressed.pdf",
+    "common": "src/assets/WWEC嘉宾邀请函mini.pdf",
+    "观礼": "src/assets/WWEC列席嘉宾邀请函.pdf",
+    "演讲": "src/assets/WWEC演讲嘉宾邀请函.pdf",
+    "致辞": "src/assets/WWEC演讲嘉宾邀请函.pdf",
+}
+font_path = "src/assets/ZhouZiSongTi7000Zi-2.otf"
+font_path = "src/assets/FangZhengHeiTiJianTi-1.ttf"
+# font = fitz.Font(fontfile=font_path)
+
+def getInvitationPDF(data):
+    pdf_path = pdf_path_dict[data["category"]]
+    doc = fitz.open(pdf_path)
+    for page in doc:
+        fields = page.widgets()
+        for field in fields:
+            if field.field_name in data:
+                # field.font = font
+                field.field_value = data[field.field_name]
+                field.update()
+    pdfStream = io.BytesIO()
+    doc.save(pdfStream)
+    doc.close()
+    # with open("output/output.pdf", 'wb') as f:
     #     f.write(pdfStream.getvalue())
-    oss.upload_file_stream_to_s3(pdfStream, "host", object_name)
-    file_url = f"https://isidoresong.us.kg/{object_name}"
-    return {"url": file_url}
-
-class ExhibitorPosterInfo(BaseModel):
-    exhibitor: str
-    logo_link: str
-    slogan: str
-    content: str
-
-@app.post("/getExhibitorPoster")
-def getExhibitorPoster(item: ExhibitorPosterInfo):
-    logo_link = item.logo_link
-    exhibitor = item.exhibitor
-    slogan = item.slogan
-    content = item.content
-    img_stream = img_process.make_exhibitor_poster(logo_link, slogan, content)
-    file_dir = "ExhibitorPoster"
-    object_name = file_dir + "/" + exhibitor + ".jpg"
-    # with open(f"output/{exhibitor + '.jpg'}", 'wb') as f:
-    #     f.write(img_stream.getvalue())
-    oss.upload_file_stream_to_s3(img_stream, "host", object_name)
-    file_url = f"https://isidoresong.us.kg/{object_name}"
-    return {"url": file_url}
-
-interface = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
-app = gr.mount_gradio_app(app, interface, path=CUSTOM_PATH)
-
-# if __name__ == "__main__":
+    return pdfStream
+
+
+def writeInvitationPDF(person, debug=False):
+    font_size=30
+    pdf_path = pdf_path_dict[person.category]
+    doc = fitz.open(pdf_path)
+    page = doc.load_page(1)
+    page.insert_font(fontname="f", fontfile=font_path)
+    page.insert_text((145, 80),
+        person.name+person.title, fontname="f", fontsize=min(font_size, font_size*5/len(person.name+person.title)), color=(1, 1, 1))
+    if person.category == "演讲" or person.category == "致辞":
+        page.insert_text((80, 375),
+            person.conference, fontname="f", fontsize=min(font_size, font_size*14/len(person.conference)), color=(1, 1, 1))
+    pdfStream = io.BytesIO()
+    doc.save(pdfStream)
+    doc.close()
+    if debug:
+        with open("output/output.pdf", 'wb') as f:
+            f.write(pdfStream.getvalue())
+    return pdfStream
+
+
+if __name__ == "__main__":
+    from collections import namedtuple
+    Person = namedtuple('Person', ['name', 'title', 'category', 'conference', 'respond_person'])
+    person = Person(name="X专业测试XXX", title="先生", category="致辞", conference="8月21日中欧智慧论坛，做主题演讲", respond_person="Eazy")
+
+    output_path = 'output/output.pdf'
+
+    # getInvitationPDF(data)
+    writeInvitationPDF(person, debug=True)
+    # with open(output_path, 'wb') as f:
+    #     f.write(pdfStream.getvalue())