Update app.py

app.py

@@ -63,13 +63,18 @@ def get_youtube_transcript(youtube_url):
         return {"error": "자막을 불러오는데 실패했습니다."}
 
 # Gradio 인터페이스 함수
-def youtube_transcript_interface(youtube_url):
-    transcript_data = get_youtube_transcript(youtube_url)
+def youtube_transcript_interface(youtube_url, language_name):
+    selected_lang = map_language_name_to_code(language_name)
+    transcript_data = get_youtube_transcript(youtube_url, selected_lang)
     
     if "error" in transcript_data:
         return transcript_data["error"]
     
-    data = transcript_data.get("data", {})
+    data_list = transcript_data.get("data", [])
+    if not data_list:
+        return "자막을 찾을 수 없습니다."
+    
+    data = data_list[0]  # 리스트의 첫 번째 요소 접근
     title = data.get("title", "제목을 찾을 수 없습니다.")
     transcription = data.get("transcriptionAsText", "자막을 찾을 수 없습니다.")
     
@@ -77,13 +82,31 @@ def youtube_transcript_interface(youtube_url):
     formatted_output = f"제목: {title}\n\n{transcription}"
     return formatted_output
 
+# 사용 가능한 언어 목록 (예시)
+available_languages = [
+    ("한국어", "ko"),
+    ("영어", "en"),
+    ("일본어", "ja"),
+    ("중국어", "zh")
+]
+
+# 언어 선택 시 내부적으로 언어 코드를 매핑하기 위한 함수
+def map_language_name_to_code(language_name):
+    for name, code in available_languages:
+        if name == language_name:
+            return code
+    return 'ko'  # 기본값
+
 # Gradio 인터페이스 생성
 interface = gr.Interface(
     fn=youtube_transcript_interface,
-    inputs=gr.Textbox(lines=2, placeholder="유튜브 URL을 입력하세요."),
+    inputs=[
+        gr.Textbox(lines=2, placeholder="유튜브 URL을 입력하세요."),
+        gr.Dropdown(choices=[lang[0] for lang in available_languages], label="자막 언어", value="한국어")
+    ],
     outputs=gr.Textbox(),
     title="YouTube 자막 추출기",
-    description="유튜브 URL을 입력하면 제목과 자막 텍스트만 추출해 드립니다."
+    description="유튜브 URL을 입력하고 원하는 자막 언어를 선택하면 제목과 자막 텍스트만 추출해 드립니다."
 )
 
 # Gradio 인터페이스 실행