修改部署方式

README.md

@@ -3,11 +3,10 @@ title: Certificate Master
 emoji: 🎓
 colorFrom: indigo
 colorTo: blue
-sdk: docker
+sdk: gradio
 pinned: false
 license: mit
-short_description: 批量证书生成大师 - 可视化拖拽设计
-app_port: 7860
+short_description: 批量证书生成大师 - Gradio 版本
 ---
 
 # 🎓 证书批量生成大师 (Certificate Master)

app.py

@@ -3,12 +3,10 @@ import json
 import io
 import zipfile
 import csv
-from flask import Flask, render_template, request, send_file, jsonify
-from PIL import Image, ImageDraw, ImageFont
+import tempfile
 
-app = Flask(__name__)
-app.config["MAX_CONTENT_LENGTH"] = 32 * 1024 * 1024
-app.config["JSON_AS_ASCII"] = False
+import gradio as gr
+from PIL import Image, ImageDraw, ImageFont
 
 DEFAULT_FONTS = [
     "/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc",
@@ -32,16 +30,18 @@ def sanitize_filename(name):
     safe = "".join([c for c in str(name) if c.isalnum() or c in (" ", "-", "_")]).strip()
     return safe or "certificate"
 
-def parse_rows(data_str, data_file):
-    if data_file:
-        content = data_file.read().decode("utf-8", errors="ignore")
+def parse_rows_from_inputs(data_str, csv_bytes):
+    if csv_bytes:
+        content = csv_bytes.decode("utf-8", errors="ignore")
         reader = csv.DictReader(io.StringIO(content))
         return [row for row in reader]
-    try:
-        data = json.loads(data_str or "[]")
-        return data if isinstance(data, list) else []
-    except json.JSONDecodeError:
-        return []
+    if data_str:
+        try:
+            data = json.loads(data_str or "[]")
+            return data if isinstance(data, list) else []
+        except json.JSONDecodeError:
+            return []
+    return []
 
 def wrap_text(text, font, max_width, draw):
     if not max_width or max_width <= 0:
@@ -60,121 +60,237 @@ def wrap_text(text, font, max_width, draw):
         lines.append(current)
     return "\n".join(lines)
 
-@app.route("/")
-def index():
-    return render_template("index.html")
+def generate_certificates_gradio(
+    bg_image,
+    font_bytes,
+    data_csv_bytes,
+    data_json_str,
+    config_str,
+    filename_tpl,
+    export_scale,
+    logo_image,
+    logo_x,
+    logo_y,
+    logo_scale,
+):
+    if bg_image is None:
+        raise gr.Error("请先上传背景图片")
 
-@app.route("/api/health")
-def health():
     try:
-        return jsonify({"status": "ok"}), 200
-    except Exception as e:
-        return jsonify({"status": "error", "message": str(e)}), 500
+        config = json.loads(config_str or "[]")
+    except json.JSONDecodeError:
+        raise gr.Error("字段配置 JSON 无效，请检查格式")
+
+    data_list = parse_rows_from_inputs(data_json_str, data_csv_bytes)
+    if not data_list:
+        raise gr.Error("没有有效的数据行，请检查 CSV 或 JSON 输入")
 
-@app.route("/api/generate", methods=["POST"])
-def generate_certificates():
     try:
-        if "background" not in request.files:
-            return jsonify({"error": "No background image provided"}), 400
-        bg_file = request.files["background"]
-        font_file = request.files.get("font")
-        data_file = request.files.get("data_file")
-        logo_file = request.files.get("logo")
-
-        config_str = request.form.get("config", "[]")
-        data_str = request.form.get("data", "[]")
-        filename_tpl = request.form.get("filename_template", "{index}.png")
-        export_scale = float(request.form.get("export_scale", "1"))
-        if export_scale <= 0:
-            export_scale = 1
-        logo_x = int(float(request.form.get("logo_x", "0")))
-        logo_y = int(float(request.form.get("logo_y", "0")))
-        logo_scale = float(request.form.get("logo_scale", "0"))
+        export_scale_val = float(export_scale or 1)
+    except ValueError:
+        export_scale_val = 1
+    if export_scale_val <= 0:
+        export_scale_val = 1
 
-        try:
-            config = json.loads(config_str)
-        except json.JSONDecodeError:
-            return jsonify({"error": "Invalid JSON in config"}), 400
-
-        data_list = parse_rows(data_str, data_file)
-        if not data_list:
-            return jsonify({"error": "No data rows provided"}), 400
-
-        zip_buffer = io.BytesIO()
-        with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zip_file:
-            base_image = Image.open(bg_file).convert("RGB")
-            if export_scale != 1:
-                w, h = base_image.size
-                base_image = base_image.resize((int(w * export_scale), int(h * export_scale)), Image.LANCZOS)
-
-            font_bytes = None
-            if font_file:
-                font_bytes = io.BytesIO(font_file.read())
-
-            for idx, row in enumerate(data_list):
-                img = base_image.copy()
-                draw = ImageDraw.Draw(img)
-                for field in config:
-                    text_tpl = field.get("template", "")
-                    x = int(field.get("x", 0)) * export_scale
-                    y = int(field.get("y", 0)) * export_scale
-                    size = int(field.get("size", 30)) * export_scale
-                    color = field.get("color", "#000000")
-                    align = field.get("align", "left")
-                    stroke_width = int(field.get("stroke_width", 0))
-                    stroke_color = field.get("stroke_color", None)
-                    shadow_dx = int(field.get("shadow_dx", 0))
-                    shadow_dy = int(field.get("shadow_dy", 0))
-                    shadow_color = field.get("shadow_color", None)
-                    max_width = int(field.get("max_width", 0)) * export_scale
-
-                    try:
-                        text = text_tpl.format(**row)
-                    except Exception:
-                        text = text_tpl
-
-                    if font_bytes:
-                        font_bytes.seek(0)
-                        font = ImageFont.truetype(font_bytes, int(size))
-                    else:
-                        font = load_font(None, int(size))
-
-                    anchor = "la"
-                    if align == "center":
-                        anchor = "ma"
-                    elif align == "right":
-                        anchor = "ra"
-
-                    text_wrapped = wrap_text(text, font, max_width, draw)
-                    if shadow_color and (shadow_dx or shadow_dy):
-                        draw.text((x + shadow_dx, y + shadow_dy), text_wrapped, fill=shadow_color, font=font, anchor=anchor,
-                                  stroke_width=stroke_width if stroke_color else 0, stroke_fill=stroke_color or None)
-                    draw.text((x, y), text_wrapped, fill=color, font=font, anchor=anchor,
-                              stroke_width=stroke_width if stroke_color else 0, stroke_fill=stroke_color or None)
-
-                if logo_file and logo_scale > 0:
-                    logo_file.stream.seek(0)
-                    logo = Image.open(logo_file).convert("RGBA")
-                    lw, lh = logo.size
-                    target_w = max(1, int(lw * logo_scale))
-                    target_h = max(1, int(lh * logo_scale))
-                    logo_resized = logo.resize((target_w, target_h), Image.LANCZOS)
-                    img.paste(logo_resized, (int(logo_x * export_scale), int(logo_y * export_scale)), logo_resized)
-
-                out = io.BytesIO()
-                img.save(out, format="PNG")
+    try:
+        logo_x_val = int(logo_x or 0)
+        logo_y_val = int(logo_y or 0)
+        logo_scale_val = float(logo_scale or 0)
+    except ValueError:
+        logo_x_val = 0
+        logo_y_val = 0
+        logo_scale_val = 0
+
+    zip_buffer = io.BytesIO()
+    with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zip_file:
+        base_image = bg_image.convert("RGB")
+        if export_scale_val != 1:
+            w, h = base_image.size
+            base_image = base_image.resize(
+                (int(w * export_scale_val), int(h * export_scale_val)),
+                Image.LANCZOS,
+            )
+
+        font_stream = None
+        if font_bytes:
+            font_stream = io.BytesIO(font_bytes)
+
+        for idx, row in enumerate(data_list):
+            img = base_image.copy()
+            draw = ImageDraw.Draw(img)
+            for field in config:
+                text_tpl = field.get("template", "")
+                x = int(field.get("x", 0)) * export_scale_val
+                y = int(field.get("y", 0)) * export_scale_val
+                size = int(field.get("size", 30)) * export_scale_val
+                color = field.get("color", "#000000")
+                align = field.get("align", "left")
+                stroke_width = int(field.get("stroke_width", 0))
+                stroke_color = field.get("stroke_color", None)
+                shadow_dx = int(field.get("shadow_dx", 0))
+                shadow_dy = int(field.get("shadow_dy", 0))
+                shadow_color = field.get("shadow_color", None)
+                max_width = int(field.get("max_width", 0)) * export_scale_val
 
                 try:
-                    name = filename_tpl.format(index=idx + 1, **row)
+                    text = text_tpl.format(**row)
                 except Exception:
-                    name = f"certificate_{idx + 1}.png"
-                name = sanitize_filename(os.path.splitext(name)[0]) + ".png"
-                zip_file.writestr(name, out.getvalue())
+                    text = text_tpl
+
+                if font_stream:
+                    font_stream.seek(0)
+                    font = ImageFont.truetype(font_stream, int(size))
+                else:
+                    font = load_font(None, int(size))
+
+                anchor = "la"
+                if align == "center":
+                    anchor = "ma"
+                elif align == "right":
+                    anchor = "ra"
+
+                text_wrapped = wrap_text(text, font, max_width, draw)
+                if shadow_color and (shadow_dx or shadow_dy):
+                    draw.text(
+                        (x + shadow_dx, y + shadow_dy),
+                        text_wrapped,
+                        fill=shadow_color,
+                        font=font,
+                        anchor=anchor,
+                        stroke_width=stroke_width if stroke_color else 0,
+                        stroke_fill=stroke_color or None,
+                    )
+                draw.text(
+                    (x, y),
+                    text_wrapped,
+                    fill=color,
+                    font=font,
+                    anchor=anchor,
+                    stroke_width=stroke_width if stroke_color else 0,
+                    stroke_fill=stroke_color or None,
+                )
+
+            if logo_image is not None and logo_scale_val > 0:
+                logo = logo_image.convert("RGBA")
+                lw, lh = logo.size
+                target_w = max(1, int(lw * logo_scale_val))
+                target_h = max(1, int(lh * logo_scale_val))
+                logo_resized = logo.resize((target_w, target_h), Image.LANCZOS)
+                img.paste(
+                    logo_resized,
+                    (int(logo_x_val * export_scale_val), int(logo_y_val * export_scale_val)),
+                    logo_resized,
+                )
+
+            out = io.BytesIO()
+            img.save(out, format="PNG")
+
+            try:
+                name = (filename_tpl or "{index}.png").format(index=idx + 1, **row)
+            except Exception:
+                name = f"certificate_{idx + 1}.png"
+            name = sanitize_filename(os.path.splitext(name)[0]) + ".png"
+            zip_file.writestr(name, out.getvalue())
+
+    zip_buffer.seek(0)
+    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".zip")
+    tmp.write(zip_buffer.getvalue())
+    tmp.flush()
+    tmp.close()
+    return tmp.name
+
+
+default_config = json.dumps(
+    [
+        {
+            "template": "{name}",
+            "x": 800,
+            "y": 600,
+            "size": 60,
+            "color": "#000000",
+            "align": "center",
+            "stroke_width": 0,
+            "stroke_color": "#000000",
+            "shadow_dx": 0,
+            "shadow_dy": 0,
+            "shadow_color": "#000000",
+            "max_width": 0,
+        }
+    ],
+    ensure_ascii=False,
+    indent=2,
+)
+
+
+with gr.Blocks(title="证书批量生成大师 | Certificate Master") as demo:
+    gr.Markdown("# 🎓 证书批量生成大师\n批量生成带个性化信息的证书图片，并打包为 ZIP 下载。")
+
+    with gr.Row():
+        with gr.Column():
+            bg_image = gr.Image(label="背景图片（必选）", type="pil")
+            font_file = gr.File(label="字体文件（可选，ttf/otf）", type="bytes")
+
+            with gr.Row():
+                data_csv = gr.File(label="学员数据 CSV（可选）", type="bytes")
+                data_json = gr.Textbox(
+                    label="或 JSON 数据（可选）",
+                    lines=4,
+                    placeholder='[{"name": "张三", "course": "Python 进阶", "date": "2024-01-01"}]',
+                )
+
+            config_box = gr.Textbox(
+                label="字段配置 JSON（高级）",
+                value=default_config,
+                lines=10,
+            )
+
+            filename_tpl = gr.Textbox(
+                label="文件名模板",
+                value="{name}.png",
+                placeholder="{name}.png 或 certificate_{index}.png",
+            )
+
+            export_scale = gr.Number(
+                label="导出缩放倍数",
+                value=1,
+            )
+
+            with gr.Row():
+                logo_image = gr.Image(label="徽章 / Logo（可选）", type="pil")
+            with gr.Row():
+                logo_x = gr.Number(label="Logo X 坐标", value=0)
+                logo_y = gr.Number(label="Logo Y 坐标", value=0)
+                logo_scale = gr.Number(label="Logo 缩放倍数", value=0.0)
+
+            generate_button = gr.Button("生成并下载 ZIP", variant="primary")
+
+        with gr.Column():
+            result_zip = gr.File(label="生成结果 ZIP")
+            gr.Markdown(
+                "提示：\n"
+                "- CSV 第一行应为表头，如 name,course,date\n"
+                "- 字段配置 JSON 中的 template 支持使用 {name}、{course} 等占位符\n"
+                "- 坐标和字号单位与背景图像像素一致"
+            )
+
+    generate_button.click(
+        fn=generate_certificates_gradio,
+        inputs=[
+            bg_image,
+            font_file,
+            data_csv,
+            data_json,
+            config_box,
+            filename_tpl,
+            export_scale,
+            logo_image,
+            logo_x,
+            logo_y,
+            logo_scale,
+        ],
+        outputs=result_zip,
+    )
 
-        zip_buffer.seek(0)
-        return send_file(zip_buffer, mimetype="application/zip", as_attachment=True, download_name="certificates.zip")
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=False)
+    demo.launch(server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -1,3 +1,2 @@
-Flask>=3.0.0
 Pillow>=10.0.0
-gunicorn>=21.2.0
+gradio>=4.0.0