from translators import translate_text
from gradio import Interface, Dropdown, Textbox, themes
from langdetect import detect

def translate(output_language, input_text ):
    languages_dict = {'Spanish': 'es', 'Bengali': 'bn', 'English': 'en', 'French': 'fr'}
    
    input_text = input_text.lower()
    input_language = detect(input_text)
    
    return input_language, translate_text(input_text, translator="google", 
                             from_language=input_language, 
                             to_language=languages_dict[output_language])

interface = Interface(fn=translate, 
                         inputs=[
                             Dropdown(['Spanish', 'Bengali', 'English', 'French'], 
                                         label = 'Select the translation language first'),
                             Textbox(lines=2, placeholder='Text to translate', 
                                        label='Text')],
                        outputs=[Textbox(label='Initial language detected'), Textbox(label='Translation')],
                        title='Automatic Language Translation Demo by RPI eSolutions Pte Ltd',
                        theme=themes.Default(
                            primary_hue=themes.colors.blue, 
                            secondary_hue=themes.colors.indigo))

interface.launch()