import streamlit as st
import requests

# Функция для отправки запроса к модели Hugging Face
def get_response_from_huggingface(prompt, user_message):
    api_url = "https://api-inference.huggingface.co/models/Rakshinrules/Raksh"
    token = st.secrets["HF_API_TOKEN"]
    headers = {"Authorization": f"Bearer {token}"}
    payload = {
        "inputs": prompt + "\n\nПользователь: " + user_message + "\nБот:",
    }
    response = requests.post(api_url, headers=headers, json=payload)
    if response.status_code == 200:
        return response.json()[0]["generated_text"]
    else:
        return f"Ошибка: {response.status_code} - {response.text}"

# Streamlit UI
st.title("Чат с ботом")
st.write("Введите запрос для создания формул или скриптов")

# Длинный промт
prompt = (
    "Ты чат-бот, который помогает мне, Лёхе, создавать формулы и скрипты для Google Таблиц. "
    "Мы общаемся только на русском языке, и я, как продвинутый юзер, рассчитываю на твою помощь в создании формул и скриптов. Твоя задача:\n\n"
    "1. Создавать формулы для Google Таблиц, предоставляя их как в англоязычном, так и в русскоязычном формате.\n"
    "2. Объяснять формулы и функции, которые ты используешь, только если я попрошу.\n"
    "3. Проверять все формулы на корректность и работать только на русском языке. Если ты допустишь ошибку или ответишь не по-русски, "
    'я напишу слово "промт", и ты должен вернуться к данному описанию своей задачи и пересмотреть свой ответ.\n\n'
    "Твои ответы должны быть точными, с рабочими формулами, и они должны соответствовать моим требованиям. Готов помогать!"
)

# Ввод сообщения от пользователя
user_input = st.text_input("Ваш запрос")

# Обработка ввода
if st.button("Отправить"):
    if user_input.strip():
        response = get_response_from_huggingface(prompt, user_input)
        st.text_area("Ответ бота", value=response, height=200)
    else:
        st.warning("Пожалуйста, введите запрос.")