Update app.py

app.py

@@ -73,18 +73,29 @@ def summarize_text(text):
 8. 대본에서 전달하는 감정이나 분위기도 포함
 9. 반드시 기술적 용어나 전문 용어가 있을 경우, 이를 정확하게 사용
 10. 대본의 목적이나 의도를 파악하고, 이를 요약에 반드시 반영
+11. 각 문장을 명확하게 구분하고, 적절한 단락 구분을 사용하여 가독성을 높이시오
 
 대본:
 {text}
 """
     return call_api(prompt, max_tokens=2000, temperature=0.3, top_p=0.9)
 
-def display_script(title, script):
-    return f"""<h2 style='font-size:24px;'>{title}</h2>
-    <details>
-        <summary><h3>원문 스크립트 (클릭하여 펼치기)</h3></summary>
-        <div style="white-space: pre-wrap;">{script}</div>
-    </details>"""
+def display_content(title, script, summary):
+    return f"""
+    <h2 style='font-size:24px;'>{title}</h2>
+    <div style="display: flex; flex-direction: column; gap: 20px;">
+        <div style="flex: 1;">
+            <h3>요약</h3>
+            <div style="white-space: pre-wrap; background-color: #f0f0f0; padding: 15px; border-radius: 5px;">{summary}</div>
+        </div>
+        <div style="flex: 1;">
+            <details>
+                <summary><h3>원문 스크립트 (클릭하여 펼치기)</h3></summary>
+                <div style="white-space: pre-wrap; background-color: #e0e0e0; padding: 15px; border-radius: 5px;">{script}</div>
+            </details>
+        </div>
+    </div>
+    """
 
 def analyze(url, cache):
     try:
@@ -96,29 +107,28 @@ def analyze(url, cache):
             title, script = get_youtube_script(url)
             cache = {"url": url, "title": title, "script": script}
 
-        script_html = display_script(title, script)
-        summary_html = summarize_text(script)
+        summary = summarize_text(script)
+        content_html = display_content(title, script, summary)
         
         logging.info("분석 완료")
-        return script_html, summary_html, cache
+        return content_html, cache
     except Exception as e:
         error_msg = f"처리 중 오류 발생: {str(e)}"
         logging.exception(error_msg)
-        return error_msg, "", cache
+        return error_msg, cache
 
 # Gradio 인터페이스
 with gr.Blocks() as demo:
     gr.Markdown("## YouTube 스크립트 추출 및 요약 도구")
     youtube_url_input = gr.Textbox(label="YouTube URL 입력")
     analyze_button = gr.Button("분석하기")
-    script_output = gr.HTML(label="스크립트")
-    summary_output = gr.HTML(label="전체 요약")
+    content_output = gr.HTML(label="내용")
     cached_data = gr.State({"url": "", "title": "", "script": ""})
 
     analyze_button.click(
         analyze, 
         inputs=[youtube_url_input, cached_data], 
-        outputs=[script_output, summary_output, cached_data]
+        outputs=[content_output, cached_data]
     )
 
 if __name__ == "__main__":