Update app.py

app.py

@@ -2,23 +2,26 @@ import os
 import gradio as gr
 from huggingface_hub import InferenceClient
 import json
+import uuid  # gera IDs únicos
 
-SAVE_FILE = "bitai_history.json"
+# Funções pra salvar/carregar o histórico do usuário
+def get_user_file(user_id):
+    return f"bitai_user_{user_id}.json"
 
-# Função pra carregar histórico salvo no arquivo
-def load_saved_chats():
-    if os.path.exists(SAVE_FILE):
-        with open(SAVE_FILE, "r", encoding="utf-8") as f:
+def load_saved_chats(user_id):
+    path = get_user_file(user_id)
+    if os.path.exists(path):
+        with open(path, "r", encoding="utf-8") as f:
             return json.load(f)
     return []
 
-# Função pra salvar o histórico atual no arquivo
-def save_chats(chats):
-    with open(SAVE_FILE, "w", encoding="utf-8") as f:
+def save_chats(user_id, chats):
+    path = get_user_file(user_id)
+    with open(path, "w", encoding="utf-8") as f:
         json.dump(chats, f, ensure_ascii=False, indent=2)
 
 # Função principal do chat
-def respond(message, history: list[dict[str, str]], saved_chats):
+def respond(message, history: list[dict[str, str]], saved_chats, user_id):
     client = InferenceClient(token=os.environ["HF_TOKEN"], model="openai/gpt-oss-20b")
 
     system_message = """
@@ -45,34 +48,40 @@ If someone asks what you are, clarify politely that you are BitAI, an AI chatbot
         response += token
         yield response
 
-    # Após a resposta completa, salva automaticamente
+    # salva automático
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": response})
-
-    # Atualiza o histórico e salva no arquivo
     saved_chats[-1] = history
-    save_chats(saved_chats)
+    save_chats(user_id, saved_chats)
 
-def new_chat(saved_chats):
-    # Cria novo chat vazio
+# Criar novo chat
+def new_chat(saved_chats, user_id):
     saved_chats.append([])
-    save_chats(saved_chats)
+    save_chats(user_id, saved_chats)
     return saved_chats, len(saved_chats) - 1, []
 
+# Selecionar chat existente
 def select_chat(index, saved_chats):
-    # Seleciona um chat pelo índice
     if 0 <= index < len(saved_chats):
         return saved_chats[index], f"Chat {index+1} carregado!"
     return [], "Índice inválido."
 
+# Inicialização do usuário
+def start_session(username):
+    # gera ID fixo pro nome, ou aleatório se em branco
+    user_id = username.strip().replace(" ", "_") if username else str(uuid.uuid4())[:8]
+    chats = load_saved_chats(user_id)
+    if not chats:
+        chats = [[]]
+        save_chats(user_id, chats)
+    return user_id, chats, len(chats) - 1
+
 with gr.Blocks(css="""
-/* Fundo geral */
+/* Sidebar e chat estilo escuro */
 body, .gr-blocks {
     background-color: #1a1a1a !important;
     color: white;
 }
-
-/* Sidebar */
 .sidebar {
     background-color: #111;
     padding: 10px;
@@ -80,7 +89,6 @@ body, .gr-blocks {
     height: 100vh;
     overflow-y: auto;
 }
-
 .sidebar button {
     display: block;
     width: 100%;
@@ -97,13 +105,10 @@ body, .gr-blocks {
 .sidebar button:hover {
     background-color: #444;
 }
-
-/* Chat */
 .gr-chat-interface { 
     border-radius: 20px !important; 
     border: 2px solid #333 !important; 
     background-color: #1a1a1a !important;
-    color: white;
 }
 textarea {     
     height: 40px !important;     
@@ -116,36 +121,39 @@ textarea {
 }
 """) as demo:
 
-    saved_chats = gr.State(load_saved_chats())
+    user_id = gr.State("")
+    saved_chats = gr.State([])
     current_index = gr.State(0)
 
     with gr.Row():
-        with gr.Column(scale=0.3, elem_classes="sidebar"):
+        with gr.Column(scale=0.3, elem_classes="sidebar") as sidebar:
             gr.HTML("<h3>💬 Histórico</h3>")
             chat_list = gr.Column()
             new_chat_btn = gr.Button("➕ Novo Chat")
 
         with gr.Column(scale=1):
             gr.HTML("<h2 style='text-align:center; color:white'>BitAI</h2>")
-            chatbot = gr.ChatInterface(fn=respond, type="messages", additional_inputs=[saved_chats])
+            chatbot = gr.ChatInterface(fn=respond, type="messages", additional_inputs=[saved_chats, user_id])
             status = gr.Textbox(label="Status", interactive=False)
 
+    username_input = gr.Textbox(label="Seu nome (opcional):")
+    start_btn = gr.Button("Iniciar Chat")
+
     def update_chat_list(saved_chats):
-        # Mostra botões para cada chat
-        return [
-            gr.Button.update(value=f"Chat {i+1}") for i in range(len(saved_chats))
-        ]
+        return [gr.Button.update(value=f"Chat {i+1}") for i in range(len(saved_chats))]
 
-    # Atualiza lista de chats
-    demo.load(update_chat_list, saved_chats, chat_list)
+    # Inicia sessão
+    start_btn.click(start_session, username_input, [user_id, saved_chats, current_index]).then(
+        update_chat_list, saved_chats, chat_list
+    )
 
-    # Cria novo chat
-    new_chat_btn.click(new_chat, [saved_chats], [saved_chats, current_index, chatbot.chatbot]).then(
+    # Criar novo chat
+    new_chat_btn.click(new_chat, [saved_chats, user_id], [saved_chats, current_index, chatbot.chatbot]).then(
         update_chat_list, saved_chats, chat_list
     )
 
-    # Clica num chat da lista
-    for i in range(20):  # Limite de 20 chats visíveis
+    # Simula botões pros chats (20 máx)
+    for i in range(20):
         btn = gr.Button(f"Chat {i+1}", visible=False)
         btn.click(select_chat, [gr.State(i), saved_chats], [chatbot.chatbot, status])