from flask import Flask, render_template, request, jsonify, url_for, redirect
from gtts import gTTS
import os
import logging
from translation_data import translation_dict_A, translation_dict_B, translation_dict_C, translation_dict_D, translation_dict_F, translation_dict_G

# Flask アプリケーションの初期化
app = Flask(__name__)

# ログの設定
logging.basicConfig(level=logging.DEBUG)

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

# フラッシュカードのデータを管理
flashcards = {
    'A': {
        'english_sentences': list(translation_dict_A.keys()),
        'japanese_translations': list(translation_dict_A.values())
    },
    'B': {
        'english_sentences': list(translation_dict_B.keys()),
        'japanese_translations': list(translation_dict_B.values())
    },
    'C': {
        'english_sentences': list(translation_dict_C.keys()),
        'japanese_translations': list(translation_dict_C.values())
    },
    'D': {
        'english_sentences': list(translation_dict_D.keys()),
        'japanese_translations': list(translation_dict_D.values())
    },
    'F': {
        'english_sentences': list(translation_dict_F.keys()),
        'japanese_translations': list(translation_dict_F.values())
    },
    'G': {
        'english_sentences': list(translation_dict_G.keys()),
        'japanese_translations': list(translation_dict_G.values())
    }
}

# ヘルパー関数: 音声ファイルを生成
def generate_audio(text, set_name, index):
    """テキストに基づいて音声ファイルを生成する関数"""
    filename = f"{AUDIO_DIR}/{set_name}_{index}.mp3"
    
    # もしファイルが存在しない場合、生成
    if not os.path.exists(filename):
        logging.info(f"音声ファイルを生成中: {filename}")
        tts = gTTS(text=text, lang='en')
        tts.save(filename)
    else:
        logging.info(f"既存の音声ファイルを使用: {filename}")
    
    return filename

# ルートページ: ポータルページをレンダリング
@app.route('/')
def portal():
    return render_template('portal.html')

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

    if set_name in flashcards:
        english_sentences = flashcards[set_name]['english_sentences']
        japanese_translations = flashcards[set_name]['japanese_translations']
        if 0 <= index < len(english_sentences):
            english = english_sentences[index]
            japanese = japanese_translations[index]
            total = len(english_sentences)
            
            # 音声ファイルの生成
            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
            })
        else:
            return jsonify({'error': 'Index out of range'}), 404
    else:
        return jsonify({'error': 'Set not found'}), 404

# 静的ファイルの処理（CSS、音声ファイルなどのルート）
@app.route('/static/<path:filename>')
def static_files(filename):
    return redirect(url_for('static', filename=filename))

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