Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-import random
+import os
 
 # 50 лучших моделей для выбора
 MODELS = {
@@ -56,20 +56,41 @@ MODELS = {
     "⚙️ FLAN-UL2": "google/flan-ul2"
 }
 
-def format_model_name(name):
-    return f"{name[:30]}..." if len(name) > 30 else name
-
 def respond(message, history, model, system_message, max_tokens, temperature, top_p, hf_token):
     try:
-        client = InferenceClient(token=hf_token.token if hf_token else None, model=MODELS[model])
+        # Исправление: проверяем тип токена
+        token = None
+        if hf_token:
+            if hasattr(hf_token, 'token'):
+                token = hf_token.token
+            elif isinstance(hf_token, str):
+                token = hf_token
+            else:
+                token = os.getenv("HF_TOKEN")  # Берем из переменных окружения
+        
+        if not token:
+            yield "❌ Please login with your Hugging Face account or set HF_TOKEN"
+            return
+        
+        client = InferenceClient(token=token, model=MODELS[model])
         
-        messages = [{"role": "system", "content": system_message}] + history + [{"role": "user", "content": message}]
+        messages = [{"role": "system", "content": system_message}]
+        for msg in history:
+            messages.append({"role": msg["role"], "content": msg["content"]})
+        messages.append({"role": "user", "content": message})
         
         response = ""
-        for chunk in client.chat_completion(messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p):
+        for chunk in client.chat_completion(
+            messages, 
+            max_tokens=max_tokens, 
+            stream=True, 
+            temperature=temperature, 
+            top_p=top_p
+        ):
             if chunk.choices and chunk.choices[0].delta.content:
                 response += chunk.choices[0].delta.content
                 yield response
+                
     except Exception as e:
         yield f"❌ Error: {str(e)}"
 
@@ -125,6 +146,7 @@ button.primary:hover {
 }
 """
 
+# Создаем интерфейс
 with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     with gr.Sidebar(open=True):
         gr.Markdown("""