GPT-Playground / app.py
Serg4451D's picture
Update app.py
71de207 verified
raw
history blame
1.27 kB
import gradio as gr
from g4f.client import Client
# Создаем экземпляр клиента
client = Client()
# Функция для общения с выбранной моделью
def chat_with_gpt(user_input, selected_model):
try:
response = client.chat.completions.create(
model=selected_model,
messages=[{"role": "user", "content": user_input}]
)
return response.choices[0].message.content
except Exception as e:
return str(e)
# Создаем интерфейс Gradio
with gr.Blocks() as demo:
gr.Markdown("# Chat with GPT Models")
user_input = gr.Textbox(label="You:", placeholder="Type your message here...")
# Выпадающий список для выбора модели
model_selector = gr.Dropdown(
label="Select Model:",
choices=["gpt-4o", "gpt-4", "gpt-3.5-turbo", "gpt-4o-mini", "o1", "o1-mini"],
value="gpt-4o" # Значение по умолчанию
)
output = gr.Textbox(label="Response:", interactive=False)
submit_button = gr.Button("Send")
submit_button.click(chat_with_gpt, inputs=[user_input, model_selector], outputs=output)
# Запускаем интерфейс
demo.launch()