File size: 3,461 Bytes
c32697e 26f365f c32697e d35de09 b70ba64 c32697e 94ef169 c32697e 26f365f c32697e 437c02a dcd3020 c32697e 83d5183 c32697e 9713d52 b1da8e3 9713d52 c32697e 94ef169 c32697e d35de09 25ad790 c32697e 25ad790 c743c6d 7dbc3ec 25ad790 7dbc3ec 14a0b3e c32697e 9713d52 904a7c2 9713d52 c32697e 94ef169 c32697e 29b5120 1571261 29b5120 1571261 24f7854 c32697e 9713d52 087b372 9713d52 c32697e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
import gradio as gr
from backend.text_to_tgt import src_txt_to_eng_translator, formatted_languages
from backend.audio_to_tgt import src_audio_to_eng_translator
from backend.image_to_tgt import src_image_to_eng_translator, pytesseract_language_dict
# from backend.video_to_tgt import src_video_to_eng_translator
heading_txt = "Text-to-TranslatedText"
description_txt = '''Enter text in any language, and get the translation in any language.'''
language_list = formatted_languages.keys()
txt_interface = gr.Interface(
fn=src_txt_to_eng_translator,
inputs=[
gr.Textbox(label="Text Input"),
gr.Dropdown(choices=language_list, label="Select Target Language", interactive=True)
],
outputs=[
gr.Textbox(label="Translation"),
gr.Textbox(label="Source Language")
],
title=heading_txt,
description=description_txt,
examples=[
["Bonjour, comment ça va ?", "Hindi"],
["Привет, как дела?", "English"],
["Hola, ¿cómo estás?", "Odia (Oriya)"],
["你好，你怎么样", "English"],
["Guten Tag! Wie geht's dir?", "English"]
]
)
heading_image = "Image-to-TranslatedText"
description_image = "Upload an image to extract text and translate it to any language. Make sure to choose language in 'Select Language'"
sorted_languages = sorted(pytesseract_language_dict.keys())
image_interface = gr.Interface(
fn=src_image_to_eng_translator,
inputs=[
gr.Image(label="Upload an Image", type="filepath"),
gr.Dropdown(choices=sorted_languages,
label="Select Image Text Language",
),
gr.Dropdown(
choices=language_list,
label="Select Target Language",
interactive=True
)
],
outputs=[
gr.Textbox(label="Image Text"),
gr.Textbox(label="Translated Text"),
],
title="Image Text Extractor and Translator",
description=description_image,
examples=[
["examples/images/hindi_image_sample.jpg", "Hindi", "English"],
["examples/images/odia_sample_image.png", "Odia", "English"],
["examples/images/russian_sample_image.png", "Russian", "English"]
]
)
heading_audio = "Audio-to-TranslatedText"
description_audio = "Upload an audio file to extract text and translate it to any language. Takes too much time without GPU."
audio_interface = gr.Interface(
fn=src_audio_to_eng_translator,
inputs=[
gr.Audio(
label="Upload an Audio file",
type="filepath"
),
gr.Dropdown(
choices=["turbo", "base", "tiny", "small", "medium", "large"],
label="Select Whisper Model size",
),
gr.Dropdown(
choices=language_list,
label="Select Target Language",
interactive=True
)
],
outputs=[gr.Textbox(label="Original text"),
gr.Textbox(label="Translated text"),
gr.Textbox(label="Original Language")],
title=heading_audio,
description=description_audio,
examples=[
["examples/audios/russian_sample_audio.mp3", "turbo", "English"]
]
)
combined_interface = gr.TabbedInterface(
[txt_interface, image_interface, audio_interface],
['Text-to-English', 'Image-to-English', 'Audio-to-English']
)
combined_interface.launch() |