from helsinki_models import helsinki_models, get_clearly_formatted_langauge_directions
from iso639_wrapper import get_name_from_iso_code
from language_detection import detect_language
from collections import OrderedDict
from utils import convert_keys_to_lowercase, match_in_keys, match_in_values


def get_all_source_languages():
    """
    Returns a human-readable `dict source_languages_names:codes` 
    based on the available models.
    """
    source_languages = {}
    language_directions = get_clearly_formatted_langauge_directions()

    for direction in language_directions:
        source_lang_code = direction.split('-')[0]
        source_language_name = get_name_from_iso_code(source_lang_code)
        if source_language_name:
            source_languages[source_language_name] = source_lang_code
    source_languages = OrderedDict(sorted(source_languages.items()))
    all_source_langs_including_auto_detect = \
        { **{'Auto Detect' : 'Auto Detect'}, **source_languages}
    return all_source_langs_including_auto_detect

def update_source_languages_dict(source_languages_dict, auto_detected_language):
    source_languages_dict[auto_detected_language] = "Auto Detect"

def get_target_languages(source_language_code, input_text=None):
    """
    Returns a human-readable `dict of target languages names to codes` 
    based on the available models and the source language passed.
    """
    include_all_languages = False
    if source_language_code == "Auto Detect":
        source_language_code, include_all_languages = auto_detect_language_code(input_text)
        
    target_languages = {}
    language_directions = get_clearly_formatted_langauge_directions()
    for direction in language_directions:
        if direction.startswith(f"{source_language_code}-") or include_all_languages:
            target_language = direction.split('-')[1]  # Extracting the last part as the target language
            target_language_name = get_name_from_iso_code(target_language)
            if target_language_name:
                target_languages[target_language_name] = target_language
    return OrderedDict(sorted(target_languages.items())), source_language_code

def auto_detect_language_code(input_text):
    DEFAULT_SOURCE_LANGUAGE = "en"
    detected_language_string = DEFAULT_SOURCE_LANGUAGE
    if not input_text:
        return DEFAULT_SOURCE_LANGUAGE, True
    language_or_code = detect_language(input_text)
    if language_or_code == "unknown":
        return DEFAULT_SOURCE_LANGUAGE, True
    else:
        detected_language_string = match_in_keys(get_all_source_languages(), language_or_code)
        if not detected_language_string:
            detected_language_string = match_in_values(get_all_source_languages(), language_or_code)
        if detected_language_string:
            return detected_language_string, False
        else:
            return DEFAULT_SOURCE_LANGUAGE, True
    

# Example usage:
# all_source_languages = get_all_source_languages()
# print("All Source Languages:", all_source_languages)

# source_language_code = "pa"  # Replace with your desired source language
# target_languages = get_target_languages(source_language_code)
# print(f"Target Languages for {source_language_code}:", target_languages)