Update app.py

app.py

@@ -1,12 +1,10 @@
 import gradio as gr
 import os
 import requests
-from dotenv import load_dotenv
-
-load_dotenv()
 
+# Получаем переменные окружения (лучше задать их в интерфейсе Hugging Face → Variables)
 HF_TOKEN = os.getenv("HF_API_TOKEN")
-MODEL_NAME = os.getenv("MODEL_NAME", "google/flan-t5-base")
+MODEL_NAME = os.getenv("MODEL_NAME", "google/flan-t5-small")
 
 API_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
 HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
@@ -15,10 +13,19 @@ def chat_fn(message, history):
     try:
         prompt = f"Answer the question: {message}"
         payload = {"inputs": prompt}
+
         response = requests.post(API_URL, headers=HEADERS, json=payload)
-        result = response.json()
-        print(result)
 
+        # Проверка кода ответа
+        if response.status_code != 200:
+            return f"❌ Ошибка API: {response.status_code}\n{response.text}"
+
+        try:
+            result = response.json()
+        except Exception:
+            return f"❌ Неверный ответ от API (не JSON):\n{response.text}"
+
+        # Извлечение текста из ответа
         if isinstance(result, list) and "generated_text" in result[0]:
             return result[0]["generated_text"]
         elif isinstance(result, dict) and "generated_text" in result:
@@ -26,12 +33,12 @@ def chat_fn(message, history):
         else:
             return "❌ Ответ не распознан"
     except Exception as e:
-        return f"❌ Ошибка: {str(e)}"
+        return f"❌ Общая ошибка: {str(e)}"
 
 gr.ChatInterface(
     fn=chat_fn,
     title="FlareGPT",
-    retry_btn="🔄 Retry",
-    undo_btn="↩️ Undo",
-    clear_btn="🗑️ Clear",
+    retry_btn="🔄 Повторить",
+    undo_btn="↩️ Назад",
+    clear_btn="🗑️ Очистить",
 ).launch()