import gradio as gr
from feifeilib.feifeichat import feifeichat

FeiFei = (
    gr.ChatInterface(
        feifeichat,
        type="messages",
        multimodal=True,
        additional_inputs=[
            gr.Checkbox(label="Feifei",value=True),
            gr.Dropdown(
                ["meta-llama/Llama-3.3-70B-Instruct",
                 "CohereForAI/c4ai-command-r-plus-08-2024",
                 "Qwen/Qwen2.5-72B-Instruct",
                 "nvidia/Llama-3.1-Nemotron-70B-Instruct-HF",
                 "NousResearch/Hermes-3-Llama-3.1-8B",
                 "mistralai/Mistral-Nemo-Instruct-2411",
                 "microsoft/phi-4"],
                value="mistralai/Mistral-Nemo-Instruct-2411",
                show_label=False,
                container=False),
            gr.Radio(
                ["pixtral","Vsiion"],
                value="pixtral",
                show_label=False,
                container=False)
        ],
    )
)
FeiFei.launch()