Update app.py

app.py

@@ -1,8 +1,11 @@
 import gradio as gr
-from transformers import MarianMTModel, MarianTokenizer
-from gtts import gTTS
+import webbrowser
 import sqlite3
 from datetime import datetime
+from googletrans import Translator
+from gtts import gTTS
+import playsound
+import pykakasi
 import os
 import pandas as pd
 
@@ -23,46 +26,10 @@ def init_db():
     conn.commit()
     conn.close()
 
-# 모델과 토크나이저 로드
-ko_en_model_name = "Helsinki-NLP/opus-mt-ko-en"  # 한국어 -> 영어
-ko_en_tokenizer = MarianTokenizer.from_pretrained(ko_en_model_name)
-ko_en_model = MarianMTModel.from_pretrained(ko_en_model_name)
-
-en_jp_model_name = "Helsinki-NLP/opus-mt-en-jap"  # 영어 -> 일본어
-en_jp_tokenizer = MarianTokenizer.from_pretrained(en_jp_model_name)
-en_jp_model = MarianMTModel.from_pretrained(en_jp_model_name)
-
-# 번역 기능 구현
-def translate_ko_to_jp(keyword):
-    # 1단계: 한국어 -> 영어
-    tokens = ko_en_tokenizer(keyword, return_tensors="pt", padding=True)
-    translated_tokens = ko_en_model.generate(**tokens)
-    translated_text_en = ko_en_tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
-    
-    # 2단계: 영어 -> 일본어
-    tokens = en_jp_tokenizer(translated_text_en, return_tensors="pt", padding=True)
-    translated_tokens = en_jp_model.generate(**tokens)
-    translated_text_jp = en_jp_tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
-    
-    return translated_text_jp
-
-# 검색 기능과 검색 기록 저장
-def search(keyword):
-    if keyword:
-        translated_text = translate_ko_to_jp(keyword)
-        url = f"https://www.irasutoya.com/search?q={translated_text}"
-        
-        # 검색 기록 저장
-        save_search_history(keyword, translated_text, url)
-        
-        return f"검색어: {keyword}\n일본어 번역: {translated_text}\n검색 결과 링크: {url}", url
-    else:
-        return "검색어를 입력하세요.", ""
-
-# 검색 기록 저장
 def save_search_history(keyword, translation, link):
     home_dir = os.path.expanduser("~")
     db_path = os.path.join(home_dir, 'search_history.db')
+
     conn = sqlite3.connect(db_path)
     cursor = conn.cursor()
     current_date = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -71,56 +38,52 @@ def save_search_history(keyword, translation, link):
     conn.commit()
     conn.close()
 
-# 검색 기록 로드
-def load_history():
-    home_dir = os.path.expanduser("~")
-    db_path = os.path.join(home_dir, 'search_history.db')
-    conn = sqlite3.connect(db_path)
-    cursor = conn.cursor()
-    cursor.execute('SELECT * FROM searches ORDER BY date DESC')
-    history = cursor.fetchall()
-    conn.close()
-    
-    df = pd.DataFrame(history, columns=['ID', '날짜', '검색어', '일본어 번역', '링크'])
-    return df
+# Kakasi 설정
+kakasi = pykakasi.kakasi()
 
-# 일본어 음성 재생 기능
-def play_japanese_audio(japanese_text):
-    tts = gTTS(japanese_text, lang='ja')
-    audio_file = "japanese_audio.mp3"
-    tts.save(audio_file)
-    return audio_file
+# 검색 함수
+def search(keyword):
+    translator = Translator()
+    japanese_keyword = translator.translate(keyword, src='auto', dest='ja').text
+    url = f"https://www.irasutoya.com/search?q={japanese_keyword}"
+    romaji = ''.join([item['hepburn'] for item in kakasi.convert(japanese_keyword)])
 
-# Gradio 인터페이스 설정
-def gradio_search_interface(keyword):
-    result, url = search(keyword)
-    return result
+    save_search_history(keyword, japanese_keyword, url)
 
-def gradio_audio_interface(japanese_text):
-    audio_file = play_japanese_audio(japanese_text)
-    return audio_file
+    result_text = f"검색어: {keyword}\n일본어 번역: {japanese_keyword} ({romaji})\n검색 결과 링크: {url}"
+    return result_text, japanese_keyword, url
 
-def gradio_history_interface():
-    history = load_history()
-    return history
+# 일본어 음성 재생
+def play_japanese_audio(japanese_text):
+    tts = gTTS(japanese_text, lang='ja')
+    tts.save("japanese_audio.mp3")
+    playsound.playsound("japanese_audio.mp3")
+    os.remove("japanese_audio.mp3")
+    return "일본어 음성이 재생되었습니다."
 
-# Gradio Blocks 사용하여 인터페이스 구성
+# Gradio 인터페이스 설정
 with gr.Blocks() as demo:
-    with gr.Tab("검색"):
+    gr.Markdown("## Irasutoya 한글 검색기")
+    with gr.Row():
         keyword_input = gr.Textbox(label="한글 검색어 입력")
         search_button = gr.Button("검색")
-        search_output = gr.Textbox(label="검색 결과")
-        audio_output = gr.Audio(label="일본어 음성 재생")
-        
-        search_button.click(gradio_search_interface, inputs=keyword_input, outputs=search_output)
-        search_button.click(gradio_audio_interface, inputs=keyword_input, outputs=audio_output)
     
-    with gr.Tab("검색 기록"):
-        history_button = gr.Button("기록 불러오기")
-        history_output = gr.Dataframe(label="검색 기록", headers=["ID", "날짜", "검색어", "일본어 번역", "링크"])
-        
-        history_button.click(gradio_history_interface, outputs=history_output)
+    result_output = gr.Textbox(label="검색 결과")
+    play_button = gr.Button("일본어 음성 재생", interactive=False)
+
+    def on_search(keyword):
+        result, japanese_keyword, _ = search(keyword)
+        play_button.interactive = True
+        return result, japanese_keyword
+    
+    def on_play(japanese_text):
+        return play_japanese_audio(japanese_text)
+
+    search_button.click(on_search, [keyword_input], [result_output, play_button])
+    play_button.click(on_play, [play_button], [])
+
+# 데이터베이스 초기화
+init_db()
 
-if __name__ == "__main__":
-    init_db()
-    demo.launch(share=True)
+# 앱 실행
+demo.launch()