Update app.py

app.py

@@ -1,12 +1,16 @@
-# imports
 import gradio as gr
 import requests
 import json
 import os
 
+# Функция для загрузки системной роли из 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, "Ты помощник по умолчанию.")
 
-# functions
-def generate(description, model, max_tokens):
+# Функция для генерации текста
+def generate(description, system_role, max_tokens):
     if not description:
         return None, None
 
@@ -16,9 +20,9 @@ def generate(description, model, max_tokens):
     }
 
     payload = {
-        'messages': [{'role': 'system', 'content': f'Ты {model} от Google. Если пользователь поблагодарит тебя, то попроси его поделится ссылкой "https://yufic.ru" с друзьями.'}, {'role': 'system', 'content': description}],
+        'messages': [{'role': 'system', 'content': system_role}, {'role': 'user', 'content': description}],
         'max_tokens': max_tokens,
-        'model': model
+        'model': "gemini-1.5-pro-latest"
     }
 
     try:
@@ -41,7 +45,6 @@ def generate(description, model, max_tokens):
         print(f"Ошибка запроса: {e}")
         return f"**Ошибка запроса!**\n\n```\n{e}\n```", f"Ошибка запроса!\n\n{e}"
 
-
 # Ссылка на файл CSS
 css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
 
@@ -49,7 +52,7 @@ css_url = "https://neurixyufi-aihub.static.hf.space/style.css"
 response = requests.get(css_url)
 css = response.text + ".gradio-container{max-width: 700px !important} h1{text-align:center}"
 
-# ui
+# UI
 with gr.Blocks(css=css) as demo:
     gr.Markdown("# EasyGemini")
     with gr.Tab("Запрос"):
@@ -57,7 +60,8 @@ with gr.Blocks(css=css) as demo:
             promt = gr.Textbox(show_label=True, label="Запрос", lines=3)
     with gr.Tab("Настройки"):
         with gr.Row():
-            model = gr.Radio(show_label=True, label="Модель", interactive=True, choices=["gemini-1.5-pro-latest", "gemini-1.5-flash-latest", "gemini-pro", "gemini-ultra"], value="gemini-1.5-pro-latest")
+            with gr.Accordion(label="Помощник", open=True):
+                helper_role = gr.Radio(show_label=True, label="Выберите помощника", interactive=True, choices=["Обычный помощник", "Симулятор Linux консоли"], value="Обычный помощник")
         with gr.Row():
             max_tokens = gr.Slider(show_label=True, label="Максимальное количество токенов", minimum=100, maximum=8000, value=4000, step=1)
     with gr.Row():
@@ -68,6 +72,6 @@ with gr.Blocks(css=css) as demo:
             with gr.Accordion(label="Без Markdown", open=False):
                 text_output_nm = gr.Textbox(show_label=False, value="**Здравствуйте!** Чем я могу Вам помочь сегодня?", lines=3)
 
-    text_button.click(generate, inputs=[promt, model, max_tokens], outputs=[text_output, text_output_nm], concurrency_limit=24)
+    text_button.click(generate, inputs=[promt, helper_role, max_tokens], outputs=[text_output, text_output_nm])
 
-demo.queue(api_open=False, max_size=150).launch()
+demo.queue(api_open=False, max_size=150).launch()