import openai
import os
import gradio as gr

# Получаем ключ API из переменной окружения
openai.api_key = os.environ.get("OPENAI_API_KEY")

# Основная функция общения с ChatGPT
def chatgpt_prompt(prompt):
    try:
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=[{"role": "user", "content": prompt}]
        )
        return response['choices'][0]['message']['content']
    except Exception as e:
        return f"Ошибка ChatGPT: {str(e)}"

# Интерфейс Gradio
iface = gr.Interface(
    fn=chatgpt_prompt,
    inputs=gr.Textbox(label="Введите ваш запрос"),
    outputs=gr.Textbox(label="Ответ от ChatGPT"),
    title="Интерфейс ChatGPT",
    description="Пример взаимодействия с API OpenAI через Hugging Face Space"
)

# Запуск
iface.launch()