import streamlit as st
import google.generativeai as genai
from db.mongodb import save_file_to_mongodb, fetch_file_from_mongodb, detect_file_type
from powerpoint.pptx import translate_pptx
from excel.xlsx import translate_xlsx
from excel.excel_translate import translate_csv
from word.word_helper import translate_docx
import dotenv
import os

dotenv.load_dotenv(".env")

genai.configure(api_key=os.getenv("GEMINI_API_KEY"))

st.title("Translate Your File Easily! 🌍")

uploaded_file = st.file_uploader("📂 Chọn file để dịch")
source_lang = st.selectbox("🌐 Chọn ngôn ngữ của tài liệu", ["english", "vietnamese"])
target_lang = st.selectbox("🌐 Chọn ngôn ngữ muốn dịch sang", ["english", "vietnamese"])


def process_file(file, file_type):
    progress_bar = st.progress(0)
    file_id, file_name = save_file_to_mongodb(uploaded_file=file, db_name=file_type.lower(), collection_name="root_file")
    progress_bar.progress(20)
    st.write(f"📂 File ID: {file_id}")

    if file_type == "PPTX":
        final_id = translate_pptx(file_id, file_name, source_lang = source_lang, target_lang = target_lang, slides_per_batch=5)
        progress_bar.progress(60)
    elif file_type == "Excel":
        final_id = translate_xlsx(file_id = file_id, file_name = file_name, source_lang = source_lang, target_lang = target_lang)
    elif file_type == "CSV":
        final_id = translate_csv(file_id = file_id, source_lang = source_lang, target_lang = target_lang)
    elif file_type == "Word":
        final_id = translate_docx(file_id = file_id, file_name = file_name , source_lang = source_lang, target_lang = target_lang)
    else:
        st.error("❌ Loại file không hỗ trợ!")
        return

    progress_bar.progress(80)
    st.write("✅ File đã được dịch xong!")
    file_io, file_name = fetch_file_from_mongodb(file_type.lower(), "final_file", final_id)
    progress_bar.progress(100)

    if file_io:
        st.download_button("⬇️ Tải file về", data=file_io.getvalue(), file_name=file_name)
    else:
        st.error("❌ Không thể tải xuống file. Vui lòng thử lại!")

if uploaded_file and st.button("🚀 Upload và dịch ngay!"):
    file_type = detect_file_type(uploaded_file)
    st.write(f"🔍 Loại file phát hiện: {file_type}")
    process_file(uploaded_file, file_type)