Update app.py

app.py

@@ -3,7 +3,7 @@ from gradio_client import Client
 import json
 import logging
 import ast
-import openai  # OpenAI 라이브러리 추가
+import openai
 import os
 import random
 import re
@@ -91,18 +91,7 @@ def call_api(prompt, max_tokens, temperature, top_p):
 
 # 텍스트 요약 함수
 def summarize_text(text):
-    prompt = f"""당신은 YouTube 비디오 스크립트를 요약하는 AI 어시스턴트입니다. 
-아래 제공된 스크립트를 간결하면서도 포괄적으로 요약해주세요.
-비디오의 주요 주제, 핵심 포인트, 전반적인 메시지에 초점을 맞추세요. 
-요약은 잘 구조화되고 이해하기 쉬워야 하며, 내용의 본질을 포착해야 합니다.
-반드시 한국어로 요약을 제공하세요.
-다음 구조로 요약을 작성해주세요:
-1. 비디오의 주요 주제 또는 테마
-2. 제시된 주요 포인트 또는 논점
-3. 중요한 결론 또는 시사점
-요약할 스크립트:
-{text}
-위 스크립트에 대한 요약을 제공해주세요. 요약은 간결하면서도 정보가 풍부해야 하며, 비디오 내용의 핵심을 포착해야 합니다."""
+    prompt = text  # 프롬프트를 원본 텍스트로 설정하여 self-discover 가능하도록 함
 
     try:
         return call_api(prompt, max_tokens=2000, temperature=0.3, top_p=0.9)
@@ -112,7 +101,7 @@ def summarize_text(text):
 
 # Gradio 인터페이스 설정
 with gr.Blocks() as demo:
-    gr.Markdown("## YouTube Script Extractor with Summary")
+    gr.Markdown("## YouTube 스크립트 추출 및 요약 도구")
 
     youtube_url_input = gr.Textbox(label="YouTube URL 입력")
     analyze_button = gr.Button("분석하기")
@@ -135,13 +124,19 @@ with gr.Blocks() as demo:
         script_html = f"""<h2 style='font-size:24px;'>{title}</h2>
         <details>
             <summary><h3>원문 스크립트 (클릭하여 펼치기)</h3></summary>
-            <pre>{formatted_script}</pre>
+            <div style="white-space: pre-wrap;">{formatted_script}</div>
         </details>"""
         return script_html
 
     def generate_summary(script):
         summary = summarize_text(script)
-        summary_html = f"<h3>요약:</h3>\n<pre>{summary}</pre>"
+        # 요약 결과를 잘 표시하기 위해 div 태그와 CSS 스타일 적용
+        summary_html = f"""
+        <h3>요약:</h3>
+        <div style="white-space: pre-wrap; max-height: 400px; overflow-y: auto; border: 1px solid #ccc; padding: 10px;">
+            {summary}
+        </div>
+        """
         return summary_html
 
     def analyze(url, cache):
@@ -166,4 +161,4 @@ with gr.Blocks() as demo:
     )
 
 # 인터페이스 실행
-demo.launch(share=True)
+demo.launch(share=True)