Update app.py

app.py

@@ -109,58 +109,48 @@ def summarize_text(title, description, text):
 """
     return call_api(prompt, max_tokens=2000, temperature=0.3, top_p=0.9)
 
-def create_collapsible_section(section_title, video_title, content):
-    if section_title == "원문 스크립트":
-        sentences = split_sentences(content)
-        content = "\n".join(sentences)
-    
-    return f"""
-    <details>
-        <summary style="cursor: pointer; font-weight: bold;">{section_title}</summary>
-        <div style="margin-top: 10px;">
-            <h3 style="font-size: 18px; margin-bottom: 10px;">{video_title}</h3>
-            <div style="white-space: pre-wrap; background-color: #f0f0f0; padding: 15px; border-radius: 5px;">{content}</div>
-        </div>
-    </details>
-    """
-
-def analyze(url, cache):
+def analyze(url, progress=gr.Progress()):
     try:
-        if url == cache["url"]:
-            logging.info(f"캐시된 데이터 사용: URL = {url}")
-            title, description, script = cache["title"], cache["description"], cache["script"]
-        else:
-            logging.info(f"새로운 데이터 추출 시작: URL = {url}")
-            title, description, script = get_youtube_script(url)
-            cache = {"url": url, "title": title, "description": description, "script": script}
-
-        # 원문 스크립트 섹션 생성
-        script_section = create_collapsible_section("원문 스크립트", title, script)
-        yield script_section, cache
-
-        # 요약 생성 및 섹션 생성
+        progress(0, desc="스크립트 추출 중...")
+        title, description, script = get_youtube_script(url)
+        
+        progress(33, desc="원문 스크립트 처리 중...")
+        script_sentences = split_sentences(script)
+        script_content = "\n".join(script_sentences)
+        
+        progress(66, desc="요약 생성 중...")
         summary = summarize_text(title, description, script)
-        summary_section = create_collapsible_section("요약", title, summary)
-        yield script_section + summary_section, cache
-
+        
+        progress(100, desc="완료")
+        return {
+            "제목": title,
+            "원문 스크립트": script_content,
+            "요약": summary
+        }
     except Exception as e:
         error_msg = f"처리 중 오류 발생: {str(e)}"
         logging.exception(error_msg)
-        yield error_msg, cache
+        return {"오류": error_msg}
 
 # Gradio 인터페이스
 with gr.Blocks() as demo:
     gr.Markdown("## YouTube 스크립트 추출 및 요약 도구")
     youtube_url_input = gr.Textbox(label="YouTube URL 입력")
     analyze_button = gr.Button("분석하기")
-    content_output = gr.HTML(label="내용")
-    cached_data = gr.State({"url": "", "title": "", "description": "", "script": ""})
+    
+    with gr.Tabs():
+        with gr.TabItem("원문 스크립트"):
+            script_output = gr.Markdown()
+        with gr.TabItem("요약"):
+            summary_output = gr.Markdown()
+    
+    title_output = gr.Textbox(label="영상 제목")
 
     analyze_button.click(
-        analyze, 
-        inputs=[youtube_url_input, cached_data], 
-        outputs=[content_output, cached_data]
+        analyze,
+        inputs=[youtube_url_input],
+        outputs=[title_output, script_output, summary_output]
     )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()