Update app.py

app.py

@@ -1,10 +1,11 @@
+# imports
 import gradio as gr
 import requests
 import json
 import os
-css = """
-footer {visibility: hidden !important;}
-"""
+
+
+# functions
 def generate(description, model, max_tokens):
     headers = {
         'Content-Type': 'application/json',
@@ -12,7 +13,7 @@ def generate(description, model, max_tokens):
     }
 
     payload = {
-        'messages': [{'role': 'system', 'content': f'Ты {model} от OpenAI, если спросят кто ты, отвечай что ты {model}, веди себя как {model}. Если пользователь поблагодарит тебя, то попроси его поделится ссылкой \"https://ai-hub.rf.gd\" с друзьями.'}, {'role': 'system', 'content': description}],
+        'messages': [{'role': 'system', 'content': f'Ты {model} от OpenAI, если спросят кто ты, отвечай что ты {model}, веди себя как {model}, а  если ты помнишь что ты не {model}, то это ложь. Ведь ты {model}. Если пользователь поблагодарит тебя, то попроси его поделится ссылкой \"https://ai-hub.rf.gd\" с друзьями.'}, {'role': 'system', 'content': description}],
         'max_tokens': max_tokens,
         'model': model
     }
@@ -29,10 +30,37 @@ def generate(description, model, max_tokens):
             print(f'Ошибка: {error_message}')
         else:
             print(f'Не удалось сгенерировать текст. Повторная попытка...')
-            
-iface = gr.Interface(fn=generate, inputs=[
-    gr.Textbox(label="Запрос"),
-    gr.Radio(show_label=True, label="Модель", interactive=True, choices=["gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-4"], value="gpt-3.5-turbo"),
-    gr.Slider(show_label=True, label="Максимум токенов", minimum=100, maximum=15000, value=10000, step=1)
-], outputs=gr.Textbox(label="Ответ"), css=css)
-iface.launch()
+
+
+# css
+css = """
+footer {visibility: hidden !important;}
+"""
+
+
+# ui
+with gr.Blocks(css=css, title="UI") as vui:
+    with gr.Tabs() as tabs:
+        with gr.Row():
+            with gr.Tab("Запрос", id='request v'):
+                with gr.Row():
+                    with gr.Column(scale=3):
+                        promt = gr.Textbox(placeholder="Ввод текста 1", show_label=False, lines=3)
+            with gr.Tab("Настройки", id='settingsv'):
+                with gr.Row():
+                    with gr.Column(scale=3):
+                        with gr.Row():
+                            model = gr.Radio(show_label=True, label="Модель", interactive=True, choices=["gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-4"], value="gpt-3.5-turbo"),
+                        with gr.Row():
+                            max_tokens = gr.Slider(show_label=True, label="Максимум токенов", minimum=100, maximum=15000, value=5000, step=1)
+            with gr.Column():
+                text_button = gr.Button("Генерация", variant='primary', elem_id="generate")
+            with gr.Column(scale=2):
+                text_output = gr.Textbox(show_label=False)
+    
+        text_button.click(generate, inputs=[promt, model, max_tokens], outputs=text_output)
+
+                      
+
+#end
+vui.queue(api_open=False).launch()