Update app.py

app.py

@@ -5,6 +5,7 @@ import logging
 import openai
 import os
 import re
+import html
 
 # 로깅 설정
 logging.basicConfig(filename='youtube_script_extractor.log', level=logging.DEBUG, 
@@ -86,12 +87,13 @@ def summarize_text(title, description, text):
   [예시]
 (변경전)
  - 유튜브를 처음 시작하는 사람들은 구독자 수와 조회수에 큰 관심을 두고 매일 유튜브 스튜디오를 확인하게 된다. 그러나 구독자가 100명, 1,000명에 도달하는 것만으로는 지속적인 성장에 도움이 되지 않는다. 구독자 수가 늘어난 후에도 유튜브 채널 운영에 대한 감을 잡지 못해 포기하는 경우가 많다.
-
 (변경후)
  - 유튜브를 처음 시작하는 사람들은 구독자 수와 조회수에 큰 관심을 두고 매일 유튜브 스튜디오를 확인하게 된다. 
  - 그러나 구독자가 100명, 1,000명에 도달하는 것만으로는 지속적인 성장에 도움이 되지 않는다. 
  - 구독자 수가 늘어난 후에도 유튜브 채널 운영에 대한 감을 잡지 못해 포기하는 경우가 많다.
-17. 반드시 **각 문장에서 핵심 부분은 [[중요 내용]] 형태로 표시하라**
+17. 반드시 **각 문장에서 핵심 부분은 <highlight>중요 내용</highlight> 형태로 표시하라**
+  [예시]
+ - 카메라는 [[2억 화소]]가 탑재될 가능성이 있으며, 가격은 [[약 280만원]]으로 예상된다.
 
 제목: {title}
 설명: {description}
@@ -153,27 +155,45 @@ def analyze(url):
     # 요약 생성
     summary = summarize_text(title, description, script)
     
-    # 마크다운 기호 제거 및 HTML 태그로 변환
+    # HTML 변환 및 이스케이프 처리
+    def escape_html_except_highlights(text):
+        parts = text.split('<highlight>')
+        escaped_parts = []
+        for i, part in enumerate(parts):
+            if i % 2 == 0:  # 하이라이트 태그 외부
+                escaped_parts.append(html.escape(part))
+            else:  # 하이라이트 태그 내부
+                highlight_parts = part.split('</highlight>')
+                if len(highlight_parts) > 1:
+                    escaped_parts.append(f'<span class="highlight">{html.escape(highlight_parts[0])}</span>{html.escape(highlight_parts[1])}')
+                else:
+                    escaped_parts.append(html.escape(part))
+        return ''.join(escaped_parts)
+    
     lines = summary.split('\n')
     formatted_lines = []
     for line in lines:
         if line.startswith('# '):
-            line = f"<h1>{line[2:]}</h1>"
+            line = f"<h1>{escape_html_except_highlights(line[2:])}</h1>"
         elif line.startswith('## '):
-            line = f"<h2>{line[3:]}</h2>"
+            line = f"<h2>{escape_html_except_highlights(line[3:])}</h2>"
         elif line.startswith('### '):
-            line = f"<h3>{line[4:]}</h3>"
+            line = f"<h3>{escape_html_except_highlights(line[4:])}</h3>"
         else:
-            # 핵심 부분 하이라이트 처리: [[중요 내용]]을 하이라이트된 HTML로 변환
-            line = line.replace('[[', '<span style="background-color: #FFDAB9; font-weight: bold;">').replace(']]', '</span>')
-            line = f"<p>{line}</p>"
+            line = f"<p>{escape_html_except_highlights(line)}</p>"
         formatted_lines.append(line)
     
     formatted_summary = '\n'.join(formatted_lines)
     
     summary_content = f"""<div style="background-color: #e6f3ff; padding: 20px; border-radius: 10px; margin-top: 20px;">
+    <style>
+        .highlight {{
+            background-color: #FFDAB9;
+            font-weight: bold;
+        }}
+    </style>
     <h3>요약</h3>
-    <h2>{title}</h2>
+    <h2>{html.escape(title)}</h2>
     {formatted_summary}
     </div>"""