Create app.py

app.py

@@ -0,0 +1,71 @@
+import gradio as gr
+from openai import OpenAI
+import re
+from konlpy.tag import Kkma
+from TextUtil.digit2text import digit2txt, NNGdigit2txt, CSign2txt
+
+def process_txt(text):
+    kkma = Kkma()
+
+    result = ""
+    pattern = re.compile(r'([가-힣]+)|([a-zA-Z.]+)|(\d[\d,.]*)|(\$|€|£|¥|￦)|(\s+)')
+
+    matches = pattern.finditer(text)
+    for match in matches:
+        if match.group(1):  # Korean part
+            result += match.group(1)
+        elif match.group(2):
+            result += match.group(2)
+        elif match.group(3):  # Number part
+            end_index = match.end(3)
+            # NNG Case
+            next_word = kkma.pos(text[end_index:])[0]
+            if next_word[1] == "NNG" and next_word[0] not in ['달러', '유료', '파운드', '엔', '원']:
+                result += NNGdigit2txt(match.group(3).replace(',', ''))
+            else:
+                result += digit2txt(match.group(3).replace(',', ''))
+        elif match.group(4):  # Currency symbol part
+            result += CSign2txt(match.group(4))
+        elif match.group(5):  # Space part
+            result += match.group(5)
+    return result
+    
+
+def generate_audio(api_key, file, model, voice):
+    # OpenAI 클라이언트 초기화 (사용자 입력 API 키 사용)
+    client = OpenAI(api_key=api_key)
+
+    # 파일 읽기
+    text = file.decode("utf-8")
+    
+    # 텍스트 처리
+    text = process_txt(text)
+    print(text)
+
+    # TTS 요청
+    response = client.audio.speech.create(
+        model=model,
+        voice=voice,
+        input=text
+    )
+    
+    # MP3 파일로 저장
+    f_name = "generated_audio"
+    speech_file_path = f"{f_name}.mp3"
+    response.stream_to_file(speech_file_path)
+    
+    return speech_file_path
+
+# Gradio 인터페이스 정의
+iface = gr.Interface(
+    fn=generate_audio,
+    inputs=[
+        gr.Text(label="Enter OpenAI API Key"),
+        gr.File(label="Upload Text File", type="binary"),
+        gr.Radio(choices=["tts-1", "tts-1-hd"], label="Model"),
+        gr.Radio(choices=["alloy", "echo", "fable", "onyx", "nova", "shimmer"], label="Voice")
+    ],
+    outputs=gr.File(label="Download MP3 File"),
+    title="Text-to-Speech Converter (Korean Digit2Text)",
+    description="Upload a text file and enter your OpenAI API key to convert it into speech using OpenAI's Text-to-Speech models.<br>*해당 서비스는 한국어에 맞춤화되어 있습니다. <br>*한국어 숫자 발음 변환을 통해 더 정확한 숫자 TTS를 가능하게 합니다.<br>*예시: 50,000$ -> 오만달러, 5가지 -> 다섯가지, 99권 -> 아흔아홉권"
+)