import gradio as gr
import numpy as np
import scipy.io.wavfile as wavfile
import soundfile as sf
import time
import uuid
import os
import csv
from supabase import create_client

# Supabase
url = "https://eecucubpmvpjkhqletul.supabase.co"
key = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImVlY3VjdWJwbXZwamtocWxldHVsIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NDQ3MTc5MzUsImV4cCI6MjA2MDI5MzkzNX0.Av-ZEQ2xczudhm2c8p1JioXRXQCf4s0m4X_w5jrkf-8"  # Cẩn thận không chia sẻ publicly!
DB = create_client(supabase_url=url, supabase_key=key)


def local_storage_set(key, value):
    value = json.dumps(value, ensure_ascii=False)  # Convert to JSON string
    st_javascript(f"localStorage.setItem('{key}', {value});")


# Function to get a value from localStorage
def local_storage_get(key):
    return st_javascript(f"localStorage.getItem('{key}');")


def get_transcripts(path: str):
    samples = []
    with open(path, newline='', encoding='utf-8') as csvfile:
        reader = csv.reader(csvfile)
        next(reader)
        for row in reader:
            samples.append(f"{row[0]} - {row[1]}")
    return samples


def get_provinces(path):
    provinces = []
    with open(path, newline='', encoding='utf-8') as csvfile:
        reader = csv.reader(csvfile)
        next(reader)
        for row in reader:
            provinces.append(f"{row[0]}")
    return provinces


def save_data(audio, transcript, age, gender, region):
    if age < 1:
        return "⚠️ Bạn cần nhập đúng số tuổi."

    if not transcript or not gender or not region.strip():
        return "⚠️ Bạn cần nhập đầy đủ các thông tin: Câu thoại, Giới tính và Tỉnh/TP."

    if audio is None:
        return "⚠️ Vui lòng ghi âm trước khi lưu."

    session_id = str(uuid.uuid4())
    audio_array = np.frombuffer(audio[-1], dtype=np.int16)
    filename = f"recorded_audio_{int(time.time())}.wav"
    wavfile.write(filename, 44100, audio_array)

    DB.storage.from_("cs-bucket").upload(file=filename, path=f"upload/{filename}", file_options={"content-type": "audio/wav"})
    url = DB.storage.from_("cs-bucket").get_public_url(path=f"upload/{filename}")

    word = transcript.split("-")[0]
    transcript_text = transcript.split("-")[1]

    DB.table("cs-data").insert({
        "user_id": session_id,
        "audio_url": url,
        "word": word,
        "transcript_text": transcript_text,
        "age": age,
        "gender": gender,
        "region": region.strip()
    }).execute()

    if os.path.exists(filename):
        os.remove(filename)

    return "✅ Dữ liệu đã được lưu thành công. Cảm ơn bạn!"

def on_submit(audio, sentence_selected, age, gender, region):
    full_transcript = sentence_to_full.get(sentence_selected, None)
    if full_transcript is None:
        return "⚠️ Lỗi: Không tìm thấy câu thoại tương ứng."
    return save_data(audio, full_transcript, age, gender, region)


transcripts = get_transcripts("scripts.csv")
provinces = get_provinces("vietnam_provinces.csv")

sentence = []

sentence_to_full = {trans.split(" - ")[-1].strip(): trans for trans in transcripts}

with gr.Blocks(theme=gr.themes.Soft()) as demo:
    gr.Markdown("""
    <h1 style="
        font-size: 32px;
        text-align: center;
    ">
    🎤 Ứng dụng ghi âm và thu thập dữ liệu
    </h1>
    """)

    with gr.Row():
        with gr.Column(scale=3):
            sentence = gr.Dropdown(label="Chọn câu thoại", choices=list(sentence_to_full.keys()))
            age = gr.Number(label="Tuổi")
            gender = gr.Dropdown(label="Giới tính", choices=["Nam", "Nữ"])
            region = gr.Dropdown(label="Tỉnh/TP", choices=provinces)
            audio = gr.Audio(sources="microphone", type="numpy", label="Ghi âm tại đây")
            submit_btn = gr.Button("Lưu dữ liệu")
            output = gr.Textbox(label="Thông báo")
            submit_btn.click(fn=on_submit, 
                             inputs=[audio, sentence, age, gender, region], 
                             outputs=output
                            )

        with gr.Column(scale=2):
            gr.Markdown("<h3 style='font-size: 22px'>📌 Hướng dẫn sử dụng</h3>")
            gr.Markdown("""
            <div style="font-size: 18px">
            <b>Bước 1:</b> Chọn đoạn thoại bạn muốn ghi âm trong hộp gợi ý.<br>
            <b>Bước 2:</b> Điền đầy đủ thông tin (<i>tuổi, giới tính, tỉnh/thành</i>).<br>
            <b>Bước 3:</b> Bấm <b>Start Recording</b> để ghi âm, sau đó bấm <b>Stop</b> và nghe lại.<br>
            <b>Bước 4:</b> Nếu ghi âm bị lỗi hoặc thiếu, bấm <b>Reset</b> để ghi lại.<br>
            <b>Bước 5:</b> Bấm <b>“Lưu dữ liệu”</b> để gửi ghi âm về cho nhóm phát triển.<br><br>

            <span style="color: red">🔊 Lưu ý:</span> Bạn có thể phát âm từ tiếng Anh theo kiểu Việt hóa.<br>
            Ví dụ: <code>ability</code> → <i>ờ bi li ti</i><br><br>
            👉 Hãy cố gắng giúp ghi âm <b>5 câu</b> nếu có thể nhé! Cảm ơn sự giúp đỡ của bạn 
            </div>
            """)

demo.launch()