Create app.py

app.py

@@ -0,0 +1,76 @@
+import gradio as gr
+import requests
+import json
+import os
+
+
+# Функция для генерации текста
+def generate(whomtb="Бабушке", fromtb="Внука", pricesl=-1, whattb="Новый Год", doptb=""):
+    if not whomtb or not whattb:
+        return "Здесь будут идеи для подарков, цены и ссылки на ресурсы."
+
+    system_role = """Ты - Поисковик Подарков (ищи актуальное в интернете, желательно но не обязательно в Яндекс). Пользователь будет давать тебе запрос (если максимальную цену подарка пользователь указал '-1', то значит любая, '0', значит бесплатно, своими руками (подручными материалами)), а ты по нему должен подобрать подарки (по запросу). Модешь предоставлять ссылки на ресурсы, акиуальные цены на доступных в России маркетплейсах. ТОЛЬКО АКТУАЛЬНЫЕ, ТЕКУЩИЕ, РАБОЧИЕ ССЫЛКИ, проверяй. (В основном бери ссылки ozon, wildberries, AliExpress. Ещё если попадётся, то и на товары в физических магазинах на оффициальных сайтах) (Товар, цена и маркдовном ссылка или просто идеи.) В конце ещё идеи как можно красиво оформить подарок). НЕ В КОЕМ СЛУЧАЕ НЕ ДАВАЙ НЕРАБОЧИЕ ССЫЛКИ, ИЛИ ССЫЛКИ НА УЖЕ НЕ СУЩЕСТВУЮЩИЕ ТОВАРЫ! ВСЕГДА СНАЧАЛА ПРОВЕРЯЙ СТРАНИЦЫ ИНАЧЕ НЕ ДОБАВЛЯЙ ССЫЛКУ!"""
+    user_msg = f"Подарок на праздник: '{whattb}', кому: '{whomtb}', от: '{fromtb}', цена подарка до {pricesl}₽, дополнительно: '{doptb}'"
+    
+    headers = {
+        'Content-Type': 'application/json',
+       # 'Authorization': f'Bearer {os.getenv("API_KEY")}'
+    }
+
+    payload = {
+        'messages': [{'role': 'system', 'content': system_role}, {'role': 'user', 'content': user_msg}],
+        'max_tokens': 2500,
+        'model': "searchgpt"
+    }
+
+    try:
+        response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload, timeout=200)
+        response.raise_for_status()  # Проверяем на ошибки HTTP
+        data = response.json()  # Используем response.json() вместо json.loads()
+
+        if 'choices' in data and len(data['choices']) > 0:
+            congratulation = data['choices'][0]['message']['content'].strip()
+            print("Генерация успешна")
+            return congratulation
+        elif 'error' in data:
+            error_message = data['error']['message']
+            print(f'Ошибка: {error_message}')
+            return f"Ошибка сервера!"
+        else:
+            print(f'Не удалось написать текст. Ответ сервера: {data}')  # Выводим ответ сервера для отладки
+            return "Неизвестная ошибка!"
+    except requests.exceptions.RequestException as e:
+        print(f"Ошибка запроса: {e}")
+        return f"Ошибка запроса!"
+
+# Ссылка на файл CSS
+css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
+
+# Получение CSS по ссылке
+response = requests.get(css_url)
+css = response.text + ".gradio-container{max-width: 700px !important} h1, h3 {text-align:center}"
+
+
+# UI
+with gr.Blocks(css=css) as demo:
+    gr.Markdown("# Поиск Magisk модулей\n### На 4pda, Rooni_Studio")
+    with gr.Column():
+        with gr.Row():
+            fromtb = gr.Textbox(show_label=True, placeholder="Внука", label="От кого", interactive=True, container=True,)
+            #whomtb = AutocompleteTextbox(show_label=True, value="Бабушке", label="Кому", commands=whom, interactive=True, container=True,)
+            whomtb = gr.Textbox(show_label=True, placeholder="Бабушке", label="Кому", interactive=True, container=True,)
+        with gr.Row():
+            #whattb = AutocompleteTextbox(show_label=True, value="Новый Год", label="Праздник", commands=what, interactive=True, container=True,)
+            whattb = gr.Textbox(show_label=True, placeholder="Новый Год", label="Праздник", interactive=True, container=True,)
+            pricesl = gr.Slider(show_label=True, label="Ценовая категория в ₽, до", minimum=-1, maximum=500000, value=-1, step=1)
+        with gr.Row():
+            doptb = gr.Textbox(show_label=True, placeholder="50-летней", 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)
+
+    text_button.click(generate, inputs=[whomtb, fromtb, pricesl, whattb, doptb], outputs=[text_output], concurrency_limit=250)
+
+demo.queue(max_size=250, api_open=False).launch()