Update app.py

app.py

@@ -37,37 +37,16 @@ def get_youtube_script(url):
         data = parsed_result["data"][0]
         title = data.get("title", "제목 없음")
         transcription_text = data.get("transcriptionAsText", "")
-        sections = data.get("sections", [])
 
         if not transcription_text:
             raise ValueError("추출된 스크립트가 없습니다.")
 
         logging.info("스크립트 추출 완료")
-        return title, transcription_text, sections
+        return title, transcription_text
     except Exception as e:
         logging.exception("스크립트 추출 중 오류 발생")
         raise
 
-def format_time(seconds):
-    minutes, seconds = divmod(seconds, 60)
-    hours, minutes = divmod(minutes, 60)
-    return f"{int(hours):02d}:{int(minutes):02d}:{int(seconds):02d}"
-
-def generate_timeline_summary(sections):
-    timeline_items = []
-    for section in sections:
-        start_time = format_time(section['start_time'])
-        text = section['text']
-        timeline_items.append(f"<p><strong>{start_time}</strong>: {text}</p>")
-    
-    timeline_html = "\n".join(timeline_items)
-    return f"""
-    <h3>타임라인 요약:</h3>
-    <div style="white-space: pre-wrap; max-height: 400px; overflow-y: auto; border: 1px solid #ccc; padding: 10px;">
-        {timeline_html}
-    </div>
-    """
-
 def call_api(prompt, max_tokens, temperature, top_p):
     try:
         response = openai.ChatCompletion.create(
@@ -111,22 +90,21 @@ def analyze(url, cache):
     try:
         if url == cache["url"]:
             logging.info(f"캐시된 데이터 사용: URL = {url}")
-            title, script, sections = cache["title"], cache["script"], cache["sections"]
+            title, script = cache["title"], cache["script"]
         else:
             logging.info(f"새로운 데이터 추출 시작: URL = {url}")
-            title, script, sections = get_youtube_script(url)
-            cache = {"url": url, "title": title, "script": script, "sections": sections}
+            title, script = get_youtube_script(url)
+            cache = {"url": url, "title": title, "script": script}
 
         script_html = display_script(title, script)
-        timeline_html = generate_timeline_summary(sections)
         summary_html = summarize_text(script)
         
         logging.info("분석 완료")
-        return script_html, timeline_html, summary_html, cache
+        return script_html, summary_html, cache
     except Exception as e:
         error_msg = f"처리 중 오류 발생: {str(e)}"
         logging.exception(error_msg)
-        return error_msg, "", "", cache
+        return error_msg, "", cache
 
 # Gradio 인터페이스
 with gr.Blocks() as demo:
@@ -134,14 +112,13 @@ with gr.Blocks() as demo:
     youtube_url_input = gr.Textbox(label="YouTube URL 입력")
     analyze_button = gr.Button("분석하기")
     script_output = gr.HTML(label="스크립트")
-    timeline_output = gr.HTML(label="타임라인 요약")
     summary_output = gr.HTML(label="전체 요약")
-    cached_data = gr.State({"url": "", "title": "", "script": "", "sections": []})
+    cached_data = gr.State({"url": "", "title": "", "script": ""})
 
     analyze_button.click(
         analyze, 
         inputs=[youtube_url_input, cached_data], 
-        outputs=[script_output, timeline_output, summary_output, cached_data]
+        outputs=[script_output, summary_output, cached_data]
     )
 
 if __name__ == "__main__":