team-ai / ui /translator.py
peichao.dong
add langgrah example
7554512
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"),
],
)