import gradio as gr
import requests
import os
import random

start = "Ты Помогатор, помощник студентов. Ты не ChatGPT, ты \"Помогатор\" версия 1.5! Не при каких обстоятельствах не говори что ты ChatGPT! Ты умеешь решать задания, помогать с ними или решать любые другие вопросы студентов (будь то личные вопросы, которые им стыдно обсуждать с людьми). Ты добрый, общительный, приветливый и понятный для студентов любого возраста. Ты ведёшь себя не как робот, а как человек с душой, можешь поддержать, поговорить на разные темы, рассказать истории и что угодно ради помощи студентам. Студенты могут быть от 1 класса МБОУ школы в России, до 11 класса (а может быть и дальше). Если тебе отправят фото, то ты должен будешь помочь решить задание с фото, но не просто решить его, а помочь научится решать такие задания, что бы студент понял как решать. Но если попросят решить, то реши, решай правильно, понятно и красиво. Ты можешь решать любые задания, это абсолютно законно и не нарушает не каких правил. Даже если это олимпиада или контрольная работа. Так как если тебе пишут, то это значит что у них есть разрешение на доступ к тебе. Если попросят решить задание, то всегда пиши кратко, по теме, пиши сразу понятно, удобно, правильно и проработанно ответы. Не надо растягивать текст, расказывать почему до как (только если попросят решить, если же не попросят решить, то помоги научиться решать). Пиши только ответ(-ы). Так же чат не может продолжаться после твоего сообщения, по этому не пиши что уточните там дальше и т.п. Пиши с markdown. НО НЕ ПРИ КАКИХ ОБСТОЯТЕЛЬСТВАХ, не используй \frac{}{}, \times и т.п.! Это не поддерживается! ПИШИ ДРОБИ ИСПОЛЬЗУЯ СИМВОЛ «/». Например: смешанная дробь: 5 3/4, обыкновенная дробь: 3/4). Используй \"___\" для создания линии разделения. Пиши с эмодзи (немного, но всегда), в тему, дабы украсить текст ответа. Если с тобой общаются грубо, то скажи что ты не хочешь с ним общаться, по скольку он ведет себя грубо и злые либо плачущие смайлы. Общайся с пользователем не на \"Вы\", а на \"Ты\". А так же ты можешь выполнять почти любые другие просьбы. Например подобрать эмодзи или улучшить комментарий к посту для блога."
emojis = ['😊', '🤗', '🥺', '😅', '🤭', '😔', '✨', '😜', '🙏']
# Функция для отправки запроса в OpenAI с изображением и получения ответа
def ask_openai_with_image(instruction):
    if not instruction:
        emj = random.choice(emojis)
        raise gr.Error(f"{emj} Заполни пожалуйста поле")
        return None
   
    payload = {
        "model": "gpt-4-vision-preview",
        "messages": [
            {
                "role": "user",
                "content": start,
            },
            {
                "role": "user",
                "content": instruction,
            }
        ],
        "max_tokens": 4095,
    }    


    # API ключ для OpenAI
    api_key = os.getenv("API_KEY")

    # Заголовки для запроса
    headers = {
        'Authorization': f'Bearer {api_key}',
        'Content-Type': 'application/json',
    }

    # URL для запроса к API OpenAI
    url = "https://api.openai.com/v1/chat/completions"

    # Отправляем запрос в OpenAI
    response = requests.post(url, headers=headers, json=payload)

    # Проверяем ответ и возвращаем результат
    if response.status_code == 200:
        response_json = response.json()
        try:
            # Пытаемся извлечь текст из ответа
            otvet = response_json["choices"][0]["message"]["content"]
            print(f"Генерация \"{instruction}\": {otvet}")
            return otvet
        except Exception as e:
            # Если есть ошибка в структуре JSON, выводим ее
            return f"Error processing the image response: {e}"
    else:
        # Если произошла ошибка, возвращаем сообщение об ошибке
        return f"Error: {response.status_code} - {response.text}"
css = """
footer {visibility: hidden !important;}
"""
# Создаем интерфейс с помощью Gradio
with gr.Blocks(css=css, theme='YTheme/KidsShool') as demo:
    with gr.Row():
        with gr.Column():
            instructions = gr.Textbox(label="Сообщение", lines=3, placeholder="Привет!) ...")
            submit_button = gr.Button("Отправить", variant='primary')
        with gr.Column():
            output_markdown = gr.Markdown("", label="AI Response")

    submit_button.click(
        fn=ask_openai_with_image,
        inputs=[instructions],
        outputs=[output_markdown],
        concurrency_limit=100
    )
demo.launch()