Update app.py

app.py

@@ -8,6 +8,7 @@ from fpdf import FPDF
 from datetime import datetime
 from zoneinfo import ZoneInfo
 from sklearn.feature_extraction.text import CountVectorizer
+from weasyprint import HTML
 
 # OpenAI API 클라이언트 설정
 openai.api_key = os.getenv("OPENAI_API_KEY")
@@ -515,8 +516,13 @@ def save_to_pdf(blog_post, user_topic):
         current_tag = ''
         buffer = ''
         is_bold = False
+
+        # 스타일 속성 초기화
+        font_size = 11
+        color = (0, 0, 0)  # 기본 글자 색 (검정)
         
         for part in re.split(tag_pattern, blog_post):
+            # 태그 시작 부분
             if part in ['h1', 'h2', 'h3', 'p', 'strong', 'li', 'br']:
                 if buffer:
                     if current_tag in ['h1', 'h2', 'h3']:
@@ -524,30 +530,49 @@ def save_to_pdf(blog_post, user_topic):
                         pdf.multi_cell(0, 10, buffer.strip(), align='L')
                         pdf.ln(5)
                     elif current_tag == 'p':
-                        pdf.set_font("NanumGothic", '', 11)
+                        pdf.set_font("NanumGothic", '', font_size)
+                        pdf.set_text_color(*color)  # 설정된 색상 적용
                         pdf.multi_cell(0, 6, buffer.strip(), align='J')
                         pdf.ln(5)
                     elif current_tag == 'li':
-                        pdf.set_font("NanumGothic", '', 11)
+                        pdf.set_font("NanumGothic", '', font_size)
+                        pdf.set_text_color(*color)
                         pdf.multi_cell(0, 6, "• " + buffer.strip(), align='J')
                     elif current_tag == 'br':
                         pdf.ln(5)
                 buffer = ''
                 current_tag = part
+
+            # 스타일 속성 반영
+            elif 'style=' in part:
+                styles = re.findall(r'style="(.*?)"', part)
+                for style in styles:
+                    if 'color:' in style:
+                        color_match = re.search(r'color: (#\w+);', style)
+                        if color_match:
+                            color_hex = color_match.group(1)
+                            r, g, b = tuple(int(color_hex[i:i+2], 16) for i in (1, 3, 5))
+                            color = (r, g, b)  # RGB 값으로 변환하여 적용
+                    if 'font-size:' in style:
+                        size_match = re.search(r'font-size: (\d+)px;', style)
+                        if size_match:
+                            font_size = int(size_match.group(1)) * 0.75  # pt 단위로 변환 (1px = 0.75pt)
+
             elif part == 'strong':
                 is_bold = True
-                pdf.set_font("NanumGothic", 'B', 11)
+                pdf.set_font("NanumGothic", 'B', font_size)
             elif part == '/strong':
                 is_bold = False
-                pdf.set_font("NanumGothic", '', 11)
+                pdf.set_font("NanumGothic", '', font_size)
             elif part.startswith('/') or part == 'div':
                 continue
             elif not tag_pattern.match(part) and part.strip():
-                pdf.write(6, part.strip() + ' ')
-        
+                buffer += part.strip() + ' '
+
         # 마지막 버퍼 처리
         if buffer:
-            pdf.set_font("NanumGothic", '', 11)
+            pdf.set_font("NanumGothic", '', font_size)
+            pdf.set_text_color(*color)
             pdf.multi_cell(0, 6, buffer.strip(), align='J')
         
         # PDF 저장