Update app.py

app.py

@@ -78,24 +78,17 @@ def summarize_text(text):
 대본:
 {text}
 """
-    return call_api(prompt, max_tokens=8000, temperature=0.4, top_p=0.95)
+    return call_api(prompt, max_tokens=2000, temperature=0.3, top_p=0.9)
 
-def display_content(title, script, summary):
+def create_collapsible_section(section_title, video_title, content):
     return f"""
-    <div style="display: flex; flex-direction: column; gap: 20px;">
-        <div style="flex: 1;">
-            <h2 style='font-size:24px;'>{title}</h2>
-            <h3>요약</h3>
-            <div style="white-space: pre-wrap; background-color: #f0f0f0; padding: 15px; border-radius: 5px;">{summary}</div>
+    <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>
-        <div style="flex: 1;">
-            <h2 style='font-size:24px;'>{title}</h2>
-            <details>
-                <summary><h3>원문 스크립트 (클릭하여 펼치기)</h3></summary>
-                <div style="white-space: pre-wrap; background-color: #e0e0e0; padding: 15px; border-radius: 5px;">{script}</div>
-            </details>
-        </div>
-    </div>
+    </details>
     """
 
 def analyze(url, cache):
@@ -108,15 +101,19 @@ def analyze(url, cache):
             title, script = get_youtube_script(url)
             cache = {"url": url, "title": title, "script": script}
 
+        # 원문 스크립트 섹션 생성
+        script_section = create_collapsible_section("원문 스크립트", title, script)
+        yield script_section, cache
+
+        # 요약 생성 및 섹션 생성
         summary = summarize_text(script)
-        content_html = display_content(title, script, summary)
-        
-        logging.info("분석 완료")
-        return content_html, cache
+        summary_section = create_collapsible_section("요약", title, summary)
+        yield script_section + summary_section, cache
+
     except Exception as e:
         error_msg = f"처리 중 오류 발생: {str(e)}"
         logging.exception(error_msg)
-        return error_msg, cache
+        yield error_msg, cache
 
 # Gradio 인터페이스
 with gr.Blocks() as demo: