Update app.py

app.py

@@ -4,6 +4,7 @@ from huggingface_hub import InferenceClient
 import requests
 import re
 from duckduckgo_search import DDGS
+
 # ==== Константы ====
 MODEL_NAME = "openai/gpt-oss-20b"
 SYSTEM_MESSAGE = "Ты Pok.Bot, ты используешь открытую локальную модель GPT-OSS от OpenAI, не GPT-4 (наверно😆). Тебя создал POKilondron. Используй емодзи 😄.If you don't know the answer, output a command like this:#search <query> Do NOT invent facts. И да использую команду #search «запрос» когда нужно и не повторяй постоянноодин и тоже запрос и не видавай запрос на 18+ 😠"
@@ -42,7 +43,7 @@ def respond(message, history: list[dict[str, str]], hf_token: gr.OAuthToken):
         delta = msg.choices[0].delta.content
         if delta:
             raw_response += delta
-            yield raw_response  # Yield partial response
+            yield raw_response
 
     # === Проверка на команду поиска ===
     if raw_response.strip().startswith("#search"):
@@ -80,31 +81,44 @@ def respond(message, history: list[dict[str, str]], hf_token: gr.OAuthToken):
 # -------------------
 def load_chats():
     if os.path.exists(FILE):
-        # Check if the file is empty
-        if os.path.getsize(FILE) == 0:
-            return {}  # Return an empty dictionary if the file is empty
-        with open(FILE, "r", encoding="utf-8") as f:
-            try:
-                return json.load(f)
-            except json.JSONDecodeError:
-                # Handle potential corruption by returning an empty dict
-                return {}
+        # Добавлена проверка на пустой файл для предотвращения JSONDecodeError
+        if os.path.getsize(FILE) > 0:
+            with open(FILE, "r", encoding="utf-8") as f:
+                try:
+                    return json.load(f)
+                except json.JSONDecodeError:
+                    return {}
     return {}
 
-    # Convert Gradio history format to the format required by the model
-    # (list of dicts with 'role' and 'content')
+def save_chats(data):
+    with open(FILE, "w", encoding="utf-8") as f:
+        json.dump(data, f, ensure_ascii=False, indent=2)
+
+# -------------------
+# Логика чат-бота
+# -------------------
+def bot_reply(message, chat_id, chat_history, hf_token: gr.OAuthToken):
+    chats = load_chats()
+
+    if chat_id not in chats:
+        chats[chat_id] = []
+
+    chats[chat_id].append({"role": "user", "content": message})
+    save_chats(chats)
+
+    # Преобразуем историю в формат, нужный модели (список словарей)
     model_history = []
     for user_msg, bot_msg in chat_history:
         model_history.append({"role": "user", "content": user_msg})
         model_history.append({"role": "bot", "content": bot_msg})
-    
-    # This is the key change: call the respond function and stream the response
+
+    # Потоковый вызов respond и сохранение ответа
     full_response = ""
     for response_chunk in respond(message, model_history, hf_token):
         full_response = response_chunk
         yield chat_history + [(message, full_response)], ""
 
-    # After the full response is received, save it to the chat history file
+    # Сохраняем полный ответ в историю чата после завершения генерации
     chats[chat_id].append({"role": "bot", "content": full_response})
     save_chats(chats)
 
@@ -113,7 +127,8 @@ def load_chats():
 # -------------------
 def new_chat():
     chats = load_chats()
-    new_id = f"chat{len(chats)+1}"
+    # Убедимся, что ID уникален, даже если какой-то чат был удалён
+    new_id = f"chat{max([int(re.sub(r'chat', '', k)) for k in chats.keys()] or [0]) + 1}"
     chats[new_id] = []
     save_chats(chats)
     return gr.update(choices=list(chats.keys()), value=new_id), []
@@ -134,7 +149,7 @@ def load_chat(chat_id):
 # -------------------
 # Интерфейс Gradio
 # -------------------
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🗨️ Чат-бот с историей диалогов")
 
     with gr.Row():
@@ -145,13 +160,20 @@ with gr.Blocks() as demo:
     msg = gr.Textbox(label="Сообщение", placeholder="Напиши что-нибудь...")
     send = gr.Button("Отправить")
 
-    # Pass the chatbot history and Hugging Face token to bot_reply
+    # Обновляем обработчик для правильной передачи всех аргументов
+    msg.submit(
+        bot_reply,
+        [msg, chat_selector, chatbot, gr.OAuthToken()],
+        [chatbot, msg]
+    )
     send.click(
         bot_reply,
         [msg, chat_selector, chatbot, gr.OAuthToken()],
         [chatbot, msg]
     )
+
     new_btn.click(new_chat, None, [chat_selector, chatbot])
     chat_selector.change(load_chat, chat_selector, chatbot)
 
 demo.launch()
+