Update app.py

app.py

@@ -5,6 +5,7 @@ import json
 from pathlib import Path
 from datetime import datetime
 import requests
+import os
 
 # === ЗАГРУЗКА МОДЕЛИ ===
 print("🚀 Загрузка модели...")
@@ -20,7 +21,9 @@ HISTORY_DIR = Path("chat_history")
 HISTORY_DIR.mkdir(exist_ok=True)
 
 def get_history_path(username):
-    return HISTORY_DIR / f"{username}.json"
+    # Очищаем имя пользователя от недопустимых символов для имени файла
+    safe_name = "".join(c for c in username if c.isalnum() or c in ('-', '_')).lower()
+    return HISTORY_DIR / f"{safe_name}.json"
 
 def load_history(username):
     path = get_history_path(username)
@@ -35,7 +38,7 @@ def save_history(username, history_data):
         json.dump(history_data, f, ensure_ascii=False, indent=2)
 
 def check_if_pro(token):
-    if not token:
+    if not token or not token.startswith("hf_"):
         return False
     try:
         response = requests.get(
@@ -53,16 +56,14 @@ def check_if_pro(token):
 # === ГЕНЕРАЦИЯ ОТВЕТА ===
 def generate_response(message, history, username, current_chat_id):
     if not username:
-        gr.Warning("Пожалуйста, войдите через Hugging Face")
+        gr.Warning("Сначала введите имя пользователя на экране входа!")
         return history, gr.update(), current_chat_id
     
     if not message.strip():
         return history, gr.update(), current_chat_id
     
-    # Добавляем сообщение пользователя
     history = history + [{"role": "user", "content": message}]
     
-    # Формируем промпт из всей истории
     prompt = ""
     for msg in history:
         if msg["role"] == "user":
@@ -87,7 +88,6 @@ def generate_response(message, history, username, current_chat_id):
     
     history = history + [{"role": "assistant", "content": ai_response}]
     
-    # Сохраняем в хранилище
     if username and current_chat_id:
         history_data = load_history(username)
         if current_chat_id not in history_data:
@@ -179,8 +179,6 @@ CUSTOM_CSS = """
 
 # === ИНТЕРФЕЙС ===
 with gr.Blocks(title="OpenAirAI-X Chat") as demo:
-    
-    # Состояния
     username_state = gr.State("")
     history_state = gr.State({})
     current_chat_id_state = gr.State(None)
@@ -194,96 +192,77 @@ with gr.Blocks(title="OpenAirAI-X Chat") as demo:
         """)
         
         with gr.Row():
-            # Боковая панель
             with gr.Column(scale=1, min_width=250):
-                user_info = gr.Textbox(
-                    label="Пользователь",
-                    value="Не авторизован",
-                    interactive=False,
-                    elem_classes="user-info"
-                )
+                user_info = gr.Textbox(label="Пользователь", value="Не авторизован", interactive=False, elem_classes="user-info")
                 pro_badge = gr.HTML('<div class="pro-badge">👑 PRO</div>', visible=False)
                 new_chat_btn = gr.Button("➕ Новый чат", variant="primary")
                 gr.Markdown("### 📚 История чатов")
-                chat_list = gr.Radio(
-                    choices=[],
-                    label="Ваши чаты",
-                    interactive=True
-                )
+                chat_list = gr.Radio(choices=[], label="Ваши чаты", interactive=True)
                 delete_chat_btn = gr.Button("🗑️ Удалить выбранный чат", variant="stop", size="sm")
                 logout_btn = gr.Button("🚪 Выйти", size="sm")
             
-            # Основная область чата
             with gr.Column(scale=3):
-                chatbot = gr.Chatbot(
-                    label="Диалог",
-                    height=500
-                )
+                # Исправлено для Gradio 6.0: убраны type и show_copy_button
+                chatbot = gr.Chatbot(label="Диалог", height=500)
                 with gr.Row():
-                    msg_input = gr.Textbox(
-                        placeholder="Напишите сообщение...",
-                        lines=2,
-                        scale=5,
-                        show_label=False
-                    )
+                    msg_input = gr.Textbox(placeholder="Напишите сообщение...", lines=2, scale=5, show_label=False)
                     send_btn = gr.Button("📤 Отправить", variant="primary", scale=1)
     
-    # Экран входа
+    # Экран входа (без gr.LoginButton, чтобы избежать ошибок в Docker)
     with gr.Column(visible=True) as login_screen:
         gr.HTML("""
         <div class="main-header">
             <h1>🤖 OpenAirAI-X Chat</h1>
-            <p>Войдите через Hugging Face, чтобы начать</p>
+            <p>Введите данные для входа</p>
         </div>
         """)
         gr.Markdown("""
         ### 🔐 Вход в систему
         
-        Для использования чата необходимо войти через аккаунт Hugging Face.
-        
-        **Преимущества входа:**
-        - 💾 Сохранение истории чатов
-        - 👑 PRO-статус для подписчиков
-        - 🔄 Доступ к предыдущим диалогам
+        Введите ваше имя пользователя Hugging Face. История чатов будет привязана к этому имени.
         """)
-        gr.LoginButton(value="🔑 Войти через Hugging Face", variant="primary")
+        with gr.Column(max_width=400):
+            username_input = gr.Textbox(label="Имя пользователя (HF Username)", placeholder="Например: RootLinux21")
+            token_input = gr.Textbox(label="HF Token (необязательно, только для проверки PRO)", type="password", placeholder="hf_...")
+            login_btn = gr.Button("🔑 Войти", variant="primary")
+            gr.Markdown("*Если у вас есть PRO-подписка, вставьте ваш токен (Settings -> Access Tokens), чтобы увидеть значок 👑*, токен не передаётся OpenRussianAI!")
     
     # === ОБРАБОТЧИКИ ===
     
-    def handle_login(request: gr.Request):
-        if request.username:
-            username = request.username
-            token = request.oauth_token.get("access_token") if request.oauth_token else None
-            is_pro = check_if_pro(token)
-            history_data = load_history(username)
+    def handle_login(username, token):
+        if not username.strip():
+            gr.Warning("Введите имя пользователя!")
+            return gr.update(), gr.update(), "Не авторизован", "", {}, None, [], gr.update(visible=False)
+        
+        is_pro = check_if_pro(token)
+        history_data = load_history(username)
+        chat_list_choices = get_chat_list(username)
+        
+        current_chat_id = None
+        if not history_data:
+            chat_id = datetime.now().strftime("%Y%m%d_%H%M%S")
+            history_data[chat_id] = {
+                "title": "Новый чат",
+                "created": datetime.now().isoformat(),
+                "messages": []
+            }
+            save_history(username, history_data)
+            current_chat_id = chat_id
             chat_list_choices = get_chat_list(username)
-            
-            current_chat_id = None
-            if not history_data:
-                chat_id = datetime.now().strftime("%Y%m%d_%H%M%S")
-                history_data[chat_id] = {
-                    "title": "Новый чат",
-                    "created": datetime.now().isoformat(),
-                    "messages": []
-                }
-                save_history(username, history_data)
-                current_chat_id = chat_id
-                chat_list_choices = get_chat_list(username)
-            
-            pro_badge_html = '<div class="pro-badge">👑 PRO</div>'
-            display_name = f"👤 {username}"
-            
-            return (
-                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())
+        
+        pro_badge_html = '<div class="pro-badge">👑 PRO</div>'
+        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=is_pro, value=pro_badge_html) # pro_badge
+        )
     
     def handle_logout():
         return (
@@ -297,11 +276,10 @@ with gr.Blocks(title="OpenAirAI-X Chat") as demo:
             gr.update(visible=False)
         )
     
-    demo.load(
+    login_btn.click(
         fn=handle_login,
-        inputs=None,
-        outputs=[main_interface, login_screen, user_info, username_state, 
-                 history_state, current_chat_id_state, chat_list, pro_badge]
+        inputs=[username_input, token_input],
+        outputs=[main_interface, login_screen, user_info, username_state, history_state, current_chat_id_state, chat_list, pro_badge]
     )
     
     send_btn.click(
@@ -345,9 +323,8 @@ with gr.Blocks(title="OpenAirAI-X Chat") as demo:
     logout_btn.click(
         fn=handle_logout,
         inputs=None,
-        outputs=[main_interface, login_screen, user_info, username_state,
-                 history_state, current_chat_id_state, chat_list, pro_badge]
+        outputs=[main_interface, login_screen, user_info, username_state, history_state, current_chat_id_state, chat_list, pro_badge]
     )
 
-# Запуск с CSS в launch()
+# Запуск: css передается в launch(), как требует Gradio 6.0
 demo.launch(server_name="0.0.0.0", server_port=7860, css=CUSTOM_CSS)