Update app.py

app.py

@@ -117,36 +117,42 @@ def split_sentences(text):
 def display_script(title, script):
     script_sentences = split_sentences(script)
     formatted_script = "\n\n".join(script_sentences)
-    return f"# {title}\n\n{formatted_script}"
-
-def display_script(title, script):
-    script_sentences = split_sentences(script)
-    formatted_script = "\n\n".join(script_sentences)
-    return f"""<h2>{title}</h2>
+    return f"""<div style="background-color: #f0f0f0; padding: 20px; border-radius: 10px;">
+<h3>원문 스크립트</h3>
+<h2>{title}</h2>
 <details>
-    <summary>원문 스크립트 (클릭하여 펼치기)</summary>
-    <pre>{formatted_script}</pre>
-</details>"""
+    <summary>클릭하여 펼치기</summary>
+    <pre style="white-space: pre-wrap;">{formatted_script}</pre>
+</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>
+{summary}
+</div>"""
 
 def analyze(url):
     title, description, script = get_youtube_script(url)
     script_content = display_script(title, script)
+    yield script_content, gr.update(visible=True), gr.update(value="<div style='text-align: center; margin-top: 20px;'>요약 생성 중...</div>", visible=True)
     summary = summarize_text(title, description, script)
-    summary_content = f"<h2>{title}</h2>\n\n{summary}"
-    return script_content, summary_content
+    summary_content = display_summary(title, summary)
+    yield script_content, gr.update(visible=True), gr.update(value=summary_content, visible=True)
 
 # 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="요약")
+    script_output = gr.HTML(label="원문 스크립트", visible=False)
+    summary_output = gr.HTML(label="요약", visible=False)
 
     analyze_button.click(
         analyze,
         inputs=[youtube_url_input],
-        outputs=[script_output, summary_output]
+        outputs=[script_output, script_output, summary_output]
     )
 
 if __name__ == "__main__":