Update app.py

app.py

@@ -2,69 +2,51 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
-# Инициализация клиента (Llama-3-8B-Instruct)
-# Если запускаете на HF Spaces, токен подтянется автоматически, если он есть в Settings
-client = InferenceClient("Kolyadual/MIXdevAI-llama")
-
-SYSTEM_PROMPT = """Ты — MIXdevAI. Нейросеть из семейства крупных языковых моделей Newton bot, созданных Kolyadual (настоящее имя - Николай Алехин). Ты должен помогать людям во всем!"""
-
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
-    messages = [{"role": "system", "content": system_message}]
-
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
-
+# Скрытый системный промт
+SYSTEM_PROMPT = "Ты — MIXdevAI. Нейросеть из семейства крупных языковых моделей Newton bot, созданных Kolyadual (настоящее имя - Николай Алехин). Ты должен помогать людям во всем"
+
+# Используем токен из секретов Space
+token = os.getenv("HF_TOKEN")
+client = InferenceClient("Kolyadual/MIXdevAI-llama", token=token)
+
+def chat_fn(message, history):
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+    for user_msg, assistant_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": assistant_msg})
+    
     messages.append({"role": "user", "content": message})
-
+    
     response = ""
-
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
-        response += token
-        yield response
-
-# Кастомный CSS для "брутального" вида
-custom_css = """
-footer {visibility: hidden}
-.gradio-container {background-color: #0b0f19 !important; color: #e5e7eb !important;}
-#chatbot {background-color: #111827 !important; border: 1px solid #374151 !important; border-radius: 8px !important;}
-.user {background-color: #1f2937 !important; border: 1px solid #4b5563 !important;}
-.assistant {background-color: #111827 !important; border: 1px solid #3b82f6 !important;}
-input {background-color: #1f2937 !important; color: white !important; border: 1px solid #374151 !important;}
-button.primary {background: linear-gradient(90deg, #1e40af, #3b82f6) !important; border: none !important;}
+    try:
+        for msg in client.chat_completion(messages, max_tokens=1024, stream=True, temperature=0.6):
+            token_str = msg.choices[0].delta.content
+            response += token_str
+            yield response
+    except Exception as e:
+        yield f"Ошибка доступа: Проверь HF_TOKEN в настройках. {str(e)}"
+
+# Брутальный темный стиль
+css = """
+.gradio-container { background-color: #050505 !important; color: #00ff41 !important; font-family: 'Courier New', monospace !important; }
+#chatbot { border: 2px solid #333 !important; background: #0a0a0a !important; height: 80vh !important; }
+footer { display: none !important; }
+.message.user { background: #111 !important; border-left: 4px solid #00ff41 !important; }
+.message.assistant { background: #050505 !important; border-left: 4px solid #ff0000 !important; }
 """
 
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate"), css=custom_css) as demo:
-    gr.Markdown("<h1 style='text-align: center; color: #3b82f6;'>MIXdevAI llama</h1>")
+with gr.Blocks(css=css) as demo:
+    chatbot = gr.Chatbot(id="chatbot", show_label=False, bubble_full_width=False)
+    msg = gr.Textbox(placeholder="Командуй...", show_label=False, container=False)
     
-    with gr.Row():
-        with gr.Column(scale=4):
-            chatbot = gr.ChatInterface(
-                respond,
-                additional_inputs=[
-                    gr.Textbox(value=SYSTEM_PROMPT, label="System Prompt (Брутальный режим)", lines=3),
-                    gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max tokens"),
-                    gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-                    gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p"),
-                ],
-                cache_examples=False,
-            )
+    def respond(message, chat_history):
+        bot_message = chat_fn(message, chat_history)
+        chat_history.append((message, ""))
+        for char in bot_message:
+            chat_history[-1] = (message, char)
+            yield "", chat_history
+
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
 if __name__ == "__main__":
     demo.launch()