Upload 8 files

.gitattributes

@@ -26,3 +26,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zstandard filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+src/assets/InvitationTemplate.pdf filter=lfs diff=lfs merge=lfs -text
+src/assets/InvitationTemplateCompressed.pdf filter=lfs diff=lfs merge=lfs -text

src/assets/InvitationTemplate.pdf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f2f0463b3a776395d96a64df9a43d4bc8117acf3e213be0759c2d71ee7a0d33c
+size 5784474

src/assets/InvitationTemplateCompressed.pdf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c7c7edc1d7019e5b2540e3b2d53fee9c91386249ff86f48e883ebbf3aa9963e7
+size 5014682

src/pyscripts/__init__.py

@@ -0,0 +1,2 @@
+def hello() -> str:
+    return "Hello from pyscripts!"

src/pyscripts/backup.py

@@ -0,0 +1,42 @@
+import pdfrw
+import io
+import base64
+
+ANNOT_KEY = '/Annots'
+ANNOT_FIELD_KEY = '/T'
+ANNOT_RECT_KEY = '/Rect'
+SUBTYPE_KEY = '/Subtype'
+WIDGET_SUBTYPE_KEY = '/Widget'
+
+
+def write_fillable_pdf(input_pdf_path, output_pdf_path, data_dict):
+    """填写pdf模板并输出新的pdf"""
+    template_pdf = pdfrw.PdfReader(input_pdf_path)
+    pdf_pages = len(template_pdf.pages)
+    template_pdf.Root.AcroForm.update(pdfrw.PdfDict(NeedAppearances=pdfrw.PdfObject('true')))
+    for pagenum in range(pdf_pages):
+        annotations = template_pdf.pages[pagenum][ANNOT_KEY]
+        if annotations is None:
+            continue
+        for annotation in annotations:
+            if annotation[SUBTYPE_KEY] == WIDGET_SUBTYPE_KEY:
+                if annotation[ANNOT_FIELD_KEY]:
+                    key = annotation[ANNOT_FIELD_KEY][1:-1]
+                    if key in data_dict.keys():
+                        annotation.update(
+                            pdfrw.PdfDict(V='{}'.format(data_dict[key]))
+                        )
+    pdf_stream = io.BytesIO()
+    pdfrw.PdfWriter().write(pdf_stream, template_pdf)
+    base64.b64encode(pdf_stream.getvalue())
+
+data_dict = {
+   'name': '宋有哲',
+   "conference": "世界教育者大会"
+}
+
+INVOICE_TEMPLATE_PATH = "src/assets/GuestPage.pdf"
+INVOICE_OUTPUT_PATH = 'output/output.pdf'
+
+if __name__ == '__main__':
+    write_fillable_pdf(INVOICE_TEMPLATE_PATH, INVOICE_OUTPUT_PATH, data_dict)

src/pyscripts/file_process.py

@@ -0,0 +1,24 @@
+import io
+import fitz  # PyMuPDF
+
+pdf_path = "src/assets/InvitationTemplateCompressed.pdf"
+
+def getInvitationPDF(data):
+    doc = fitz.open(pdf_path)
+    for page in doc:
+        fields = page.widgets()
+        for field in fields:
+            if field.field_name in data:
+                field.field_value = data[field.field_name]
+                field.update()
+    pdfStream = io.BytesIO()
+    doc.save(pdfStream)
+    doc.close()
+    return pdfStream
+
+if __name__ == "__main__":
+    output_path = 'output/output.pdf'
+    data = {
+        'name': '宋',
+        "conference": "世界教育者大会"
+    }

src/pyscripts/oss.py

@@ -0,0 +1,67 @@
+import io
+import os
+import boto3
+from botocore.exceptions import NoCredentialsError
+
+ACCESS_KEY = os.environ.get('ACCESS_KEY')
+SECRET_KEY = os.environ.get('SECRET_KEY')
+ENDPOINT = os.environ.get("ENDPOINT")
+BUCKET_NAME = 'host'
+
+def create_s3_client(access_key, secret_key, endpoint_url):
+    try:
+        s3_client = boto3.client(
+            's3',
+            aws_access_key_id=access_key,
+            aws_secret_access_key=secret_key,
+            endpoint_url=endpoint_url
+        )
+        return s3_client
+    except NoCredentialsError:
+        print("Credentials not available")
+        return None
+
+# 上传文件到 S3（或兼容 S3 的服务，如 IDrive e2）
+def upload_file_stream_to_s3(file_stream, bucket_name, key):
+    s3_client = create_s3_client(ACCESS_KEY, SECRET_KEY, ENDPOINT)
+    file_stream.seek(0)
+    s3_client.upload_fileobj(
+        file_stream,
+        bucket_name,
+        key
+    )
+    # with open("output/output.pdf", 'wb') as f:
+    #     file_stream.seek(0)
+    #     f.write(file_stream.getvalue())
+    print(f"{key} has been uploaded to {bucket_name}")
+    return True
+
+def download_file_stream_from_s3(bucket_name, key):
+    fileobj = io.BytesIO()
+    s3_client = create_s3_client(ACCESS_KEY, SECRET_KEY, ENDPOINT)
+    s3_client.download_fileobj(
+        bucket_name,
+        key,
+        fileobj,
+    )
+    return fileobj
+
+
+# FILE_PATH = "src/assets/InvitationTemplate.pdf"
+# OBJECT_NAME = 'test.pdf'
+# ACCESS_KEY = os.environ.get('ACCESS_KEY')
+# SECRET_KEY = os.environ.get('SECRET_KEY')
+# ENDPOINT = os.environ.get("ENDPOINT")
+# BUCKET_NAME = 'host'
+# FILE_PATH = "src/assets/InvitationTemplate.pdf"
+# OBJECT_NAME = 'test.pdf'
+
+# # 创建 S3 客户端
+# s3_client = create_s3_client(ACCESS_KEY, SECRET_KEY, ENDPOINT)
+
+# # 上传文件
+# if s3_client:
+#     # upload_file_to_s3(s3_client, FILE_PATH, BUCKET_NAME, OBJECT_NAME)
+#     fileobj = download_file_stream_from_s3(s3_client, BUCKET_NAME, OBJECT_NAME)
+#     with open(OBJECT_NAME, 'wb') as f:
+#         f.write(fileobj.getvalue())

src/pyscripts/run.py

@@ -0,0 +1,38 @@
+from fastapi import FastAPI, Path, Query
+import gradio as gr
+from src.pyscripts import file_process, oss
+
+
+CUSTOM_PATH = "/gradio"
+
+app = FastAPI()
+
+pdf_path = "src/assets/InvitationTemplate.pdf"
+output_path = 'output/output.pdf'
+data = {
+   'name': '宋',
+   "conference": "世界教育者大会"
+}
+
+
+@app.get("/")
+def read_main():
+    return {"message": "This is your main app"}
+
+@app.post("/getInvitationPDF/{name}/{conference}")
+def getInvitationPDF(name, conference):
+    data = {
+        'name': name,
+        "conference": conference
+    }
+    object_name = f"invitation/{name}_{conference}.pdf"
+    pdfStream = file_process.getInvitationPDF(data)
+    oss.upload_file_stream_to_s3(pdfStream, "host", object_name)
+    file_url = f"https://pub-429c75a96a8f4597984dd7ebc525d652.r2.dev/{object_name}"
+    # with open("output/output.pdf", 'wb') as f:
+    #     f.write(pdfStream.getvalue())
+    return file_url
+
+
+interface = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
+app = gr.mount_gradio_app(app, interface, path=CUSTOM_PATH)

src/pyscripts/tempCodeRunnerFile.py

@@ -0,0 +1 @@
+output_path