Update app.py

app.py

@@ -60,9 +60,7 @@ def generate_response(message, history, username, current_chat_id):
         return history, gr.update(), current_chat_id
     
     # Новый формат Gradio 6+: список словарей
-    history = history + [
-        {"role": "user", "content": message},
-    ]
+    history = history + [{"role": "user", "content": message}]
     
     # Формируем промпт из всей истории
     prompt = ""
@@ -87,7 +85,6 @@ def generate_response(message, history, username, current_chat_id):
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     ai_response = response.split("AI:")[-1].strip()
     
-    # Добавляем ответ AI
     history = history + [{"role": "assistant", "content": ai_response}]
     
     # Сохраняем в хранилище
@@ -181,7 +178,7 @@ CUSTOM_CSS = """
 """
 
 # === ИНТЕРФЕЙС ===
-with gr.Blocks(title="OpenAirAI-X Chat") as demo:
+with gr.Blocks(css=CUSTOM_CSS, title="OpenAirAI-X Chat") as demo:
     
     # Состояния
     username_state = gr.State("")
@@ -221,7 +218,6 @@ with gr.Blocks(title="OpenAirAI-X Chat") as demo:
                 chatbot = gr.Chatbot(
                     label="Диалог",
                     height=500,
-                    show_copy_button=True,
                     type="messages"
                 )
                 with gr.Row():
@@ -279,13 +275,13 @@ with gr.Blocks(title="OpenAirAI-X Chat") as demo:
             display_name = f"👤 {username}"
             
             return (
-                gr.update(visible=True),    # main_interface
-                gr.update(visible=False),   # login_screen
-                display_name,               # user_info
-                username,                   # username_state
-                history_data,               # history_state
-                current_chat_id,            # current_chat_id_state
-                chat_list_choices,          # chat_list
+                gr.update(visible=True),
+                gr.update(visible=False),
+                display_name,
+                username,
+                history_data,
+                current_chat_id,
+                chat_list_choices,
                 gr.update(visible=is_pro, value=pro_badge_html)
             )
         return (gr.update(), gr.update(), gr.update(), "", {}, None, [], gr.update())
@@ -302,7 +298,6 @@ with gr.Blocks(title="OpenAirAI-X Chat") as demo:
             gr.update(visible=False)
         )
     
-    # При загрузке страницы проверяем авторизацию
     demo.load(
         fn=handle_login,
         inputs=None,
@@ -310,7 +305,6 @@ with gr.Blocks(title="OpenAirAI-X Chat") as demo:
                  history_state, current_chat_id_state, chat_list, pro_badge]
     )
     
-    # Отправка сообщения
     send_btn.click(
         fn=generate_response,
         inputs=[msg_input, chatbot, username_state, current_chat_id_state],
@@ -331,28 +325,24 @@ with gr.Blocks(title="OpenAirAI-X Chat") as demo:
         outputs=[chat_list]
     )
     
-    # Новый чат
     new_chat_btn.click(
         fn=new_chat,
         inputs=[username_state],
         outputs=[chatbot, current_chat_id_state, chat_list]
     )
     
-    # Загрузка чата
     chat_list.change(
         fn=load_chat,
         inputs=[chat_list, username_state, current_chat_id_state],
         outputs=[chatbot, current_chat_id_state]
     )
     
-    # Удаление чата
     delete_chat_btn.click(
         fn=delete_chat,
         inputs=[chat_list, username_state],
         outputs=[chatbot, history_state, current_chat_id_state, chat_list]
     )
     
-    # Выход
     logout_btn.click(
         fn=handle_logout,
         inputs=None,
@@ -360,5 +350,4 @@ with gr.Blocks(title="OpenAirAI-X Chat") as demo:
                  history_state, current_chat_id_state, chat_list, pro_badge]
     )
 
-# Запуск с CSS
-demo.launch(server_name="0.0.0.0", server_port=7860, css=CUSTOM_CSS)
+demo.launch(server_name="0.0.0.0", server_port=7860)