Update app.py

app.py

@@ -1,55 +1,80 @@
 import os
 import requests
 import gradio as gr
+from datasets import load_dataset
 
-# Токен берётся из Secrets Hugging Face
-HF_TOKEN = os.environ.get("HF_TOKEN")
-if not HF_TOKEN:
-    raise ValueError("❌ Токен не найден. Добавьте HF_TOKEN в Secrets!")
-
-# Настройки модели (используем облегчённую версию для стабильности)
-MODEL_NAME = "IlyaGusev/saiga_mistral_7b-lora"  # Альтернатива: "IlyaGusev/saiga_mistral_7b"
+# --- Конфигурация ---
+HF_TOKEN = os.environ.get("HF_TOKEN")  # Из Secrets Spaces
+MODEL_NAME = "IlyaGusev/saiga_mistral_7b-lora"
 API_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
-def generate_response(prompt, history):
-    """Генерация ответа с учётом истории диалога"""
+# --- Загрузка датасета ---
+try:
+    dataset = load_dataset("sberbank-ai/ru_helpdesk", split="train[:50]")  # Первые 50 примеров
+    examples = [d["question"] for d in dataset]
+except:
+    examples = [
+        "Мой заказ #12345 не прибыл вовремя",
+        "Как вернуть товар?",
+        "Проблема с оплатой на сайте"
+    ]
+
+# --- Функция генерации ответа ---
+def generate_response(message, history):
+    prompt = f"""Ты - ИИ-ассистент поддержки. Ответь клиенту профессионально.
+Контекст: {history}
+Клиент: {message}
+Ассистент:"""
+    
     try:
-        # Форматируем историю для модели (если нужно)
-        formatted_prompt = f"Диалог:\n{history}\nПользователь: {prompt}\nАссистент:"
-        
-        payload = {
-            "inputs": formatted_prompt,
-            "parameters": {
-                "max_new_tokens": 500,
-                "temperature": 0.7,
-                "do_sample": True,
+        response = requests.post(
+            API_URL,
+            headers=headers,
+            json={
+                "inputs": prompt,
+                "parameters": {
+                    "max_new_tokens": 300,
+                    "temperature": 0.3  # Для меньшей креативности
+                }
             }
-        }
-        
-        response = requests.post(API_URL, headers=headers, json=payload)
-        
-        if response.status_code == 200:
-            return response.json()[0]["generated_text"]
-        else:
-            return f"⚠️ Ошибка API: {response.text}"
+        )
+        return response.json()[0]["generated_text"].split("Ассистент:")[-1].strip()
     except Exception as e:
-        return f"❌ Ошибка: {str(e)}"
+        return f"⚠️ Ошибка: {str(e)}"
 
-# Интерфейс Gradio (аналог ChatGPT)
-with gr.Blocks(title="🇷🇺 Russian Saiga Chat") as demo:
-    gr.Markdown("## 💬 Чат с Saiga Mistral 7B (через HF API)")
-    chatbot = gr.Chatbot(label="Диалог")
-    msg = gr.Textbox(label="Сообщение", placeholder="Напишите что-то...")
-    clear = gr.Button("Очистить")
+# --- Интерфейс Gradio ---
+theme = gr.themes.Default(
+    primary_hue="blue",
+    font=[gr.themes.GoogleFont("Noto Sans"), "sans-serif"]
+)
 
-    def respond(message, chat_history):
-        bot_message = generate_response(message, chat_history)
-        chat_history.append((message, bot_message))
-        return "", chat_history
+with gr.Blocks(theme=theme) as demo:
+    gr.Markdown("""<center><h1>📞 Поддержка клиентов</h1></center>""")
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            chatbot = gr.Chatbot(height=400, bubble_full_width=False)
+            msg = gr.Textbox(label="Опишите проблему", placeholder="У меня проблема с...")
+            btn = gr.Button("Отправить", variant="primary")
+        
+        with gr.Column(scale=1):
+            gr.Examples(
+                examples=examples,
+                inputs=msg,
+                label="Примеры обращений"
+            )
+            gr.Markdown("""
+            **Советы для клиентов:**
+            1. Указывайте номер заказа
+            2. Описывайте проблему подробно
+            3. Сохраняйте вежливый тон
+            """)
 
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    clear.click(lambda: None, None, chatbot, queue=False)
+    btn.click(generate_response, [msg, chatbot], [msg, chatbot])
+    msg.submit(generate_response, [msg, chatbot], [msg, chatbot])
 
-demo.launch(debug=True)
+# --- Запуск ---
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0")