Update app.py

app.py

@@ -2,62 +2,44 @@ import os
 import gradio as gr
 import requests
 
-# Загружаем токен и имя модели из secrets
-API_TOKEN = os.getenv("HF_API_TOKEN")
-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 {API_TOKEN}"}
-
+HF_TOKEN = os.getenv("HF_API_TOKEN")
+MODEL_NAME = os.getenv("MODEL_NAME", "google/flan-t5-base")
 
-def query_huggingface(prompt):
-    if not API_TOKEN:
-        return "❌ Токен HF_API_TOKEN не задан! Добавь его в Secrets."
-    
-    payload = {"inputs": prompt}
+headers = {
+    "Authorization": f"Bearer {HF_TOKEN}"
+}
 
-    try:
-        response = requests.post(API_URL, headers=headers, json=payload)
-    except Exception as e:
-        return f"🚫 Ошибка подключения: {str(e)}"
+API_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
 
-    if response.status_code != 200:
-        return f"❌ Ошибка API: {response.status_code}\n{response.text}"
+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}"}
 
-    try:
-        result = response.json()
-    except Exception as e:
-        return f"❌ Ошибка парсинга ответа: {str(e)}\nОтвет: {response.text}"
+def chat(message, history):
+    history.append((message, ""))
+    input_text = message
 
-    # Разбор разных форматов
-    if isinstance(result, list) and "generated_text" in result[0]:
-        return result[0]["generated_text"]
+    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:
-        return result["generated_text"]
-    elif isinstance(result, dict) and "error" in result:
-        return f"⚠️ Ошибка модели: {result['error']}"
+        output = result["generated_text"]
     else:
-        return f"⚠️ Неожиданный формат ответа: {result}"
-
+        output = str(result)
 
-def chat(user_input, history):
-    history = history or []
-    prompt = f"Q: {user_input}\nA:"
-    response = query_huggingface(prompt)
-    history.append((user_input, response))
+    history[-1] = (message, output)
     return history, history
 
+chat_ui = gr.ChatInterface(fn=chat, title="FlareGPT", theme="soft")
 
-# UI через Gradio Blocks
-with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 FlareGPT — чат с моделью Hugging Face")
-
-    chatbot = gr.Chatbot(height=400)
-    message = gr.Textbox(placeholder="Введите сообщение и нажмите Enter...")
-    clear = gr.Button("🧹 Очистить")
-
-    state = gr.State([])
-
-    message.submit(chat, [message, state], [chatbot, state])
-    clear.click(lambda: ([], []), None, [chatbot, state])
-
-demo.launch()
+if __name__ == "__main__":
+    chat_ui.launch()