Update app.py

app.py

@@ -119,52 +119,34 @@ def display_script(title, script):
     formatted_script = "\n\n".join(script_sentences)
     return f"# {title}\n\n{formatted_script}"
 
-def analyze(url, progress=gr.Progress()):
-    try:
-        # 스크립트 추출 (진행 상황 표시 없이)
-        title, description, script = get_youtube_script(url)
-        
-        # 원문 스크립트 처리 및 출력
-        script_content = display_script(title, script)
-        yield script_content, "요약 준비 중..."
-        
-        # 요약 작업 시작 (진행 상황 표시)
-        for i in progress.tqdm(range(3), desc="요약 생성 중..."):
-            if i == 0:
-                # 요약 생성
-                summary = summarize_text(title, description, script)
-            elif i == 1:
-                # 요약 내용 정리
-                summary_content = f"# {title}\n\n{summary}"
-            else:
-                # 완료
-                pass
-            yield script_content, f"요약 생성 중... ({i+1}/3)"
-        
-        # 최종 결과 출력
-        yield script_content, summary_content
-    except Exception as e:
-        error_msg = f"처리 중 오류 발생: {str(e)}"
-        logging.exception(error_msg)
-        yield error_msg, error_msg
+def display_script(title, script):
+    script_sentences = split_sentences(script)
+    formatted_script = "\n\n".join(script_sentences)
+    return f"""<h2>{title}</h2>
+<details>
+    <summary>원문 스크립트 (클릭하여 펼치기)</summary>
+    <pre>{formatted_script}</pre>
+</details>"""
+
+def analyze(url):
+    title, description, script = get_youtube_script(url)
+    script_content = display_script(title, script)
+    summary = summarize_text(title, description, script)
+    summary_content = f"<h2>{title}</h2>\n\n{summary}"
+    return script_content, summary_content
 
 # Gradio 인터페이스
 with gr.Blocks() as demo:
     gr.Markdown("## YouTube 스크립트 추출 및 요약 도구")
     youtube_url_input = gr.Textbox(label="YouTube URL 입력")
     analyze_button = gr.Button("분석하기")
-    
-    with gr.Tabs():
-        with gr.TabItem("원문 스크립트"):
-            script_output = gr.Markdown()
-        with gr.TabItem("요약"):
-            summary_output = gr.Markdown()
+    script_output = gr.HTML(label="원문 스크립트")
+    summary_output = gr.HTML(label="요약")
 
     analyze_button.click(
         analyze,
         inputs=[youtube_url_input],
-        outputs=[script_output, summary_output],
-        show_progress=True
+        outputs=[script_output, summary_output]
     )
 
 if __name__ == "__main__":