Update app.py

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
 import json
-import os
-import docx
-from docx.oxml.ns import qn
-from docx import Document
-from docx.shared import Inches, Pt, Cm, Mm, RGBColor
-from docx.enum.table import WD_TABLE_ALIGNMENT
 import pandas as pd
+from reportlab.lib import colors
+from reportlab.pdfbase import pdfmetrics
+from reportlab.pdfbase.ttfonts import TTFont
+from reportlab.platypus import Table, TableStyle, Image, SimpleDocTemplate, PageBreak
+from reportlab.lib import colors
+from reportlab.lib.pagesizes import A4
+from reportlab.lib.units import cm
 
 with open('pokemon.json', 'r') as f:
     pokemons = json.load(f)
@@ -27,27 +28,30 @@ generation = gr.Dropdown(
             [f"{k}세대" for k in range(1, 10)], value="1세대", label="포켓몬 세대", info="원하는 포켓몬 세대를 선택하세요."
         )
 
-download = gr.File(label="Download a file")
-text = gr.DataFrame()
-
-def write_docx(gen, progress=gr.Progress()):
-    filename = f'포켓몬{gen}.docx'
-
-    document = Document()
-    section = document.sections[0]
-    section.page_height = Mm(297)
-    section.page_width = Mm(210)
-    #changing the page margins
-    margin = 1.27
-    sections = document.sections
-    for section in sections:
-        section.top_margin = Cm(margin)
-        section.bottom_margin = Cm(margin)
-        section.left_margin = Cm(margin)
-        section.right_margin = Cm(margin)
-    document.styles['Normal'].font.name = 'NanumSquareRound'
-    document.styles['Normal']._element.rPr.rFonts.set(qn('w:eastAsia'), 'NanumSquareRound')
-
+download = gr.File(label="파란 글씨를 눌러서 다운로드 받으세요.")
+text = gr.DataFrame(label="포켓몬 리스트")
+
+def write_pdf(gen, progress=gr.Progress()):
+    filename = f'포켓몬{gen}.pdf'
+    pdfmetrics.registerFont(TTFont("나눔고딕", "NanumGothic.ttf"))
+    # PDF 문서를 생성합니다.
+    pdfmetrics.registerFont(TTFont("나눔고딕", "NanumGothic.ttf"))
+    doc = SimpleDocTemplate(filename, pagesize=A4)
+    
+    # 테이블 스타일을 정의합니다.
+    style = TableStyle([
+        ('TEXTCOLOR', (0, 1), (-1, -1), colors.gray),
+        ('ALIGN', (0, 0), (-1, -1), 'CENTER'),
+        ('FONTNAME', (0, 0), (-1, -1), '나눔고딕'),
+        ('SIZE', (0, 0), (-1, -1), 50),
+        ('BACKGROUND', (0, 1), (-1, 1), colors.white),
+        ('GRID', (0, 0), (-1, -1), 1, colors.black),
+        ('VALIGN', (0, 0), (-1, -1), 'MIDDLE'),
+        ('LEADING', (0, 0), (-1, -1), 14),  # 라인 간격 조정
+        ('BOTTOMPADDING', (0, 0), (-1, -1), 60),
+    ])
+
+    story = []
     data_dict = []
     start, end = GEN_RANGE[gen]
     for k in progress.tqdm(range(start, end+1)):
@@ -60,42 +64,26 @@ def write_docx(gen, progress=gr.Progress()):
             dict(이름=name, No=number, 타입='+'.join(types))
         )
 
-        df = pd.DataFrame(data_dict)
-        # Document
-        table = document.add_table(rows=4, cols=1)
-        table.alignment = WD_TABLE_ALIGNMENT.CENTER
-        table.style = 'Table Grid'
-
-        hdr_cells = table.rows[0].cells
-        hdr_cells[0].text = f"{number}"
-        hdr_cells[0].paragraphs[0].runs[0].font.size = Pt(50)
-        hdr_cells[0].paragraphs[0].alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.CENTER
-
-        hdr_cells = table.rows[1].cells
-        p = hdr_cells[0].add_paragraph()
-        p.alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.CENTER
-        r = p.add_run()
-        r.add_picture(image_path, width=Cm(14.5), height=Cm(14.5))
-        r.add_break(docx.enum.text.WD_BREAK.LINE)
-
-        hdr_cells = table.rows[3].cells
-        hdr_cells[0].text = f"{name}"
-        hdr_cells[0].paragraphs[0].runs[0].font.size = Pt(70)
-        hdr_cells[0].paragraphs[0].runs[0].font.color.rgb = RGBColor(192, 192, 192)
-        hdr_cells[0].paragraphs[0].alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.CENTER
-
-        hdr_cells = table.rows[2].cells
-        hdr_cells[0].text = f"{'+'.join(types)}"
-        hdr_cells[0].paragraphs[0].runs[0].font.size = Pt(70)
-        hdr_cells[0].paragraphs[0].runs[0].font.color.rgb = RGBColor(192, 192, 192)
-        hdr_cells[0].paragraphs[0].alignment = docx.enum.text.WD_ALIGN_PARAGRAPH.CENTER
-
-        document.add_page_break()
-
-    document.save(filename)
+        # PDF에 이미지를 추가합니다. 이미지 파일의 경로를 지정하세요.
+        image = Image(image_path, width=13.5*cm, height=13.5*cm)
+
+        # 테이블 데이터를 준비합니다.
+        data = [
+            [number],
+            [image],  # 2번째 행에 이미지를 추가할 것입니다.
+            ['+'.join(types)],
+            [name],
+        ]
+        # 테이블 객체를 생성합니다.
+        table = Table(data)
+        table.setStyle(style)
+        story.append(table)
+
+    doc.build(story)
+    df = pd.DataFrame(data_dict)
     return df[['No', '이름', '타입']], filename
 
-demo = gr.Interface(write_docx, generation, [text, download], title="대치동 포켓몬 도감 생성기",
+demo = gr.Interface(write_pdf, generation, [text, download], title="대치동 포켓몬 도감 생성기",
                     description="원하는 포켓몬 세대를 선택하고, 다운로드를 눌러주세요.")
 demo.queue(concurrency_count=3)
 demo.launch()