Update app.py

app.py

@@ -1,73 +1,288 @@
+import gradio as gr
+import requests
+import re
 import os
-import openai
+import openai  # openai 최신버전 (>=1.0.0) 가정
 
-def refine_with_llm_new(old_changes_summary_text: str):
+def debug_print(*args):
+    """디버깅용 콘솔 출력"""
+    print("[DEBUG]", *args)
+
+########################################################
+# (1) 기존 함수들: parse_track_number, fetch_script, show_next_field
+########################################################
+
+def parse_track_number(input_text):
+    debug_print("parse_track_number() called with:", input_text)
+    match = re.search(r'<track[^>]*\skind="captions"[^>]*\ssrc="([^"]+)"', input_text)
+    if not match:
+        debug_print("트랙 태그(kinds='captions')에서 src 추출 실패")
+        return None
+
+    track_number = match.group(1).strip()
+    debug_print("초기 추출 track_number:", track_number)
+
+    # blob: 제거
+    if 'blob:' in track_number:
+        track_number = track_number.replace('blob:', '')
+        debug_print("blob: 제거 후:", track_number)
+
+    # https://player.vimeo.com 제거
+    if 'https://player.vimeo.com' in track_number:
+        track_number = track_number.replace('https://player.vimeo.com', '')
+        debug_print("도메인 제거 후:", track_number)
+
+    track_number = track_number.strip()
+    if not track_number.startswith('/'):
+        track_number = '/' + track_number
+        debug_print("앞에 '/' 추가 후:", track_number)
+
+    debug_print("최종 track_number:", track_number)
+    return track_number
+
+def fetch_script(url):
+    """URL(문자열)에서 트랙번호 추출 후, Vimeo 자막 URL로 GET 요청."""
+    track_number = parse_track_number(url)
+    if not track_number:
+        return "트랙번호를 찾지 못했습니다. 예: <track kind='captions' src='/1234567.vtt' ...>"
+
+    target_url = f"https://player.vimeo.com{track_number}"
+    debug_print("접속할 URL:", target_url)
+    try:
+        resp = requests.get(target_url)
+        if resp.status_code == 200:
+            debug_print("자막 스크립트 가져오기 성공")
+            return resp.text
+        else:
+            debug_print("HTTP 에러:", resp.status_code)
+            return f"HTTP 에러: {resp.status_code}"
+    except Exception as e:
+        debug_print("예외 발생:", str(e))
+        return f"에러 발생: {str(e)}"
+
+def show_next_field(num_visible):
+    """
+    + 버튼 클릭 시, URL 입력칸 하나씩 추가해서 최대 5개까지 표시.
+    """
+    debug_print("show_next_field() called. Current:", num_visible)
+    if num_visible < 5:
+        num_visible += 1
+    updates = []
+    for i in range(1, 6):
+        updates.append(gr.update(visible=(i <= num_visible)))
+
+    return [num_visible] + updates
+
+########################################################
+# (2) (기존) 전체 스크립트 만들기
+########################################################
+
+def is_vtt_timeline_line(line: str) -> bool:
+    # 예: 00:49:33.920 --> 00:49:38.700
+    pattern = r'^\d{2}:\d{2}:\d{2}(\.\d+)?\s*-->\s*\d{2}:\d{2}:\d{2}(\.\d+)?$'
+    return bool(re.match(pattern, line.strip()))
+
+def is_number_line(line: str) -> bool:
+    # 예: 1, 2., 3)
+    pattern = r'^\d+[\.\)]?$'
+    return bool(re.match(pattern, line.strip()))
+
+def is_webvtt_header(line: str) -> bool:
+    # 예: WEBVTT
+    return line.strip() == "WEBVTT"
+
+def combine_scripts(script1, script2, script3, script4, script5):
+    """
+    - (기존) 전체 스크립트: 원본 그대로 합침 + [강의N] 헤더
+    - (기존) 수정사항: 타임라인/숫자/WEBVTT 제거 + [강의N] 헤더
+    """
+    scripts = [script1, script2, script3, script4, script5]
+    merged_full = []
+    merged_filtered = []
+
+    for idx, sc in enumerate(scripts, start=1):
+        if not sc or not sc.strip():
+            continue
+
+        lines = sc.splitlines()
+
+        # (A) 전체 스크립트
+        merged_full.append(f"[강의{idx}]")
+        merged_full.extend(lines)
+
+        # (B) 수정사항
+        merged_filtered.append(f"[강의{idx}]")
+        for line in lines:
+            if is_vtt_timeline_line(line):
+                continue
+            if is_number_line(line):
+                continue
+            if is_webvtt_header(line):
+                continue
+            merged_filtered.append(line)
+
+    old_combined_script = "\n".join(merged_full)
+    old_changes_summary = "\n".join(merged_filtered)
+    return old_combined_script, old_changes_summary
+
+########################################################
+# (3) LLM 호출 로직 - 새로운 지시사항
+########################################################
+
+def refine_with_llm(old_changes_summary_text):
     """
-    1. (기존) 수정사항 텍스트를 입력받아,
-    2. 줄바꿈/띄어쓰기를 문맥에 맞게 재구성,
-    3. 내용은 절대 요약/삭제/임의 변경 금지,
-    4. 오타·맞춤법만 수정, 그 수정 내역을 별도로 정리해 알려준다.
-
-    출력:
-      (교정�� 본문, 수정내역 정리)
-      - 교정된 본문(문맥상 문단 나누거나 붙이기, 맞춤법 수정 반영)
-      - 수정내역(이전 -> 이후, 모든 수정사항)
+    [새로운 요구사항]:
+    1. (기존) 수정사항 내용 우선 기억
+    2. 줄바꿈/띄어쓰기 문맥에 맞게 재구성 (붙이고/나누고)
+    3. 절대 내용 줄이거나 요약/삭제/임의변경하지 말기
+    4. 오타/맞춤법만 수정
+    5. 수정한 부분 모두 정리하여 알려주기
+
+    출력형식:
+      (교정된 본문, 수정내역)
+
+    LLM 응답 예시:
+    교정된 본문
+    === 수정내역 ===
+    (이전 -> 이후 목록)
     """
     openai.api_key = os.getenv("OPENAI_API_KEY")
     if not openai.api_key:
-        return (
-            "오류: OPENAI_API_KEY가 설정되지 않았습니다.",
-            "수정내역 없음"
-        )
+        return "오류: OPENAI_API_KEY가 설정되지 않았습니다.", "수정내역 없음"
 
     system_prompt = (
         "당신은 한국어 문장 교정 전문가입니다.\n"
-        "아래의 [규칙]을 지키면서 텍스트를 교정해 주세요.\n"
         "[규칙]\n"
-        "1) (기존) 수정사항 텍스트를 우선 그대로 기억.\n"
-        "2) 줄바꿈, 띄어쓰기를 문맥에 맞게 재구성.\n"
-        "   - 분리되어야 할 문단은 분리하고\n"
-        "   - 붙여야 할 문단은 붙이되\n"
-        "   - 내용은 절대 삭제하거나 요약하지 말 것.\n"
-        "3) 오타나 맞춤법이 안 맞는 것만 수정.\n"
-        "   - 수정한 부분은 모두 별도로 정리.\n"
-        "4) 절대 임의로 내용을 바꾸지 않는다.\n"
-    )
-
-    user_prompt = (
-        f"아래 텍스트를 [규칙]에 따라 교정해 주세요.\n"
-        f"```\n{old_changes_summary_text}\n```"
-        "\n\n"
+        "1) (기존) 수정사항 텍스트를 우선 그대로.\n"
+        "2) 줄바꿈/띄어쓰기를 문맥에 맞게 재구성.\n"
+        "   - 붙여야 할 문단은 붙이고\n"
+        "   - 분리해야 할 문단은 분리.\n"
+        "3) 절대 내용을 요약/삭제/임의 변경하지 말 것.\n"
+        "4) 오타나 맞춤법만 수정.\n"
+        "5) 수정한 부분을 모두 정리해 알려줄 것.\n"
         "출력 형식:\n"
-        "교정된 본문(문맥상 문단 재구성 반영)\n"
+        "=== 교정된 본문 ===\n"
+        "(재구성+맞춤법 수정된 본문)\n"
         "=== 수정내역 ===\n"
-        "모든 교정 사항(이전 -> 이후) 리스트\n"
+        "(이전 -> 이후) 등 모든 변경사항 리스트.\n"
     )
 
+    user_prompt = f"아래 텍스트를 교정해 주세요:\n```\n{old_changes_summary_text}\n```"
+
     try:
-        # (예: 최신 openai 1.0.0+ 에서는 ChatCompletion 방식을 계속 지원)
         response = openai.ChatCompletion.create(
-            model="gpt-4o-mini",  # 실제 모델 존재 여부는 별도 확인 필요
+            model="gpt-4o-mini",  # 실제 존재X 가능성. 필요시 gpt-3.5-turbo 등 사용
             messages=[
                 {"role": "system", "content": system_prompt},
-                {"role": "user", "content": user_prompt}
+                {"role": "user", "content": user_prompt},
             ],
             temperature=0.0
         )
         raw_output = response.choices[0].message["content"].strip()
     except Exception as e:
-        return (
-            f"LLM 호출 오류: {str(e)}",
-            "수정내역 정보 생성 실패"
-        )
+        return f"LLM 호출 오류: {str(e)}", "수정내역 생성 실패"
 
-    # 아래는 가정된 출력 형식:
-    # 교정된 본문
+    # 출력 예시:
+    # === 교정된 본문 ===
+    # (본문)
     # === 수정내역 ===
-    # (수정 사항들)
-    if "=== 수정내역 ===" in raw_output:
-        corrected_text, changes_text = raw_output.split("=== 수정내역 ===", 1)
-        return corrected_text.strip(), changes_text.strip()
+    # (수정리스트)
+    corrected_sep = "=== 교정된 본문 ==="
+    changes_sep = "=== 수정내역 ==="
+
+    corrected_text = ""
+    changes_text = ""
+
+    if corrected_sep in raw_output and changes_sep in raw_output:
+        # 교정된 본문
+        part1 = raw_output.split(corrected_sep, 1)[1]
+        if changes_sep in part1:
+            corrected_text, changes_text = part1.split(changes_sep, 1)
+            corrected_text = corrected_text.strip()
+            changes_text = changes_text.strip()
+        else:
+            corrected_text = part1.strip()
+            changes_text = "수정내역 구분 없음"
     else:
-        # 수정내역 구분이 없으면, 통째로 본문으로 반환
-        return raw_output, "수정내역 구분 없음"
+        corrected_text = raw_output
+        changes_text = "LLM 응답에 수정내역 구분 없음"
+
+    return corrected_text, changes_text
+
+########################################################
+# (4) build_app: Gradio UI
+########################################################
+
+def build_app():
+    with gr.Blocks() as demo:
+        gr.Markdown("## 강의 URL 입력 및 스크립트 처리")
+
+        num_visible = gr.State(value=2)
+        script_outputs = []
+        row_list = []
+
+        # 최대 5개 URL 입력
+        for i in range(1, 6):
+            with gr.Row(visible=(i <= 2)) as row:
+                with gr.Column():
+                    url_inp = gr.Textbox(
+                        label=f"강의{i} URL 입력",
+                        placeholder="예: <track kind='captions' src='/1234567.vtt' ...>"
+                    )
+                    fetch_btn = gr.Button("스크립트 가져오기")
+                    script_box = gr.Textbox(label=f"강의{i} 스크립트", lines=8)
+
+                    fetch_btn.click(fetch_script, inputs=url_inp, outputs=script_box)
+            row_list.append(row)
+            script_outputs.append(script_box)
+
+        plus_button = gr.Button("+")
+        plus_button.click(show_next_field, inputs=num_visible, outputs=[num_visible]+row_list)
+
+        # (기존) 전체 스크립트 만들기
+        with gr.Row():
+            old_combine_btn = gr.Button("(기존) 전체 스크립트 만들기")
+        with gr.Row():
+            old_combined_box = gr.Textbox(
+                label="(기존) 전체 스크립트",
+                lines=8
+            )
+            old_changes_box = gr.Textbox(
+                label="(기존) 수정사항",
+                lines=8
+            )
+
+        # combine_scripts -> (기존) 전체 스크립트, (기존) 수정사항
+        old_combine_btn.click(
+            fn=combine_scripts,
+            inputs=script_outputs,
+            outputs=[old_combined_box, old_changes_box]
+        )
+
+        # LLM 교정
+        with gr.Row():
+            llm_btn = gr.Button("LLM(새 교정) - 강의내용 정리")
+
+        with gr.Row():
+            corrected_box = gr.Textbox(
+                label="교정된 본문",
+                lines=10,
+                placeholder="줄바꿈/띄어쓰기 재구성 + 맞춤법 수정 결과"
+            )
+            changes_box = gr.Textbox(
+                label="수정내역",
+                lines=8,
+                placeholder="오타/맞춤법 수정된 부분 요약"
+            )
+
+        llm_btn.click(
+            fn=refine_with_llm,
+            inputs=old_changes_box,   # (기존) 수정사항
+            outputs=[corrected_box, changes_box]
+        )
+
+    return demo
+
+if __name__ == "__main__":
+    demo = build_app()
+    demo.launch()