Update app.py

app.py

@@ -1,23 +1,36 @@
 import gradio as gr
 from gradio_client import Client
 import requests
+import json
 
 # Создаем клиент для взаимодействия с API
 client = Client("llamameta/Pixtral-Large-Instruct-2411")
 
+# Функция для загрузки системной роли из JSON файла
+def load_system_role(role_name):
+    with open('system_roles.json', 'r', encoding='utf-8') as file:
+        roles = json.load(file)
+    return roles.get(role_name, "Ты помощник по умолчанию.")
+
+# Функция для загрузки названий ролей из JSON файла
+def load_role_names():
+    with open('system_roles.json', 'r', encoding='utf-8') as file:
+        roles = json.load(file)
+    return list(roles.keys())
+
 def respond(
     message,
     history: list[tuple[str, str]],
-    system_message,
+    system_role_name,
     max_tokens,
     temperature,
     top_p,
 ):
     if not message:
         return history, ""
-    
+    system_role = load_system_role(system_role_name)
     # Формируем сообщения для отправки в API
-    messages = [{"role": "system", "content": system_message}]
+    messages = [{"role": "system", "content": system_role}]
     
     for val in history:
         if val[0]:
@@ -42,6 +55,9 @@ def respond(
     
     return history, ""
 
+# Загрузка названий ролей из JSON файла
+role_names = load_role_names()
+
 # Ссылка на файл CSS
 css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
 
@@ -60,7 +76,8 @@ with gr.Blocks(css=css) as demo:
             submit = gr.Button("Отправить")
     
     with gr.Accordion("Настройки модели", open=False):
-        system_message = gr.Textbox(value="Ты - дружелюбный ии, способный помочь в чём угодно и в любой ситуации!", placeholder="Ты - дружелюбный ии, способный помочь в чём угодно и в любой ситуации!", label="Системное сообщение", lines=2)
+        with gr.Accordion(label="Помощник", open=False):
+            helper_role = gr.Radio(show_label=True, label="Выберите помощника", interactive=True, choices=role_names, value=role_names[0])
         max_tokens = gr.Slider(minimum=100, maximum=18000, value=18000, step=1, label="Максимальное количество новых токенов")
         temperature = gr.Slider(minimum=0.0, maximum=1.0, value=0.7, step=0.1, label="Температура")
         top_p = gr.Slider(minimum=0.0, maximum=1.0, value=0.95, step=0.05, label="Top-p (нуклеарное сэмплирование)")