import requests
import json

def translate_text(text, target_lang='ja', source_lang='en'):
    api_key = "AIzaSyAEzK5_n6zKTimD9yoXS-C8O0xN_4LaVBQ"
    # 元の改行を `<br>` で置き換え
    text = text.replace('\n', '<br>')
    url = f"https://translation.googleapis.com/language/translate/v2?key={api_key}"
    payload = {
        'q': text,
        'source': source_lang,
        'target': target_lang,
        'format': 'html'  # フォーマットを 'html' に変更
    }
    headers = {
        'Content-Type': 'application/json'
    }

    response = requests.post(url, headers=headers, json=payload)
    if response.status_code == 200:
        translated_text = json.loads(response.text)['data']['translations'][0]['translatedText']
        # 翻訳されたテキストの `<br>` を改行に戻す
        translated_text = translated_text.replace('<br>', '\n')
        
        # "##選択肢" と "【正解選択肢】" の間に改行を確実に挿入
        translated_text = translated_text.replace("##選択肢【正解選択肢】", "##選択肢\n【正解選択肢】")

        return translated_text
    else:
        return "Error: " + response.text