Update app.py

app.py

@@ -40,12 +40,13 @@ def get_youtube_script(url):
         title = data.get("title", "제목 없음")
         description = data.get("description", "설명 없음")
         transcription_text = data.get("transcriptionAsText", "")
+        thumbnails = data.get("thumbnails", [])
 
         if not transcription_text:
             raise ValueError("추출된 스크립트가 없습니다.")
 
         logging.info("스크립트 추출 완료")
-        return title, description, transcription_text
+        return title, description, transcription_text, thumbnails
     except Exception as e:
         logging.exception("스크립트 추출 중 오류 발생")
         raise
@@ -69,7 +70,8 @@ def summarize_text(title, description, text):
 [유튜브 요약 규칙]
 1. 너는 유튜브 영상 전문 해설가로서 지침에 맞게 이 글을 작성하라
 2. 아래의 제목과 설명은 이 유튜브 영상의 원본 메타데이터이다.
-3. 반드시 제목과 설명으로 주제와 문맥을 먼저 파악하고, 아래의 대본을 반드시 지침에 맞게 상세하게 요약하라
+3. 반드시 제목과 설명으로 주제와 문맥, 철자(Spelling)을 먼저 파악하고, 아래의 대본을 반드시 지침에 맞게 상세하게 요약하라
+  - 반드시 주어진 제목, 설명에 있는 철자(Spelling)를 요약에 반영하라(원문 대본에는 오탈자가 있을 수 있다)
 4. 반드시 한글로 작성하라
 5. 반드시 '이 유튜브 대본은', '이 영상은', '이 유튜브는'등의 소개식 표현은 제외하라
 6. 요약문만으로도 영상을 직접 시청한 것과 동일한 수준으로 내용을 이해할 수 있도록 상세히 작성
@@ -80,7 +82,6 @@ def summarize_text(title, description, text):
 11. 등장인물, 장소, 사건 등 중요한 요소를 정확하게 작성
 12. 대본에서 전달하는 감정이나 분위기도 포함
 13. 반드시 기술적 용어나 전문 용어가 있을 경우, 이를 정확하게 사용
-
 14. 반드시 핵심 섹션(소주제)를 파악하여 섹션에 맞게 글을 요약하라(글의 양을 고려하여 섹션의 개수를 탄력적으로 설정)
 15. 각 섹션의 제목(소주제)에는 내용과 어울리는 적절한 이모지로 소주제를 시작하라
 16. 각 섹션의 내용은 Bullet Point를 사용하여 가독성을 높여라(문장 단위로 구분)
@@ -125,57 +126,72 @@ def split_sentences(text):
 def display_script(title, script):
     script_sentences = split_sentences(script)
     formatted_script = "\n\n".join(script_sentences)
-    return f"""<div style="background-color: #f0f0f0; padding: 20px; border-radius: 10px;">
-<h3>원문 스크립트</h3>
+    return f"""<div class="script-box">
 <details>
     <summary>클릭하여 펼치기</summary>
-    <h2>{title}</h2>
-    <pre style="white-space: pre-wrap;">{formatted_script}</pre>
+    <div class="output-title">{title}</div>
+    <p style="white-space: pre-wrap;">{formatted_script}</p>
 </details>
 </div>"""
 
 def display_summary(title, summary):
-    return f"""<div style="background-color: #e6f3ff; padding: 20px; border-radius: 10px; margin-top: 20px;">
-<h3>요약</h3>
-<h2>{title}</h2>
+    return f"""<div class="script-box">
+<div class="output-title">{title}</div>
 {summary}
 </div>"""
 
+def get_thumbnail_url(thumbnails):
+    for thumbnail in thumbnails:
+        if thumbnail.get("width") == 640 and thumbnail.get("height") == 480:
+            return thumbnail.get("url")
+    return "640x480 크기의 썸네일을 찾을 수 없습니다."
+    
 def analyze(url):
     # 스크립트 추출
-    yield "스크립트 추출 중...", "스크립트 추출 중..."
-    title, description, script = get_youtube_script(url)
+    yield "스크립트 추출 중...", "스크립트 추출 중...", ""
+    title, description, script, thumbnails = get_youtube_script(url)
     script_content = display_script(title, script)
+    thumbnail_url = get_thumbnail_url(thumbnails)
     
     # 원문 스크립트 표시 및 요약 시작
-    yield script_content, "요약 생성 중..."
+    yield script_content, "요약 생성 중...", thumbnail_url
     
     # 요약 생성
     summary = summarize_text(title, description, script)
     
+    # HTML로 변환 (convert_to_html 로직을 직접 통합)
     lines = summary.split('\n')
     formatted_lines = []
     for line in lines:
-        if line.startswith('# '):
-            line = f"<h1>{html.escape(line[2:])}</h1>"
-        elif line.startswith('## '):
-            line = f"<h2>{html.escape(line[3:])}</h2>"
-        elif line.startswith('### '):
-            line = f"<h3>{html.escape(line[4:])}</h3>"
-        else:
-            line = f"<p>{html.escape(line)}</p>"
-        formatted_lines.append(line)
+        line = line.strip()
+        if line.startswith('####'):
+            formatted_lines.append(f"<h4>{html.escape(line[4:].strip())}</h4>")
+        elif line.startswith('###'):
+            formatted_lines.append(f"<h3>{html.escape(line[3:].strip())}</h3>")
+        elif line.startswith('##'):
+            formatted_lines.append(f"<h2>{html.escape(line[2:].strip())}</h2>")
+        elif line.startswith('#'):
+            formatted_lines.append(f"<h1>{html.escape(line[1:].strip())}</h1>")
+        elif line.startswith('- '):  # 리스트 아이템
+            content = html.escape(line[2:])
+            bold_content = re.sub(r'\*\*(.*?)\*\*', r'<strong>\1</strong>', content)
+            formatted_lines.append(f"<li>{bold_content}</li>")
+        elif line:  # 일반 텍스트 (빈 줄 제외)
+            content = html.escape(line)
+            bold_content = re.sub(r'\*\*(.*?)\*\*', r'<strong>\1</strong>', content)
+            formatted_lines.append(f"<p>{bold_content}</p>")
+        else:  # 빈 줄
+            formatted_lines.append("<br>")
     
     formatted_summary = '\n'.join(formatted_lines)
     
-    summary_content = f"""<div style="background-color: #e6f3ff; padding: 20px; border-radius: 10px; margin-top: 20px;">
-    <h3>요약</h3>
-    <h2>{html.escape(title)}</h2>
+    summary_content = f"""<div class="script-box">
+    <div class="output-title">{html.escape(title)}</div>
     {formatted_summary}
     </div>"""
     
     # 최종 결과 표시
-    yield script_content, summary_content
+    yield script_content, summary_content, thumbnail_url
 
 # Gradio 인터페이스
 with gr.Blocks() as demo:
@@ -184,11 +200,12 @@ with gr.Blocks() as demo:
     analyze_button = gr.Button("분석하기")
     script_output = gr.HTML(label="원문 스크립트")
     summary_output = gr.HTML(label="요약")
+    thumbnail_output = gr.Textbox(label="썸네일 URL (640x480)")  # 이 줄 추가
 
     analyze_button.click(
         analyze,
         inputs=[youtube_url_input],
-        outputs=[script_output, summary_output]
+        outputs=[script_output, summary_output, thumbnail_output]  # thumbnail_output 추가
     )
 
 if __name__ == "__main__":