Update app.py

app.py

@@ -16,14 +16,15 @@ def generate(description, model, max_tokens):
     }
 
     payload = {
-        'messages': [{'role': 'system', 'content': f'Ты {model} от OpenAI. Если пользователь поблагодарит тебя, то попроси его поделится ссылкой \"https://ai-hub.rf.gd\" с друзьями.'}, {'role': 'system', 'content': description}],
+        'messages': [{'role': 'system', 'content': f'Ты {model} от Google. Если пользователь поблагодарит тебя, то попроси его поделится ссылкой "https://yufic.ru" с друзьями.'}, {'role': 'system', 'content': description}],
         'max_tokens': max_tokens,
         'model': model
     }
 
-    while True:
+    try:
         response = requests.post(os.getenv("BASE_URL"), headers=headers, json=payload)
-        data = json.loads(response.text)
+        response.raise_for_status()  # Проверяем на ошибки HTTP
+        data = response.json()  # Используем response.json() вместо json.loads()
 
         if 'choices' in data and len(data['choices']) > 0:
             command = data['choices'][0]['message']['content'].strip()
@@ -34,8 +35,11 @@ def generate(description, model, max_tokens):
             print(f'Ошибка: {error_message}')
             return f"**Ошибка сервера!**\n\n```\n{error_message}\n```", f"Ошибка сервера!\n\n{error_message}"
         else:
-            print(f'Не удалось сгенерировать текст. Повторите попытку позже...')
+            print(f'Не удалось сгенерировать текст. Ответ сервера: {data}')  # Выводим ответ сервера для отладки
             return "**Неизвестная ошибка!**", "Неизвестная ошибка!"
+    except requests.exceptions.RequestException as e:
+        print(f"Ошибка запроса: {e}")
+        return f"**Ошибка запроса!**\n\n```\n{e}\n```", f"Ошибка запроса!\n\n{e}"
 
 
 # css
@@ -43,28 +47,24 @@ css = """
 footer {visibility: hidden !important;}
 """
 
-
 # ui
-with gr.Blocks(css=css) as vui:
-    with gr.Tab("Запрос", id='request v'):
+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():
-            with gr.Column(scale=3):
-                promt = gr.Textbox(show_label=True, label="Запрос", lines=3)
-    with gr.Tab("Настройки", id='settingsv'):
+            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():
-            with gr.Column(scale=3):
-                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.Column():
-        text_button = gr.Button("Генерация", variant='primary', elem_id="generate")
-    with gr.Column(scale=2):
+            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)
 
-#end
-vui.queue(api_open=False).launch()
+demo.queue(api_open=False).launch()