Update app.py

app.py

@@ -11,6 +11,8 @@ import re
 logging.basicConfig(filename='youtube_script_extractor.log', level=logging.DEBUG, 
                     format='%(asctime)s - %(levelname)s - %(message)s')
 
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
 def parse_api_response(response):
     try:
         if isinstance(response, str):
@@ -68,8 +70,6 @@ def get_youtube_script(url):
         logging.exception(error_msg)
         return "", "", []
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
-
 def call_api(prompt, max_tokens, temperature, top_p):
     try:
         response = openai.ChatCompletion.create(
@@ -155,6 +155,8 @@ with gr.Blocks() as demo:
         return title, script, sections, new_cache
 
     def display_script(title, script):
+        if not script:
+            return "<p>스크립트를 추출하지 못했습니다. URL을 확인하고 다시 시도해 주세요.</p>"
         formatted_script = "\n".join(split_sentences(script))
         script_html = f"""<h2 style='font-size:24px;'>{title}</h2>
         <details>
@@ -164,6 +166,8 @@ with gr.Blocks() as demo:
         return script_html
 
     def display_timeline(sections):
+        if not sections:
+            return "<p>타임라인을 생성하지 못했습니다. 스크립트 추출에 실패했을 수 있습니다.</p>"
         timeline_summary = generate_timeline_summary(sections)
         timeline_html = f"""
         <h3>타임라인 요약:</h3>
@@ -174,6 +178,8 @@ with gr.Blocks() as demo:
         return timeline_html
 
     def generate_summary(script):
+        if not script:
+            return "<p>전체 요약을 생성하지 못했습니다. 스크립트 추출에 실패했을 수 있습니다.</p>"
         summary = summarize_text(script)
         summary_html = f"""
         <h3>전체 요약:</h3>
@@ -187,21 +193,13 @@ with gr.Blocks() as demo:
         title, script, sections, new_cache = extract_and_cache(url, cache)
         script_html = display_script(title, script)
         timeline_html = display_timeline(sections)
-        return script_html, timeline_html, new_cache
-
-    def update_summary(cache):
-        if not cache["script"]:
-            return "스크립트가 없습니다. 먼저 YouTube URL을 입력하고 분석을 실행해주세요."
-        return generate_summary(cache["script"])
+        summary_html = generate_summary(script)
+        return script_html, timeline_html, summary_html, new_cache
 
     analyze_button.click(
         analyze, 
         inputs=[youtube_url_input, cached_data], 
-        outputs=[script_output, timeline_output, cached_data]
-    ).then(
-        update_summary,
-        inputs=[cached_data],
-        outputs=summary_output
+        outputs=[script_output, timeline_output, summary_output, cached_data]
     )
 
 demo.launch(share=True)