Update app.py

app.py

@@ -1,64 +1,114 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
+from openai import OpenAI
 
-"""
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-"""
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
+# Инициализация клиента для DeepSeek (укажите ваш ключ!)
+client = OpenAI(
+    api_key="sk-a02694cf3c8640c9ae60428ee2c5a62e",  # <-- ЗАМЕНИТЕ на свой ключ
+    base_url="https://api.deepseek.com"
+)
 
+# Функция, которая будет вызываться при каждом новом сообщении
+# Она принимает текущее сообщение пользователя и "history" — историю чата 
+# в формате Gradio (список кортежей).
+def chat_with_deepseek(user_message, history):
+    """
+    :param user_message: текст последнего сообщения от пользователя
+    :param history: список кортежей ([(user_msg, bot_msg), ...]),
+                    где хранится вся история переписки
+    :return: обновлённая история с добавленным ответом от deepseek-reasoner
+    """
 
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
-    messages = [{"role": "system", "content": system_message}]
+    # Преобразуем history из формата Gradio в формат messages для DeepSeek
+    messages = []
+    for user_msg, bot_msg in history:
+        if user_msg:
+            messages.append({"role": "user", "content": user_msg})
+        if bot_msg:
+            messages.append({"role": "assistant", "content": bot_msg})
 
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
+    # Добавляем новое пользовательское сообщение
+    messages.append({"role": "user", "content": user_message})
 
-    messages.append({"role": "user", "content": message})
+    # Обращаемся к deepseek-reasoner, передавая всю историю
+    try:
+        response = client.chat.completions.create(
+            model="deepseek-reasoner",
+            messages=messages
+        )
+        bot_reply = response.choices[0].message.content
+    except Exception as e:
+        bot_reply = f"Ошибка при обращении к API: {str(e)}"
 
-    response = ""
+    # Возвращаем историю, дополнив её ответом бота
+    history.append((user_message, bot_reply))
+    return history, history
 
-    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
+# Настраиваем интерфейс Gradio
+# - elem_id / css / theme позволяют стилизовать под "тёмный" чат
+# - "chatbot" виджет в Gradio уже отображает историю в виде «баблов»
+with gr.Blocks(
+    theme=gr.themes.Base(
+        primary_hue="slate",  # оттенок для кнопок
+        secondary_hue="blue", # оттенок для второстепенных элементов
+        neutral_hue="slate",  
+        text_size="md",
+        font=["Arial", "sans-serif"],
+    ),
+    css="""
+    body {
+        background-color: #111111 !important;
+    }
+    .block.block--main {
+        background-color: #111111 !important;
+    }
+    .gradio-container {
+        color: #ffffff !important;
+    }
+    /* Дополнительные правки под темный фон */
+    #chatbot { 
+        background-color: #222222 !important; 
+    }
+    """) as demo:
 
-        response += token
-        yield response
+    gr.Markdown(
+        "<h1 style='text-align: center; color: #ffffff;'>Чат с deepseek-reasoner</h1>"
+        "<p style='text-align: center; color: #bbbbbb;'>Тёмная тема, многошаговый диалог</p>"
+    )
 
+    # Компонент Chatbot для отображения диалога
+    chatbot = gr.Chatbot(label="Диалог").style(height=400)
 
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new 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 (nucleus sampling)",
-        ),
-    ],
-)
+    # Поле для ввода текста
+    msg = gr.Textbox(
+        label="Ваш вопрос",
+        placeholder="Напишите сообщение...",
+        lines=3
+    )
+    
+    # Кнопка отправки
+    send_btn = gr.Button("Отправить", variant="primary")
+
+    # state для хранения истории
+    state = gr.State([])  # пустой список истории
+
+    # Привязываем функцию chat_with_deepseek к нажатию кнопки "Отправить"
+    # input: user_message (msg), history (state)
+    # output: (chatbot, state) — обновлённая история чата
+    send_btn.click(
+        fn=chat_with_deepseek, 
+        inputs=[msg, state],
+        outputs=[chatbot, state],
+        scroll_to_output=True
+    )
 
+    # Также можно отправлять сообщение, нажав Enter в Textbox
+    msg.submit(
+        fn=chat_with_deepseek,
+        inputs=[msg, state],
+        outputs=[chatbot, state],
+        scroll_to_output=True
+    )
 
+# Запуск 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)