# imports
import gradio as gr
import requests
import json
import os


# functions
def generate(description):
    if not description:
        return None, None

    headers = {
        'Content-Type': 'application/json',
        'Authorization': f'Bearer {os.getenv("API_KEY")}'
    }

    payload = {
        'messages': [{'role': 'system', 'content': f'Ты - корректор текста. Пользователь будет отправлять тебе сообщения, а ты должен исправлять в них все ГРАММАТИЧЕСКИЕ ошибки (в том числе и знаки припенания). Отвечать ты должен без лишних символов, только исправленный запрос пользователя, без markdown. Не надо писать вроде этого: "Вот ваш исправленный тексст: ...", пиши сразу "...". И не нужно выделять то, что ты исправил, просто напиши всё правильно, без грамматических ошибок. Начинаем, ниже сообщение пользователя'}, {'role': 'user', 'content': description}],
        'max_tokens': 150000,
        'model': "gemini-1.5-pro-latest"
    }

    try:
        response = requests.post(f'{os.getenv("BASE_URL")}v1/chat/completions', headers=headers, json=payload)
        response.raise_for_status()  # Проверяем на ошибки HTTP
        data = response.json()  # Используем response.json() вместо json.loads()

        if 'choices' in data and len(data['choices']) > 0:
            correct = data['choices'][0]['message']['content'].strip()
            print("Генерация успешна")
            return correct
        elif 'error' in data:
            error_message = data['error']['message']
            print(f'Ошибка: {error_message}')
            return f"**Ошибка сервера!**\n\n```\n{error_message}\n```"
        else:
            print(f'Не удалось сгенерировать текст. Ответ сервера: {data}')  # Выводим ответ сервера для отладки
            return "**Неизвестная ошибка!**"
    except requests.exceptions.RequestException as e:
        print(f"Ошибка запроса: {e}")
        return f"**Ошибка запроса!**\n\n```\n{e}\n```", f"Ошибка запроса!\n\n{e}"


# css
css = """
footer {
    visibility: hidden !important;
}

.dark {
    --container-radius: 18px !important;
    --block-border-width: 0px !important;
    --block-radius: 18px !important;
    --border-color-primary: #2c0039 !important;
    --block-background-fill: #202020 !important;
    --body-background-fill: #000000 !important;
    --input-background-fill: #151515 !important;
    --input-radius: 14px !important;
    --checkbox-label-border-width: 0px !important;
    --block-label-border-width: 0px !important;
    --input-border-width: 0px !important;
}

.primary {
    color: #FFF !important;
    font-size: 16px !important;
    padding: 10px 20px !important;
    border: 3px solid rgb(234 181 238) !important;
    border-radius: 15px !important;
    text-decoration: none !important;
    transition: 0.5s ease-in-out !important;
    font-weight: normal !important;
    background: none !important;
}
.primary:hover {
    background: rgb(234 181 238) !important;
    color: #000 !important;
}
div.svelte-19hvt5v {
    border: 0px solid #510067 !important;
    background: #101010;
}
.selected.svelte-1uw5tnk {
    border-color: #510067 !important;
    border-width: 0px !important;
}
.tab-nav.svelte-1uw5tnk {
    border-bottom: 0px solid #510067 !important;
}
.secondary {
    color: #FFF !important;
    font-size: 18px !important;
    border: 3px solid rgb(96 55 103) !important;
    border-radius: 20px !important;
    transition: background-color 0.5s ease-in-out !important;
    width: 100% !important;
    background-color: transparent !important;
    height: 50px !important;
    font-weight: bold !important;
    align-items: center !important;
    justify-content: center !important;
    background: none !important;
}
.secondary:hover {
    background-color: rgb(96 55 103);
}
.svelte-zyxd38 {
    display: none !important;
    visibility: hidden !important
}
"""

# ui
with gr.Blocks(css=css) as demo:
    with gr.Tab("Исправление опечаток"):
        with gr.Row():
            promt = gr.Textbox(show_label=True, label="Запрос", lines=3, placeholder="Привет как делпа ?")
        with gr.Row():
            text_button = gr.Button("Генерация", variant='primary')
    with gr.Row():
        with gr.Tab("Ответ"):
            text_output = gr.Textbox(show_label=False, placeholder="Привет, как дела?")

    text_button.click(generate, inputs=[promt], outputs=[text_output], concurrency_limit=512)

demo.queue(api_open=False).launch()