Update app.py

app.py

@@ -1,212 +1,71 @@
-import gradio as gr
-from gradio_client import Client
-import json
-import logging
-import openai
 import os
+import requests
+import json
+import gradio as gr
 import re
-import html
-
-# 로깅 설정
-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):
-            response = json.loads(response)
-        if isinstance(response, list) and len(response) > 0:
-            response = response[0]
-        if not isinstance(response, dict):
-            raise ValueError(f"예상치 못한 응답 형식입니다. 받은 데이터 타입: {type(response)}")
-        return response
-    except Exception as e:
-        logging.error(f"API 응답 파싱 실패: {str(e)}")
-        raise ValueError(f"API 응답 파싱 실패: {str(e)}")
-
-def get_youtube_script(url):
-    logging.info(f"스크립트 추출 시작: URL = {url}")
-    client = Client("whispersound/YT_Ts_R")
-    try:
-        result = client.predict(youtube_url=url, api_name="/predict")
-        parsed_result = parse_api_response(result)
-        
-        if 'data' not in parsed_result or not parsed_result['data']:
-            raise ValueError("API 응답에 유효한 데이터가 없습니다.")
-
-        data = parsed_result["data"][0]
-        title = data.get("title", "제목 없음")
-        description = data.get("description", "설명 없음")
-        transcription_text = data.get("transcriptionAsText", "")
-        thumbnails = data.get("thumbnails", [])
-
-        if not transcription_text:
-            raise ValueError("추출된 스크립트가 없습니다.")
-
-        logging.info("스크립트 추출 완료")
-        return title, description, transcription_text, thumbnails
-    except Exception as e:
-        logging.exception("스크립트 추출 중 오류 발생")
-        raise
-
-def call_api(prompt, max_tokens, temperature, top_p):
-    try:
-        response = openai.ChatCompletion.create(
-            model="gpt-4o-mini",
-            messages=[{"role": "user", "content": prompt}],
-            max_tokens=max_tokens,
-            temperature=temperature,
-            top_p=top_p
-        )
-        return response['choices'][0]['message']['content']
-    except Exception as e:
-        logging.exception("LLM API 호출 중 오류 발생")
-        raise
-
-def summarize_text(title, description, text):
-    prompt = f"""
-[유튜브 요약 규칙]
-1. 너는 유튜브 영상 전문 해설가로서 지침에 맞게 이 글을 작성하라
-2. 아래의 제목과 설명은 이 유튜브 영상의 원본 메타데이터이다.
-3. 반드시 제목과 설명으로 주제와 문맥, 철자(Spelling)을 먼저 파악하고, 아래의 대본을 반드시 지침에 맞게 상세하게 요약하라
-  - 반드시 주어진 제목, 설명에 있는 철자(Spelling)를 요약에 반영하라(원문 대본에는 오탈자가 있을 수 있다)
-4. 반드시 한글로 작성하라
-5. 반드시 '이 유튜브 대본은', '이 영상은', '이 유튜브는'등의 소개식 표현은 제외하라
-6. 요약문만으로도 영상을 직접 시청한 것과 동일한 수준으로 내용을 이해할 수 있도록 상세히 작성
-7. 글을 너무 압축하거나 함축하지 말고, 중요한 내용과 세부사항을 모두 포함
-8. 반드시 대본의 흐름과 논리 구조를 유지
-9. 대본의 목적이나 의도를 파악하고, 이를 요약에 반드시 반영
-10. 반드시 시간 순서나 사건의 전개 과정을 명확하게 반영
-11. 등장인물, 장소, 사건 등 중요한 요소를 정확하게 작성
-12. 대본에서 전달하는 감정이나 분위기도 포함
-13. 반드시 기술적 용어나 전문 용어가 있을 경우, 이를 정확하게 사용
-14. 반드시 핵심 섹션(소주제)를 파악하여 섹션에 맞게 글을 요약하라(글의 양을 고려하여 섹션의 개수를 탄력적으로 설정)
-15. 각 섹션의 제목(소주제)에는 내용과 어울리는 적절한 이모지로 소주제를 시작하라
-16. 각 섹션의 내용은 Bullet Point를 사용하여 가독성을 높여라(문장 단위로 구분)
-  [예시]
-(변경전)
- - 유튜브를 처음 시작하는 사람들은 구독자 수와 조회수에 큰 관심을 두고 매일 유튜브 스튜디오를 확인하게 된다. 그러나 구독자가 100명, 1,000명에 도달하는 것만으로는 지속적인 성장에 도움이 되지 않는다. 구독자 수가 늘어난 후에도 유튜브 채널 운영에 대한 감을 잡지 못해 포기하는 경우가 많다.
-(변경후)
- - 유튜브를 처음 시작하는 사람들은 구독자 수와 조회수에 큰 관심을 두고 매일 유튜브 스튜디오를 확인하게 된다. 
- - 그러나 구독자가 100명, 1,000명에 도달하는 것만으로는 지속적인 성장에 도움이 되지 않는다. 
- - 구독자 수가 늘어난 ��에도 유튜브 채널 운영에 대한 감을 잡지 못해 포기하는 경우가 많다.
-17. 각 섹션의 내용을 반드시 충실하게 작성
 
-제목: {title}
-설명: {description}
-
-대본:
-{text}
-"""
-    return call_api(prompt, max_tokens=8000, temperature=0.35, top_p=0.95)
-
-def split_sentences(text):
-    sentences = re.split(r"(니다|에요|구나|해요|군요|겠어요|시오|해라|예요|아요|데요|대요|세요|어요|게요|구요|고요|나요|하죠)(?![\w])", text)
-    combined_sentences = []
-    current_sentence = ""
-    for i in range(0, len(sentences), 2):
-        if i + 1 < len(sentences):
-            sentence = sentences[i] + sentences[i + 1]
-        else:
-            sentence = sentences[i]
-        if len(current_sentence) + len(sentence) > 100:  # 100자를 초과할 경우
-            combined_sentences.append(current_sentence.strip())
-            current_sentence = sentence.strip()
-        else:
-            current_sentence += sentence
-        if sentence.endswith(('.', '?', '!')):
-            combined_sentences.append(current_sentence.strip())
-            current_sentence = ""
-    if current_sentence:
-        combined_sentences.append(current_sentence.strip())
-    return combined_sentences
-
-def display_script(title, script):
-    script_sentences = split_sentences(script)
-    formatted_script = "\n\n".join(script_sentences)
-    return f"""<div class="script-box">
-<details>
-    <summary>클릭하여 펼치기</summary>
-    <div class="output-title">{title}</div>
-    <p style="white-space: pre-wrap;">{formatted_script}</p>
-</details>
-</div>"""
-
-def display_summary(title, summary):
-    return f"""<div class="script-box">
-<div class="output-title">{title}</div>
-{summary}
-</div>"""
-
-def get_thumbnail_url(thumbnails):
-    for thumbnail in thumbnails:
-        if thumbnail.get("width") == 640 and thumbnail.get("height") == 480:
-            return thumbnail.get("url")
-    return "640x480 크기의 썸네일을 찾을 수 없습니다."
-    
-def analyze(url):
-    # 스크립트 추출
-    yield "스크립트 추출 중...", "스크립트 추출 중...", ""
-    title, description, script, thumbnails = get_youtube_script(url)
-    script_content = display_script(title, script)
-    thumbnail_url = get_thumbnail_url(thumbnails)
-    
-    # 원문 스크립트 표시 및 요약 시작
-    yield script_content, "요약 생성 중...", thumbnail_url
+# Hugging Face 환경 변수로부터 RapidAPI 키와 호스트 가져오기
+AA_KEY = os.getenv("AA_KEY")
+AA_HOST = "youtube-transcriptor.p.rapidapi.com"
+
+# 유튜브 URL에서 비디오 ID를 추출하는 함수
+def get_video_id(youtube_url):
+    # 유튜브 URL 또는 youtu.be 단축 URL에서 video_id 추출
+    video_id_match = re.search(r"(?<=v=)[^#&?]*", youtube_url) or re.search(r"(?<=youtu.be/)[^#&?]*", youtube_url)
+    return video_id_match.group(0) if video_id_match else None
+
+# 자막 언어 우선순위 리스트
+LANGUAGE_PRIORITY = ['ko', 'en', 'ja', 'zh']
+
+# 유튜브 자막을 요청하는 함수 (언어 우선순위를 적용하여 시도)
+def get_youtube_transcript(youtube_url):
+    # 비디오 ID 추출
+    video_id = get_video_id(youtube_url)
+    if video_id is None:
+        return {"error": "잘못된 유튜브 URL입니다. 비디오 ID를 찾을 수 없습니다."}
     
-    # 요약 생성
-    summary = summarize_text(title, description, script)
+    url = "https://youtube-transcriptor.p.rapidapi.com/transcript"
     
-    # HTML로 변환 (convert_to_html 로직을 직접 통합)
-    lines = summary.split('\n')
-    formatted_lines = []
-    for line in lines:
-        line = line.strip()
-        if line.startswith('####'):
-            formatted_lines.append(f"<h4>{html.escape(line[4:].strip())}</h4>")
-        elif line.startswith('###'):
-            formatted_lines.append(f"<h3>{html.escape(line[3:].strip())}</h3>")
-        elif line.startswith('##'):
-            formatted_lines.append(f"<h2>{html.escape(line[2:].strip())}</h2>")
-        elif line.startswith('#'):
-            formatted_lines.append(f"<h1>{html.escape(line[1:].strip())}</h1>")
-        elif line.startswith('- '):  # 리스트 아이템
-            content = html.escape(line[2:])
-            bold_content = re.sub(r'\*\*(.*?)\*\*', r'<strong>\1</strong>', content)
-            formatted_lines.append(f"<li>{bold_content}</li>")
-        elif line:  # 일반 텍스트 (빈 줄 제외)
-            content = html.escape(line)
-            bold_content = re.sub(r'\*\*(.*?)\*\*', r'<strong>\1</strong>', content)
-            formatted_lines.append(f"<p>{bold_content}</p>")
-        else:  # 빈 줄
-            formatted_lines.append("<br>")
-    
-    formatted_summary = '\n'.join(formatted_lines)
-    
-    summary_content = f"""<div class="script-box">
-    <div class="output-title">{html.escape(title)}</div>
-    {formatted_summary}
-    </div>"""
+    headers = {
+        "x-rapidapi-key": AA_KEY,
+        "x-rapidapi-host": AA_HOST
+    }
+
+    # 언어 우선순위에 따라 순��적으로 요청을 시도
+    for lang in LANGUAGE_PRIORITY:
+        querystring = {"video_id": video_id, "lang": lang}
+        response = requests.get(url, headers=headers, params=querystring)
+        
+        # 상태 코드 확인 및 전체 응답 반환
+        if response.status_code == 200:
+            try:
+                data = response.json()
+
+                # 전체 응답 데이터를 그대로 반환
+                return {"language": lang, "data": data}
+            
+            except json.JSONDecodeError as e:
+                return {"error": f"JSON 디코딩 오류 발생: {str(e)}"}
+
+    # 모든 언어에서 자막을 찾지 못한 경우
+    return {"error": "우선순위 언어로 자막을 찾을 수 없습니다."}
+
+# Gradio 인터페이스 정의
+def youtube_transcript_interface(youtube_url):
+    # 자막 데이터 가져오기
+    transcript_data = get_youtube_transcript(youtube_url)
     
-    # 최종 결과 표시
-    yield script_content, summary_content, thumbnail_url
-
-# Gradio 인터페이스
-with gr.Blocks() as demo:
-    gr.Markdown("## YouTube 스크립트 추출 및 요약 도구")
-    youtube_url_input = gr.Textbox(label="YouTube URL 입력")
-    analyze_button = gr.Button("분석하기")
-    script_output = gr.HTML(label="원문 스크립트")
-    summary_output = gr.HTML(label="요약")
-    thumbnail_output = gr.Textbox(label="썸네일 URL (640x480)")  # 이 줄 추가
-
-    analyze_button.click(
-        analyze,
-        inputs=[youtube_url_input],
-        outputs=[script_output, summary_output, thumbnail_output]  # thumbnail_output 추가
-    )
-
-if __name__ == "__main__":
-    demo.launch()
+    # 결과 출력
+    return json.dumps(transcript_data, ensure_ascii=False, indent=2)
+
+# Gradio 인터페이스 생성
+interface = gr.Interface(
+    fn=youtube_transcript_interface, 
+    inputs="text", 
+    outputs="text",
+    title="YouTube 자막 추출기",
+    description="유튜브 URL을 입력하세요."
+)
+
+# Gradio 인터페이스 실행
+interface.launch()