Update app.py

app.py

@@ -5,41 +5,48 @@ import requests
 HF_TOKEN = os.getenv("HF_API_TOKEN")
 MODEL_NAME = os.getenv("MODEL_NAME", "google/flan-t5-base")
 
-headers = {
-    "Authorization": f"Bearer {HF_TOKEN}"
-}
-
 API_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
+HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
 
 def query(payload):
-    response = requests.post(API_URL, headers=headers, json=payload)
-    if response.status_code == 200:
-        return response.json()
-    elif response.status_code == 404:
-        return {"error": "❌ Ошибка API: 404 Not Found"}
-    elif response.status_code == 401:
-        return {"error": "❌ Ошибка API: Unauthorized (проверь токен)"}
-    else:
-        return {"error": f"❌ Ошибка API: {response.status_code}"}
+    response = requests.post(API_URL, headers=HEADERS, json=payload)
+    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 = []
+    # Добавляем запрос пользователя с пустым ответом
     history.append((message, ""))
-    input_text = message
-
-    result = query({"inputs": input_text})
-    if "error" in result:
-        output = result["error"]
-    elif isinstance(result, list) and "generated_text" in result[0]:
-        output = result[0]["generated_text"]
-    elif isinstance(result, dict) and "generated_text" in result:
-        output = result["generated_text"]
+
+    response = query({"inputs": message})
+
+    # Парсим ответ — пытаемся получить строку
+    output = ""
+    if isinstance(response, dict):
+        if "error" in response:
+            output = response["error"]
+        elif "generated_text" in response:
+            output = response["generated_text"]
+        elif isinstance(response.get("choices"), list) and len(response["choices"]) > 0:
+            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]:
+        output = response[0]["generated_text"]
     else:
-        output = str(result)
+        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", theme="soft")
+chat_ui = gr.ChatInterface(fn=chat, title="FlareGPT")
 
 if __name__ == "__main__":
     chat_ui.launch()