add system status

requirements.txt

@@ -4,3 +4,4 @@ scipy
 huggingface_hub>=0.27.0
 gradio>=5.9.0
 python-multipart
+psutil

ui.py

@@ -1,20 +1,51 @@
+import os
+
 import gradio as gr
+import psutil
 
 import config
 from chat_logic import bot_response, set_interactive, user_input
 from styles import CSS
 
 
+def get_system_status():
+    """Возвращает текущую загрузку системы"""
+    # CPU
+    cpu_usage = psutil.cpu_percent(interval=None)
+
+    # RAM
+    process = psutil.Process(os.getpid())
+    memory_info = process.memory_info()
+    ram_usage_mb = memory_info.rss / 1024 / 1024
+
+    # Общая память системы
+    virtual_mem = psutil.virtual_memory()
+    ram_percent = virtual_mem.percent
+
+    return f"""
+    ### 🖥️ System Health
+    **CPU:** {cpu_usage}%
+    **RAM (App):** {ram_usage_mb:.1f} MB
+    **RAM (Total):** {ram_percent}%
+    """
+
+
 def create_ui():
     theme = gr.themes.Soft(primary_hue="blue", text_size="lg")
 
-    # Убрали fill_height=True, полагаемся на наш CSS
     with gr.Blocks(theme=theme, css=CSS, title="Code LLM") as demo:
-        # Добавили elem_id="main-row"
         with gr.Row(equal_height=True, variant="default", elem_id="main-row"):
             # --- ЛЕВАЯ КОЛОНКА ---
             with gr.Sidebar(elem_classes=["sidebar"]):
                 gr.Markdown("### ⚙️ Settings")
+
+                # --- МОНИТОРИНГ ---
+                system_status = gr.Markdown(value=get_system_status())
+                refresh_btn = gr.Button(
+                    "🔄 Refresh Status", size="sm", variant="secondary"
+                )
+                # ------------------
+
                 system_prompt = gr.Textbox(
                     label="System Prompt",
                     value="Вы опытный программист. Пишите чистый и эффективный код.",
@@ -30,16 +61,15 @@ def create_ui():
 
             # --- ПРАВАЯ КОЛОНКА ---
             with gr.Column(scale=1, elem_id="col-chat-main"):
-                # Чатбот
                 chatbot = gr.Chatbot(
                     elem_id="chatbot",
                     label="Code Assistant",
                     avatar_images=(None, "https://api.iconify.design/noto:robot.svg"),
+                    type="messages",
                     layout="bubble",
                     render_markdown=True,
                 )
 
-                # Поле ввода
                 with gr.Row(elem_id="input-area"):
                     msg = gr.Textbox(
                         show_label=False,
@@ -55,16 +85,29 @@ def create_ui():
                     )
 
         # --- СОБЫТИЯ ---
+
+        # Обновление статуса по клику
+        refresh_btn.click(get_system_status, None, system_status)
+
+        # Обновление статуса автоматически при генерации ответа (чтобы видеть пиковую нагрузку)
         msg.submit(user_input, [msg, chatbot], [msg, chatbot], queue=False).then(
             lambda: set_interactive(False), None, [msg, submit_btn], queue=False
         ).then(
             bot_response, [chatbot, system_prompt, temperature, max_tokens], chatbot
+        ).then(
+            get_system_status,
+            None,
+            system_status,  # <-- Обновляем статус после ответа
         ).then(lambda: set_interactive(True), None, [msg, submit_btn], queue=False)
 
         submit_btn.click(user_input, [msg, chatbot], [msg, chatbot], queue=False).then(
             lambda: set_interactive(False), None, [msg, submit_btn], queue=False
         ).then(
             bot_response, [chatbot, system_prompt, temperature, max_tokens], chatbot
+        ).then(
+            get_system_status,
+            None,
+            system_status,  # <-- Обновляем статус после ответа
         ).then(lambda: set_interactive(True), None, [msg, submit_btn], queue=False)
 
         clear_btn.click(lambda: [], None, chatbot, queue=False)