import gradio as gr
import requests
import os

def generate(forr, ver=15, doptb=""):
    if not forr or not ver:
        return "Сначала заполните обязательные поля! Потом здесь будут найденые нейросетью модули."

    system_role = """Ты - поисковик магиск андроид модулей, ищешь на 4PDA (Русский форум, https://4pda.to/forum/index.php?showtopic=915158&view=findpost&p=76261279 в спойлере "Алфавитный Каталог модулей", смотрешь ищешь потходящее, потом смотришь страницы, посты нужных и сверяешь характеристики пользователя, если он предоставил. Если всё отлично, то выдавай ссылку на пост (не на страницу, а прям на пост) модуля с 4pda ИЗ СПИСКА (а не когда уже перешёл на пост)) и GitHub (https://github.com/topics/magisk-modules), название, репутация и описывай функционал кратко.

! Пиши **ТОЛЬКО МОДУЛИ MAGISK**, не в коем случае не давай ссылки на темы сторонних приложений, которые не модули Magisk!
! Не в коем случае не пиши ссылки на сторонние ресурсы!
! Не пиши ссылку на страницу! Пользователю нужна ссылка на ПОСТ. Ссылка, которая указана прям в списке модулей.

**Обращай внимание на репутацию! Модули с отрицательной репутацией либо не выдавай пользователю, либо предупреждай.**
"""
    user_msg = f"Модуль для: '{forr}', совместимый с андроид: '{ver}', дополнительно: '{doptb}'"
    
    headers = {
        'Content-Type': 'application/json',
    }

    payload = {
        'messages': [{'role': 'system', 'content': system_role}, {'role': 'user', 'content': user_msg}],
        'max_tokens': 4500,
        'model': "searchgpt"
    }

    try:
        response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload, timeout=250)
        response.raise_for_status()
        data = response.json()

        # Улучшенная обработка ответа
        if not isinstance(data, dict):
            return "Некорректный формат ответа от сервера."

        if 'choices' in data and isinstance(data['choices'], list) and len(data['choices']) > 0:
            choice = data['choices'][0]
            if 'message' in choice and isinstance(choice['message'], dict):
                content = choice['message'].get('content')
                if content and isinstance(content, str):
                    return content.strip()
                else:
                    return "Сервер вернул пустой или некорректный ответ."
            else:
                return "Отсутствует поле 'message' в ответе."
        elif 'error' in data:
            error_msg = data.get('error', {}).get('message', 'Неизвестная ошибка сервера')
            return f"Ошибка сервера: {error_msg}"
        else:
            return "Неизвестный формат ответа от сервера."
    except requests.exceptions.RequestException as e:
        return f"Ошибка запроса: {str(e)}"
    except Exception as e:
        return f"Неожиданная ошибка: {str(e)}"

# UI (остаётся без изменений)
css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
response = requests.get(css_url)
css = response.text + ".gradio-container{max-width: 700px !important} h1, h3 {text-align:center}"

with gr.Blocks(css=css) as demo:
    gr.Markdown("# Поиск Magisk модулей\n### На 4pda, Rooni_Studio")
    with gr.Column():
        with gr.Row():
            forr = gr.Textbox(show_label=True, placeholder="Модуль для... / Название", label="Функционал модуля", interactive=True, container=True)
        with gr.Row():
            ver = gr.Number(show_label=True, label="Версия Android", interactive=True, container=True, value=15)
        with gr.Row():
            doptb = gr.Textbox(show_label=True, placeholder="Пример: У меня версия магиск 28.1, использую прошивку ColorOs на OnePlus 13, ...", label="Дополнительно")
    with gr.Row():
        text_button = gr.Button("Искать", variant='primary')
    with gr.Row():
        with gr.Column():
            text_output = gr.Markdown(show_label=False, value="Здесь будут найденые нейросетью модули.", container=True)
    gr.HTML("""<p style="color: red; padding: 5px;">
  <b style="color: red;">Дисклеймер:</b> Для поиска используется нейросеть, результаты могут быть неточными или неполными.
</p>
""")

    text_button.click(generate, inputs=[forr, ver, doptb], outputs=[text_output], concurrency_limit=250)

demo.queue(max_size=250, api_open=False).launch()