Update app.py

app.py

@@ -130,24 +130,24 @@ 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>
-<p>{summary}</p>
+{summary}
 </div>"""
 
 def analyze(url):
     # 스크립트 추출
-    yield gr.update(value="<div>스크립트 추출 중...</div>", visible=True), gr.update(visible=False)
+    yield "스크립트 추출 중...", gr.update(visible=True), gr.update(visible=False)
     title, description, script = get_youtube_script(url)
     script_content = display_script(title, script)
     
     # 원문 스크립트 표시 및 요약 시작
-    yield gr.update(value=script_content, visible=True), gr.update(value="<div>요약 생성 중...</div>", visible=True)
+    yield script_content, gr.update(visible=True), "요약 생성 중..."
     
     # 요약 생성
     summary = summarize_text(title, description, script)
-    summary_content = f"# {title}\n\n{summary}"
+    summary_content = display_summary(title, summary)
     
     # 최종 결과 표시
-    yield gr.update(value=script_content, visible=True), gr.update(value=summary_content, visible=True)
+    yield script_content, gr.update(visible=True), summary_content
 
 # Gradio 인터페이스
 with gr.Blocks() as demo:
@@ -160,7 +160,7 @@ with gr.Blocks() as demo:
     analyze_button.click(
         analyze,
         inputs=[youtube_url_input],
-        outputs=[script_output, summary_output]
+        outputs=[script_output, script_output, summary_output]
     )
 
 if __name__ == "__main__":