Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 
 
 # functions
-def generate(description, model, max_tokens):
+def generate(description):
     if not description:
         return None, None
 
@@ -16,9 +16,9 @@ def generate(description, model, max_tokens):
     }
 
     payload = {
-        'messages': [{'role': 'system', 'content': f'Ты {model} от Google. Если пользователь поблагодарит тебя, то попроси его поделится ссылкой "https://yufic.ru" с друзьями.'}, {'role': 'system', 'content': description}],
-        'max_tokens': max_tokens,
-        'model': model
+        'messages': [{'role': 'system', 'content': f'Ты - корректор текста. Пользователь будет отправлять тебе сообщения, а ты должен исправлять в них все ГРАММАТИЧЕСКИЕ ошибки (в том числе и знаки припенания). Отвечать ты должен без лишних символов, только исправленный запрос пользователя, без markdown. Не надо писать вроде этого: "Вот ваш исправленный тексст: ...", пиши сразу "..."'}, {'role': 'user', 'content': description}],
+        'max_tokens': 150000,
+        'model': "gemini-1.5-pro-latest"
     }
 
     try:
@@ -27,16 +27,16 @@ def generate(description, model, max_tokens):
         data = response.json()  # Используем response.json() вместо json.loads()
 
         if 'choices' in data and len(data['choices']) > 0:
-            command = data['choices'][0]['message']['content'].strip()
+            correct = data['choices'][0]['message']['content'].strip()
             print("Генерация успешна")
-            return command, command
+            return correct
         elif 'error' in data:
             error_message = data['error']['message']
             print(f'Ошибка: {error_message}')
-            return f"**Ошибка сервера!**\n\n```\n{error_message}\n```", f"Ошибка сервера!\n\n{error_message}"
+            return f"**Ошибка сервера!**\n\n```\n{error_message}\n```"
         else:
             print(f'Не удалось сгенерировать текст. Ответ сервера: {data}')  # Выводим ответ сервера для отладки
-            return "**Неизвестная ошибка!**", "Неизвестная ошибка!"
+            return "**Неизвестная ошибка!**"
     except requests.exceptions.RequestException as e:
         print(f"Ошибка запроса: {e}")
         return f"**Ошибка запроса!**\n\n```\n{e}\n```", f"Ошибка запроса!\n\n{e}"
@@ -52,19 +52,12 @@ with gr.Blocks(css=css) as demo:
     with gr.Tab("Запрос"):
         with gr.Row():
             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.Row():
-            max_tokens = gr.Slider(show_label=True, label="Максимальное количество токенов", minimum=100, maximum=8000, value=4000, step=1)
-    with gr.Row():
         text_button = gr.Button("Генерация", variant='primary')
     with gr.Row():
         with gr.Tab("Ответ"):
             text_output = gr.Markdown(show_label=False, value="**Здравствуйте!** Чем я могу Вам помочь сегодня?")
-            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], outputs=[text_output], concurrency_limit=512)
 
 demo.queue(api_open=False).launch()