Update app.py

app.py

@@ -1,71 +1,35 @@
-import os
 import gradio as gr
+import os
 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"]
-        elif "generated_text" in response:
-            output = response["generated_text"]
-        elif isinstance(response.get("choices"), list) and response["choices"]:
-            output = response["choices"][0].get("message", {}).get("content", "")
+def chat_fn(message, history):
+    try:
+        payload = {"inputs": message}
+        response = requests.post(API_URL, headers=HEADERS, json=payload)
+        if response.status_code != 200:
+            return f"❌ Ошибка API: {response.status_code}\n{response.text}"
+        result = response.json()
+
+        # Обработка разных форматов ответа
+        if isinstance(result, list) and "generated_text" in result[0]:
+            return result[0]["generated_text"]
+        elif isinstance(result, dict) and "generated_text" in result:
+            return result["generated_text"]
         else:
-            output = str(response)
-    elif isinstance(response, list) and "generated_text" in response[0]:
-        output = response[0]["generated_text"]
-    else:
-        output = str(response)
-
-    if not isinstance(output, str):
-        output = str(output)
+            return "❌ Ответ не распознан"
+    except Exception as e:
+        return f"❌ Ошибка: {str(e)}"
 
-    history[-1] = (message, output)
-    return history, history
-
-
-# Интерфейс Gradio
-chat_ui = gr.ChatInterface(
-    fn=chat,
+gr.ChatInterface(
+    fn=chat_fn,
     title="FlareGPT",
     retry_btn="🔄 Retry",
     undo_btn="↩️ Undo",
-    clear_btn="🗑️ Clear"
-)
-
-# Запуск
-if __name__ == "__main__":
-    chat_ui.launch()
+    clear_btn="🗑️ Clear",
+).launch()