| from backend.chain import chainRespond |
| import gradio as gr |
|
|
| from backend.translator import translateRespond |
| from backend.models import models |
|
|
| """ |
| For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface |
| """ |
| translatorTab = gr.ChatInterface( |
| translateRespond, |
| chatbot=gr.Chatbot(placeholder="<strong>Smart Translator</strong>"), |
| textbox=gr.Textbox(placeholder="Enter text to translate", container=False, scale=7), |
| theme="soft", |
| retry_btn=None, |
| undo_btn=None, |
| clear_btn=None, |
| additional_inputs=[ |
| gr.Dropdown(choices=models.keys(), value = "gpt-4o-mini", label="Models"), |
| gr.Dropdown(choices=["Chinese", "English", "Spanish", "French", "German", "Japanese", "Korean", "Russian", "Turkish"], value = "Chinese", label="Lnaguage"), |
| gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"), |
| gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"), |
| ], |
| ) |
|
|