from flask import Flask, render_template, request, redirect, url_for, jsonify
from gtts import gTTS
import os

app = Flask(__name__)

# 音声ファイルの保存ディレクトリ
AUDIO_DIR = 'static/audio'

# リストに変換
english_sentences_A = list(translation_dict_A.keys())
english_sentences_B = list(translation_dict_B.keys())
english_sentences_C = list(translation_dict_C.keys())
english_sentences_D = list(translation_dict_D.keys())
english_sentences_F = list(translation_dict_F.keys())
english_sentences_G = list(translation_dict_G.keys())

# 音声ファイルを生成する関数
def generate_audio(text, set_name, index):
    filename = f"{AUDIO_DIR}/{set_name}_{index}.mp3"
    
    # ファイルが存在しない場合のみ生成
    if not os.path.exists(filename):
        tts = gTTS(text=text, lang='en')
        tts.save(filename)
    return filename

# フラッシュカードの内容をJSONで返す
@app.route('/flashcards')
def index():
    set_name = request.args.get('set', 'A')
    index = int(request.args.get('index', 0))

    if set_name == 'A':
        english = english_sentences_A[index]
        japanese = translation_dict_A[english]
        total = len(english_sentences_A)
    elif set_name == 'B':
        english = english_sentences_B[index]
        japanese = translation_dict_B[english]
        total = len(english_sentences_B)
    elif set_name == 'C':
        english = english_sentences_C[index]
        japanese = translation_dict_C[english]
        total = len(english_sentences_C)
    elif set_name == 'D':
        english = english_sentences_D[index]
        japanese = translation_dict_D[english]
        total = len(english_sentences_D)
    elif set_name == 'F':
        english = english_sentences_F[index]
        japanese = translation_dict_F[english]
        total = len(english_sentences_F)
    elif set_name == 'G':
        english = english_sentences_G[index]
        japanese = translation_dict_G[english]
        total = len(english_sentences_G)

    # 音声ファイルの生成
    audio_url = url_for('static', filename=f"audio/{set_name}_{index}.mp3")
    generate_audio(english, set_name, index)

    # フラッシュカードの情報をJSONで返す
    return jsonify({
        'set_name': set_name,
        'index': index,
        'total': total,
        'english': english,
        'japanese': japanese,
        'audio_url': audio_url
    })

if __name__ == '__main__':
    app.run(debug=True, host="0.0.0.0", port=7860)