Update app.py

app.py

@@ -2,51 +2,70 @@ import os
 import gradio as gr
 import requests
 
+# Загружаем токен и имя модели из переменных окружения
 HF_TOKEN = os.getenv("HF_API_TOKEN")
 MODEL_NAME = os.getenv("MODEL_NAME", "google/flan-t5-base")
 
+# URL запроса к модели через Hugging Face Inference API
 API_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
 HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
 
+
+# Функция запроса к модели
 def query(payload):
+    print("⏳ Отправка запроса к модели...")
     response = requests.post(API_URL, headers=HEADERS, json=payload)
+    print("📡 Статус:", response.status_code)
+    print("📩 Ответ:", response.text)
+
     if response.status_code != 200:
         return f"❌ Ошибка API: {response.status_code} {response.text}"
     return response.json()
 
+
+# Функция чата
 def chat(message, history):
     if history is None:
         history = []
-    # Добавляем запрос пользователя с пустым ответом
+
+    print("👤 Сообщение пользователя:", message)
     history.append((message, ""))
 
+    # Запрос к модели
     response = query({"inputs": message})
 
-    # Парсим ответ — пытаемся получить строку
+    # Обработка ответа
     output = ""
     if isinstance(response, dict):
         if "error" in response:
-            output = response["error"]
+            output = "❌ Ошибка: " + response["error"]
         elif "generated_text" in response:
             output = response["generated_text"]
-        elif isinstance(response.get("choices"), list) and len(response["choices"]) > 0:
+        elif isinstance(response.get("choices"), list) and response["choices"]:
             output = response["choices"][0].get("message", {}).get("content", "")
         else:
             output = str(response)
-    elif isinstance(response, list) and len(response) > 0 and "generated_text" in response[0]:
+    elif isinstance(response, list) and "generated_text" in response[0]:
         output = response[0]["generated_text"]
     else:
         output = str(response)
 
-    # Гарантируем, что output — строка
     if not isinstance(output, str):
         output = str(output)
 
-    # Обновляем последний элемент в истории — теперь с ответом
     history[-1] = (message, output)
     return history, history
 
-chat_ui = gr.ChatInterface(fn=chat, title="FlareGPT")
 
+# Интерфейс Gradio
+chat_ui = gr.ChatInterface(
+    fn=chat,
+    title="FlareGPT",
+    retry_btn="🔄 Retry",
+    undo_btn="↩️ Undo",
+    clear_btn="🗑️ Clear"
+)
+
+# Запуск
 if __name__ == "__main__":
     chat_ui.launch()