import gradio as gr
import requests
import json
import os

def gpt3_turbo(image=None, text=None):
    # Замените 'YOUR_API_KEY' на ваш API ключ от OpenAI
    api_key = os.getenv("API_KEY")
    
    # Формируем данные для отправки запроса
    data = {}
    if image is not None:
        with open(image.name, "rb") as img_file:
            data['image'] = img_file.read()
    if text:
        data['text'] = text

    # Отправляем запрос на API GPT-3.5-turbo
    headers = {
        'Content-Type': 'application/json',
        'Authorization': f'Bearer {api_key}'
    }
    endpoint = os.getenv("BASE_URL")
    response = requests.post(endpoint, headers=headers, json={"prompt": data, "max_tokens": 50})

    if response.status_code == 200:
        result = response.json()
        return result['choices'][0]['text']
    else:
        return "Ошибка при получении ответа от модели."

# Создаем интерфейс Gradio
iface = gr.Interface(
    fn=gpt3_turbo,
    inputs=[
        gr.Image(type="file", label="Выберите изображение (необязательно)"),
        "text"
    ],
    outputs="text",
    title="GPT",
    description="Помощь ученикам, решение заданий"
)

# Запускаем интерфейс
iface.launch()