# Standard libraries
import base64
import io
import json
import os
import re
import tempfile
import time
from os import getenv
from typing import Any
from typing import Dict
from typing import IO
from typing import List
from typing import Optional
from typing import Tuple
from typing import Union

# Third-party libraries
import streamlit as st


from var_app import __version__

LANGUAGES_EMOJI = {
    "Afrikaans": "🇿🇦", "Arabic": "🇸🇦", "Armenian": "🇦🇲", "Azerbaijani": "🇦🇿", "Belarusian": "🇧🇾", 
    "Bosnian": "🇧🇦", "Bulgarian": "🇧🇬", "Catalan": "🇪🇸", "Chinese": "🇨🇳", "Croatian": "🇭🇷", 
    "Czech": "🇨🇿", "Danish": "🇩🇰", "Dutch": "🇳🇱", "English": "🇬🇧", "Estonian": "🇪🇪", 
    "Finnish": "🇫🇮", "French": "🇫🇷", "Galician": "🇪🇸", "German": "🇩🇪", "Greek": "🇬🇷", 
    "Hebrew": "🇮🇱", "Hindi": "🇮🇳", "Hungarian": "🇭🇺", "Icelandic": "🇮🇸", "Indonesian": "🇮🇩", 
    "Italian": "🇮🇹", "Japanese": "🇯🇵", "Kannada": "🇮🇳", "Kazakh": "🇰🇿", "Korean": "🇰🇷", 
    "Latvian": "🇱🇻", "Lithuanian": "🇱🇹", "Macedonian": "🇲🇰", "Malay": "🇲🇾", "Marathi": "🇮🇳", 
    "Maori": "🇳🇿", "Nepali": "🇳🇵", "Norwegian": "🇳🇴", "Persian": "🇮🇷", "Polish": "🇵🇱", 
    "Portuguese": "🇵🇹", "Romanian": "🇷🇴", "Russian": "🇷🇺", "Serbian": "🇷🇸", "Slovak": "🇸🇰", 
    "Slovenian": "🇸🇮", "Spanish": "🇪🇸", "Swahili": "🇰🇪", "Swedish": "🇸🇪", "Tagalog": "🇵🇭", 
    "Tamil": "🇮🇳", "Thai": "🇹🇭", "Turkish": "🇹🇷", "Ukrainian": "🇺🇦", "Urdu": "🇵🇰", 
    "Vietnamese": "🇻🇳", "Welsh": "🏴󠁧󠁢󠁷󠁬󠁳󠁿"
}



def load_ui_language(file_path: Optional[str] = "ui_lang_support.json") -> Dict[str, Any]:
    """
    Charge les traductions de l'interface utilisateur à partir d'un fichier JSON.

    Args:
        file_path (Optional[str]): Chemin vers le fichier JSON contenant les traductions.

    Returns:
        Dict[str, Any]: Un dictionnaire contenant les traductions de l'interface utilisateur.
    """
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            return json.load(file)
    except FileNotFoundError:
        print(f"{get_translation('erreur_fichier_non_trouve')} {file_path}")
        return {}
    except json.JSONDecodeError:
        print(f"{get_translation('erreur_lecture_fichier')} JSON decoding error")
        return {}
    except IOError as e:
        print(f"{get_translation('erreur_lecture_fichier')} {e}")
        return {}


def get_translation(key: str) -> str:
    """
    Obtient la traduction pour une clé donnée basée sur la langue d'interface sélectionnée.
    """
    lang = st.session_state.get('interface_language', 'English')
    return translations.get(lang, {}).get(key, key)

# Dictionary to store translations
translations = load_ui_language()

def language_selection_page():
    st.set_page_config(initial_sidebar_state="collapsed")
    st.set_page_config(page_icon="👹")
    st.set_page_config(layout="wide")
    # Fonction de rappel pour mettre à jour la langue temporaire lorsque la sélection change
    def update_temp_language():
        st.session_state['temp_interface_language'] = st.session_state['language_selector']
        
    # Initialiser la langue temporaire si elle n'existe pas
    if 'temp_interface_language' not in st.session_state:
        st.session_state['temp_interface_language'] = st.session_state.get('interface_language', 'English')
    
    
    # Sélecteur de langue avec un callback pour mettre à jour la langue temporaire 
    with st.container(
            border=True, 
            key='lang_choices_container', 
            height=None
        ):

        TITRE_LANGSELMENU=translations[st.session_state['temp_interface_language']].get('selection_de_la_langue', 'Language Selection')
        # Titre basé sur la langue temporaire
        st.header(
            f":blue[{TITRE_LANGSELMENU}]"
            ,divider=True)
        
        selecteur_du_choix, boutton_de_validation = st.columns(2, 
                                vertical_alignment='top',
                                gap="small")

        selected_language = selecteur_du_choix.selectbox(
            translations[st.session_state['temp_interface_language']].get('choix_selection_langue', 'Choose the interface language'),
            options=list(translations.keys()),
            index=list(translations.keys()).index(st.session_state['temp_interface_language']),
            format_func=lambda lang: f"{LANGUAGES_EMOJI.get(lang, '')} {lang}",
            key='language_selector',
            on_change=update_temp_language,
            label_visibility="collapsed"
        )

        # Bouton de confirmation avec traduction
        if boutton_de_validation.button(
            translations[st.session_state['temp_interface_language']].get(
                'confirmer', 'Confirm'), 
            key='confirm_button_ui_lang_choice',
            type="secondary",
            use_container_width=True,
            disabled = False
            ):
                st.session_state.interface_language = selected_language
                st.session_state.init_launch_app = False
                st.switch_page(st.Page("pages/main.py", title="Main page"))
                #st.rerun()
        
    ## Afficher les instructions basées sur la langue temporaire
    ###st.write(translations[st.session_state['temp_interface_language']].get('instruction_selection_langue', 'Please select a language.'))
    



language_selection_page()