Update app.py

app.py

@@ -1,5 +1,6 @@
 from pytube import YouTube
 from google.cloud import speech_v1p1beta1 as speech
+from pydub import AudioSegment  # 오디오 변환을 위한 라이브러리
 import io
 import openai
 import os
@@ -24,12 +25,18 @@ def get_yt_video_id(url):
     else:
         raise ValueError("유효한 유튜브 링크가 아닙니다.")
 
-# 오디오 추출 함수
-def download_audio(youtube_url):
+# 오디오 추출 및 변환 함수 (WAV 형식으로 변환)
+def download_and_convert_audio(youtube_url):
     yt = YouTube(youtube_url)
     stream = yt.streams.filter(only_audio=True).first()
     audio_path = stream.download(filename="audio.mp4")
-    return audio_path
+    
+    # 오디오 파일을 WAV로 변환 (16000Hz 샘플 레이트)
+    audio = AudioSegment.from_file(audio_path)
+    wav_audio_path = "converted_audio.wav"
+    audio.set_frame_rate(16000).export(wav_audio_path, format="wav")
+    
+    return wav_audio_path
 
 # Google Speech-to-Text API를 사용하여 오디오를 텍스트로 변환
 def speech_to_text(audio_path):
@@ -40,8 +47,8 @@ def speech_to_text(audio_path):
 
     audio = speech.RecognitionAudio(content=content)
     config = speech.RecognitionConfig(
-        encoding=speech.RecognitionConfig.AudioEncoding.ENCODING_UNSPECIFIED,
-        sample_rate_hertz=16000,
+        encoding=speech.RecognitionConfig.AudioEncoding.LINEAR16,
+        sample_rate_hertz=16000,  # 16000Hz 샘플 레이트
         language_code="ko-KR"  # 한국어 인식
     )
 
@@ -70,8 +77,8 @@ def textToSummary(text):
 # 전체 요약 프로세스를 처리하는 함수
 def summarize(url):
     try:
-        # 유튜브 오디오 다운로드
-        audio_path = download_audio(url)
+        # 유튜브 오디오 다운로드 및 변환
+        audio_path = download_and_convert_audio(url)
         # 음성을 텍스트로 변환
         transcript = speech_to_text(audio_path)
         # 텍스트 요약