Update app.py

app.py

@@ -1,19 +1,16 @@
 import gradio as gr
 import requests
-import json
 import os
 
-
-# Функция для генерации текста
 def generate(forr, ver=13, 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), название, репутация и описывай функционал кратко.
 
-! Пиши **ТОЛЬКО МОДУЛИ NAGISK**, не в коем случае не давай ссылки на темы сторонних приложений, которые не модули Magisk!
+! Пиши **ТОЛЬКО МОДУЛИ MAGISK**, не в коем случае не давай ссылки на темы сторонних приложений, которые не модули Magisk!
 ! Не в коем случае не пиши ссылки на сторонние ресурсы!
-! Не пиши ссылку на страницу! Пользователю нцжна ссылка на ПОСТ. Ссылка, которая указана прям в списке модулей.
+! Не пиши ссылку на страницу! Пользователю нужна ссылка на ПОСТ. Ссылка, которая указана прям в списке модулей.
 
 **Обращай внимание на репутацию! Модули с отрицательной репутацией либо не выдавай пользователю, либо предупреждай.**
 """
@@ -21,7 +18,6 @@ def generate(forr, ver=13, doptb=""):
     
     headers = {
         'Content-Type': 'application/json',
-       # 'Authorization': f'Bearer {os.getenv("API_KEY")}'
     }
 
     payload = {
@@ -32,43 +28,43 @@ def generate(forr, ver=13, doptb=""):
 
     try:
         response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload, timeout=250)
-        response.raise_for_status()  # Проверяем на ошибки HTTP
-        data = response.json()  # Используем response.json() вместо json.loads()
+        response.raise_for_status()
+        data = response.json()
+
+        # Улучшенная обработка ответа
+        if not isinstance(data, dict):
+            return "Некорректный формат ответа от сервера."
 
-        if 'choices' in data and len(data['choices']) > 0:
-            content = data['choices'][0]['message'].get('content')
-            if content is not None:
-                congratulation = content.strip()
-                print(f"Генерация успешна: '{forr}' с версией '{ver}'")
-                return congratulation
+        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:
-                print(f'Пустой ответ от сервера для запроса: "{forr}"')
-                return "Сервер вернул пустой ответ. Попробуйте изменить параметры запроса."
+                return "Отсутствует поле 'message' в ответе."
         elif 'error' in data:
-            error_message = data['error']['message']
-            print(f'Ошибка: {error_message}')
-            return f"Ошибка сервера!"
+            error_msg = data.get('error', {}).get('message', 'Неизвестная ошибка сервера')
+            return f"Ошибка сервера: {error_msg}"
         else:
-            print(f'Не удалось написать текст: "{forr}". Ответ сервера: {data}')  # Выводим ответ сервера для отладки
-            return "Неизвестная ошибка!"
+            return "Неизвестный формат ответа от сервера."
     except requests.exceptions.RequestException as e:
-        print(f"Ошибка запроса: {e}")
-        return f"Ошибка запроса!"
+        return f"Ошибка запроса: {str(e)}"
+    except Exception as e:
+        return f"Неожиданная ошибка: {str(e)}"
 
-# Ссылка на файл CSS
+# UI (остаётся без изменений)
 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():
-            forr = gr.Textbox(show_label=True, placeholder="Модуль для... / Название", label="Функционал модуля", interactive=True, container=True,)
+            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=13)
         with gr.Row():