import gradio as gr
import openai
import os

# Установите свои учетные данные API от OpenAI
openai.api_key = os.getenv("API_KEY")

# Установите базовый URL
base_url = os.getenv("BASE_URL")

def generate_solution(prompt):
    # Используйте GPT-3.5-turbo для генерации решения
    response = openai.Completion.create(
        engine="text-davinci-003",  # Выберите движок, поддерживающий GPT-3.5-turbo
        prompt=prompt,
        max_tokens=150  # Установите максимальное количество токенов в ответе
    )
    return response.choices[0].text.strip()

# Определите интерфейс Gradio
iface = gr.Interface(
    fn=generate_solution,
    inputs="text",  # Ввод текста
    outputs="text",  # Вывод текста
    live=True,
    theme="huggingface",  # Используйте тему Hugging Face
    layout="vertical",
    title="GPT-3.5-turbo Решение Генератор",
    description="Загрузите фото, введите промпт и нажмите на кнопку 'Решить' для генерации решения.",
)

# Запустите приложение Gradio
iface.launch(share=True, base_url=base_url)