Update app.py

app.py

@@ -1,10 +1,8 @@
-# app.py
-
 import gradio as gr
 from bs4 import BeautifulSoup
 import re
 import logging
-import requests  # 추가코드에서 사용
+import requests
 
 # 디버깅 로깅 설정 (전체 통합을 위해 포맷 포함)
 logging.basicConfig(level=logging.DEBUG, format="%(asctime)s - %(levelname)s - %(message)s")
@@ -207,26 +205,25 @@ def remove_timeline(script_text, lecture_number):
         stripped_line = line.strip()
         if stripped_line == "":
             continue
-        # 번호만 있는 줄 제거
         if re.match(r'^\d+$', stripped_line):
             continue
-        # 타임라인 형식 제거 (예: 00:00:00.000 --> 00:00:02.000)
         if re.match(r'^\d{1,2}:\d{2}(?::\d{2}(?:\.\d{3})?)?\s*-->\s*\d{1,2}:\d{2}(?::\d{2}(?:\.\d{3})?)?$', stripped_line):
             continue
         valid_lines.append(stripped_line)
     cleaned_text = "".join(valid_lines)
-    # 마침표(.) 다음에 여백이 없으면 여백 추가
     cleaned_text = re.sub(r'\.(\S)', r'. \1', cleaned_text)
-    # 글 가장 앞에 있는 "WEBVTT" 제거
     cleaned_text = re.sub(r'^WEBVTT\s*', '', cleaned_text)
     return cleaned_text
 
+# 기존의 "스크립트 URL 만들기", "스크립트 가져오기", "타임라인 제거" 관련 컴포넌트는 제거하고,
+# 아래 [모듈2 구조]에 맞게 새롭게 구성합니다.
+
 # 새로운 전체 처리 함수: 강의 URL -> 스크립트 URL 생성 -> 스크립트 가져오기 -> 타임라인 제거
 def process_full(lecture_url):
     """
     입력된 강의 URL로부터 스크립트 URL을 생성하고,
     해당 스크립트를 가져온 후 타임라인을 제거한 최종 강의 내용을 반환하는 함수.
-    중간 단계는 출력하지 않고 최종 결과만 반환합니다.
+    (중간 단계는 출력하지 않고 최종 결과만 반환)
     """
     script_url = create_script_url(lecture_url)
     if not script_url:
@@ -236,59 +233,39 @@ def process_full(lecture_url):
     if "오류 발생" in script_text:
         return script_text
 
-    cleaned = remove_timeline(script_text, 1)  # lecture_number는 기본값으로 1 사용
+    cleaned = remove_timeline(script_text, 1)
     return cleaned
 
+# [모듈2 구조]
 with gr.Blocks() as additional_demo:
-    # 모듈2 상단 안내 문구
-    gr.Markdown("※ 아래 순서대로 각 단계별 버튼을 눌러 진행하세요.")
+    # 안내 문구 수정
+    gr.Markdown("※ '강의 내용 가져오기' 버튼을 클릭하면 해당 영상의 강의 내용을 추출할 수 있습니다.") 
     with gr.Row():
         url1 = gr.Textbox(label="강의1 URL", elem_id="url1")
         url2 = gr.Textbox(label="강의2 URL", elem_id="url2")
         url3 = gr.Textbox(label="강의3 URL", elem_id="url3")
     with gr.Row():
-        gen_url_btn1 = gr.Button("스크립트 URL 만들기", elem_id="gen_url_btn1")
-        gen_url_btn2 = gr.Button("스크립트 URL 만들기", elem_id="gen_url_btn2")
-        gen_url_btn3 = gr.Button("스크립트 URL 만들기", elem_id="gen_url_btn3")
-    with gr.Row():
-        script_url1 = gr.Textbox(label="강의1 스크립트 URL", interactive=False, elem_id="script_url1")
-        script_url2 = gr.Textbox(label="강의2 스크립트 URL", interactive=False, elem_id="script_url2")
-        script_url3 = gr.Textbox(label="강의3 스크립트 URL", interactive=False, elem_id="script_url3")
-    with gr.Row():
-        fetch_btn1 = gr.Button("스크립트 가져오기", elem_id="fetch_btn1")
-        fetch_btn2 = gr.Button("스크립트 가져오기", elem_id="fetch_btn2")
-        fetch_btn3 = gr.Button("스크립트 가져오기", elem_id="fetch_btn3")
+        get_btn1 = gr.Button("강의 내용 가져오기", elem_id="get_btn1") 
+        get_btn2 = gr.Button("강의 내용 가져오기", elem_id="get_btn2") 
+        get_btn3 = gr.Button("강의 내용 가져오기", elem_id="get_btn3") 
     with gr.Row():
-        script1 = gr.Textbox(label="강의1 스크립트", lines=10, elem_id="script1")
-        script2 = gr.Textbox(label="강의2 스크립트", lines=10, elem_id="script2")
-        script3 = gr.Textbox(label="강의3 스크립트", lines=10, elem_id="script3")
+        lecture_content1 = gr.Textbox(label="강의 내용", lines=10, elem_id="lecture_content1")
+        lecture_content2 = gr.Textbox(label="강의 내용", lines=10, elem_id="lecture_content2")
+        lecture_content3 = gr.Textbox(label="강의 내용", lines=10, elem_id="lecture_content3")
     with gr.Row():
-        remove_btn1 = gr.Button("타임라인 제거", elem_id="remove_btn1")
-        remove_btn2 = gr.Button("타임라인 제거", elem_id="remove_btn2")
-        remove_btn3 = gr.Button("타임라인 제거", elem_id="remove_btn3")
+        copy_btn1 = gr.Button("강의 내용 복사하기", elem_id="copy_btn1")
+        copy_btn2 = gr.Button("강의 내용 복사하기", elem_id="copy_btn2")
+        copy_btn3 = gr.Button("강의 내용 복사하기", elem_id="copy_btn3")
     with gr.Row():
-        cleaned1 = gr.Textbox(label="강의1 타임라인 제거", lines=10, interactive=False, elem_id="cleaned1")
-        cleaned2 = gr.Textbox(label="강의2 타임라인 제거", lines=10, interactive=False, elem_id="cleaned2")
-        cleaned3 = gr.Textbox(label="강의3 타임라인 제거", lines=10, interactive=False, elem_id="cleaned3")
-    with gr.Row():
-        copy_btn1 = gr.Button("강의 내용 복사", elem_id="copy_btn1")
-        copy_btn2 = gr.Button("강의 내용 복사", elem_id="copy_btn2")
-        copy_btn3 = gr.Button("강의 내용 복사", elem_id="copy_btn3")
-    with gr.Row():
-        copy_result1 = gr.Textbox(label="강의 복사 결과", interactive=False, elem_id="copy_result1")
-        copy_result2 = gr.Textbox(label="강의 복사 결과", interactive=False, elem_id="copy_result2")
-        copy_result3 = gr.Textbox(label="강의 복사 결과", interactive=False, elem_id="copy_result3")
-
-    gen_url_btn1.click(fn=lambda url: create_script_url(url), inputs=url1, outputs=script_url1)
-    gen_url_btn2.click(fn=lambda url: create_script_url(url), inputs=url2, outputs=script_url2)
-    gen_url_btn3.click(fn=lambda url: create_script_url(url), inputs=url3, outputs=script_url3)
-    fetch_btn1.click(fn=lambda url: fetch_script(url), inputs=script_url1, outputs=script1)
-    fetch_btn2.click(fn=lambda url: fetch_script(url), inputs=script_url2, outputs=script2)
-    fetch_btn3.click(fn=lambda url: fetch_script(url), inputs=script_url3, outputs=script3)
-    remove_btn1.click(fn=lambda script: remove_timeline(script, 1), inputs=script1, outputs=cleaned1)
-    remove_btn2.click(fn=lambda script: remove_timeline(script, 2), inputs=script2, outputs=cleaned2)
-    remove_btn3.click(fn=lambda script: remove_timeline(script, 3), inputs=script3, outputs=cleaned3)
-
+        copy_result1 = gr.Textbox(label="강의 내용 복사하기 결과", interactive=False, elem_id="copy_result1")
+        copy_result2 = gr.Textbox(label="강의 내용 복사하기 결과", interactive=False, elem_id="copy_result2")
+        copy_result3 = gr.Textbox(label="강의 내용 복사하기 결과", interactive=False, elem_id="copy_result3")
+    
+    get_btn1.click(fn=lambda url: process_full(url), inputs=url1, outputs=lecture_content1)
+    get_btn2.click(fn=lambda url: process_full(url), inputs=url2, outputs=lecture_content2)
+    get_btn3.click(fn=lambda url: process_full(url), inputs=url3, outputs=lecture_content3)
+    
+    # 강의 내용 복사 기능은 클라이언트 사이드 JavaScript로 처리합니다.
     custom_script = """
     <script>
     function setupCopy(copyBtnId, textBoxId, resultBoxId) {
@@ -319,9 +296,9 @@ with gr.Blocks() as additional_demo:
         });
     }
     document.addEventListener("DOMContentLoaded", function(){
-        setupCopy("copy_btn1", "cleaned1", "copy_result1");
-        setupCopy("copy_btn2", "cleaned2", "copy_result2");
-        setupCopy("copy_btn3", "cleaned3", "copy_result3");
+        setupCopy("copy_btn1", "lecture_content1", "copy_result1");
+        setupCopy("copy_btn2", "lecture_content2", "copy_result2");
+        setupCopy("copy_btn3", "lecture_content3", "copy_result3");
     });
     </script>
     """
@@ -329,14 +306,6 @@ with gr.Blocks() as additional_demo:
     gr.Markdown("디버깅 모드 활성화됨: 로그가 콘솔에 출력됩니다.")
 # ---------- [모듈2: 추가코드] 끝 ----------
 
-# 새로운 전체 처리 인터페이스: 강의 URL -> 타임라인 제거 결과만 반환
-full_interface = gr.Interface(
-    fn=process_full,
-    inputs=gr.Textbox(label="강의 URL", placeholder="강의 URL을 입력하세요..."),
-    outputs=gr.Textbox(label="타임라인 제거 결과", lines=10),
-    description="강의 URL을 입력하면 스크립트 URL 생성, 스크립트 가져오기, 타임라인 제거까지 한 번에 수행합니다."
-)
-
 # ---------- 통합 Gradio 앱 구성 (한 페이지에 모두 표시) ----------
 with gr.Blocks() as app:
     gr.Markdown("# 캐롤라인대학 강의 추출기")
@@ -347,56 +316,86 @@ with gr.Blocks() as app:
             <strong>사용방법</strong>
             <ol>
                 <li>추출을 원하는 강의 페이지에서 "Ctrl + U"를 누르고 "페이지 소스 보기" 페이지로 이동합니다.
-                또는 해당페이지에서 마우스 우클릭 후 "페이지 소스 보기" 클릭 (엣지의 경우 "페이지 원본 보기")</li>
-                <li>"페이지 소스 보기" 페이지에서 전체 복사를 합니다. "Ctrl+A" 다음 "Ctrl + C"</li>
-                <li>"전체 페이지 HTML 입력"창에 복사한 내용을 붙혀넣기 합니다. "Ctrl + V"</li>
+                또는 해당 페이지에서 마우스 우클릭 후 "페이지 소스 보기" 클릭 (엣지의 경우 "페이지 원본 보기")</li>
+                <li>"페이지 소스 보기" 페이지에서 전체 복사를 합니다. ("Ctrl+A" 후 "Ctrl+C")</li>
+                <li>"전체 페이지 HTML 입력"창에 복사한 내용을 붙여넣기 합니다. ("Ctrl+V")</li>
                 <li>"Submit" 버튼을 클릭하면 동영상 강의 목록��� 나옵니다.</li>
-                <li>내용 추출을 원하는 강의 URL을 복사해서 아래 "강의 URL" 창에 붙혀넣기 합니다.</li>
-                <li>순서대로 버튼을 클릭하면 해당 영상의 강의 내용을 추출 할 수 있습니다.</li>
-                <li>각각 추출된 강의 내용을 "ChatGPT"에 전체 붙혀넣고 강의 요약을 지시하면 됩니다.</li>
+                <li>내용 추출을 원하는 강의 URL을 복사해서 아래 "강의 URL" 창에 붙여넣기 합니다.</li>
+                <li>"강의내용 가져오기" 버튼을 클릭하면 해당 영상의 강의 내용을 추출할 수 있습니다.</li>
+                <li>각각 추출된 강의 내용을 "ChatGPT"에 붙여넣고 강의 요약을 요청하면 됩니다.</li>
             </ol>
         </div>
         """
     )
     iface.render()
-    # Module1의 Submit 버튼 클릭 시 Module2 내용 지우기 위한 JavaScript
-    gr.HTML(
-        """
-        <script>
-        function clearModule2Fields() {
-            var ids = ["url1", "url2", "url3", "script_url1", "script_url2", "script_url3", "script1", "script2", "script3", "cleaned1", "cleaned2", "cleaned3", "copy_result1", "copy_result2", "copy_result3"];
-            ids.forEach(function(id) {
-                var elem = document.getElementById(id);
-                if (elem) {
-                    var textarea = elem.querySelector("textarea");
-                    if (textarea) {
-                        textarea.value = "";
+    gr.Markdown("---")
+    gr.Markdown("동영상 강의 목록에서 필요한 강의의 URL을 복사해서 넣어주세요")
+    additional_demo.render()
+    
+    # ---------- [모듈3: 강의내용 합치기 추가코드] 시작 ----------
+    gr.Markdown("## 강의내용 합치기")
+    with gr.Row():
+        merge_btn = gr.Button("강의내용 합치기", elem_id="merge_btn")
+    merged_content = gr.Textbox(label="전체 강의 내용", lines=10, elem_id="merged_content")
+    with gr.Row():
+        merge_copy_btn = gr.Button("전체 강의 내용 복사하기", elem_id="merge_copy_btn")
+    merge_copy_result = gr.Textbox(label="전체 강의 내용 복사 결과", interactive=False, elem_id="merge_copy_result")
+    
+    def merge_contents(l1, l2, l3):
+        merged = ""
+        if l1.strip():
+            merged += "[강의1]\n" + l1.strip()
+        if l2.strip():
+            if merged:
+                merged += "\n\n"
+            merged += "[강의2]\n" + l2.strip()
+        if l3.strip():
+            if merged:
+                merged += "\n\n"
+            merged += "[강의3]\n" + l3.strip()
+        return merged
+    
+    merge_btn.click(fn=merge_contents, inputs=[lecture_content1, lecture_content2, lecture_content3], outputs=merged_content)
+    
+    # 전체 강의 내용 복사 기능을 위한 클라이언트 사이드 JavaScript 추가
+    merge_copy_script = """
+    <script>
+    function setupMergeCopy(copyBtnId, textBoxId, resultBoxId) {
+        const copyBtn = document.getElementById(copyBtnId);
+        if (!copyBtn) {
+            console.error("버튼 " + copyBtnId + "를 찾을 수 없습니다.");
+            return;
+        }
+        copyBtn.addEventListener("click", function(){
+            const textBoxElem = document.getElementById(textBoxId);
+            const resultBoxElem = document.getElementById(resultBoxId);
+            if(textBoxElem && resultBoxElem) {
+                const textarea = textBoxElem.querySelector("textarea");
+                const resultTextarea = resultBoxElem.querySelector("textarea");
+                if(textarea && resultTextarea) {
+                    var text = textarea.value;
+                    if(text.trim() === ""){
+                        resultTextarea.value = "복사할 내용이 없습니다.";
                     } else {
-                        elem.value = "";
+                        navigator.clipboard.writeText(text).then(function(){
+                            resultTextarea.value = "복사완료";
+                        }, function(err){
+                            resultTextarea.value = "복사 실패";
+                        });
                     }
                 }
-            });
-        }
-        document.addEventListener("DOMContentLoaded", function(){
-            var buttons = document.getElementsByTagName("button");
-            for (var i = 0; i < buttons.length; i++) {
-                if (buttons[i].textContent.trim() === "Submit") {
-                    buttons[i].addEventListener("click", function(){
-                        clearModule2Fields();
-                    });
-                }
             }
         });
-        </script>
-        """
-    )
-    gr.Markdown("---")
-    gr.Markdown("동영상 강의 목록에서 필요한 강의의 URL을 복사해서 넣어주세요")
-    additional_demo.render()
-    gr.Markdown("---")
-    gr.Markdown("### 전체 처리 (강의 URL -> 타임라인 제거)")
-    full_interface.render()
-
+    }
+    document.addEventListener("DOMContentLoaded", function(){
+        setupMergeCopy("merge_copy_btn", "merged_content", "merge_copy_result");
+    });
+    </script>
+    """
+    gr.HTML(merge_copy_script)
+    # ---------- [모듈3: 강의내용 합치기 추가코드] 끝 ----------
+    
+# ---------- 통합 Gradio 앱 구성 끝 ----------
 if __name__ == "__main__":
     logging.debug("통합 Gradio 앱 실행 중")
     app.launch(debug=True)