someshb07's picture
Create app.py
691539b verified
raw
history blame
1.43 kB
def get_fores_code(language):
# Loop through the list of languages
for entry in languages['languages']:
if entry['name'].lower() == language.lower(): # Compare to the 'name' field in the dictionary
return entry['code']
return None # Return None if no match is found
def translate_text(text, destination_language):
# Convert language name to language code using get_fores_code function
dest_code = get_fores_code(destination_language)
# Ensure the code is valid (for debugging purposes)
print(f"Destination language code: {dest_code}")
# Call the translation pipeline with the correct language codes
translation = pipe(text,
src_lang='eng_Latn',
tgt_lang=dest_code) # Pass the actual code (no quotes)
return translation[0]['translation_text']
demo = gr.Interface(fn=translate_text,
inputs= [
gr.TextArea(label='Insert the text in English to translate'),
gr.Dropdown(
label='Select the language to translate',
choices=[lang['name'] for lang in languages['languages']], # Extract the 'name' field
value='English'
)
],
outputs=[gr.TextArea(label='This is the translated text')])
demo.launch()