Update app.py

app.py

@@ -1,9 +1,24 @@
 import os
 import gradio as gr
 from huggingface_hub import InferenceClient
+import json
 
+SAVE_FILE = "bitai_history.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:
+            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:
+        json.dump(chats, f, ensure_ascii=False, indent=2)
+
+# Função principal do chat
 def respond(message, history: list[dict[str, str]], saved_chats):
-    # Puxa o token do secret do Hugging Face
     client = InferenceClient(token=os.environ["HF_TOKEN"], model="openai/gpt-oss-20b")
 
     system_message = """
@@ -26,62 +41,70 @@ If someone asks what you are, clarify politely that you are BitAI, an AI chatbot
         temperature=0.7,
         top_p=0.95,
     ):
-        choices = msg.choices
-        token = ""
-        if len(choices) and choices[0].delta.content:
-            token = choices[0].delta.content
+        token = msg.choices[0].delta.content if msg.choices and msg.choices[0].delta else ""
         response += token
         yield response
 
-def save_chat(history, saved_chats):
-    # Salva o histórico atual
-    if history:
-        saved_chats.append(history.copy())
-    return saved_chats, "Chat salvo! ✅"
+    # Após a resposta completa, salva automaticamente
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": response})
 
-def load_chat(index, saved_chats):
-    # Carrega um chat salvo
+    # Atualiza o histórico e salva no arquivo
+    saved_chats[-1] = history
+    save_chats(saved_chats)
+
+def new_chat(saved_chats):
+    # Cria novo chat vazio
+    saved_chats.append([])
+    save_chats(saved_chats)
+    return saved_chats, len(saved_chats) - 1, []
+
+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 saved_chats[index], f"Chat {index+1} carregado!"
     return [], "Índice inválido."
 
 with gr.Blocks(css="""
-/* Mesmo CSS que você fez */
+/* Fundo geral */
+body, .gr-blocks {
+    background-color: #1a1a1a !important;
+    color: white;
+}
+
+/* Sidebar */
+.sidebar {
+    background-color: #111;
+    padding: 10px;
+    border-right: 2px solid #333;
+    height: 100vh;
+    overflow-y: auto;
+}
+
+.sidebar button {
+    display: block;
+    width: 100%;
+    margin-bottom: 10px;
+    background-color: #222;
+    color: white;
+    border-radius: 10px;
+    padding: 10px;
+    text-align: left;
+    border: none;
+    cursor: pointer;
+    transition: background-color 0.3s;
+}
+.sidebar button:hover {
+    background-color: #444;
+}
+
+/* Chat */
 .gr-chat-interface { 
     border-radius: 20px !important; 
-    overflow: hidden !important; 
     border: 2px solid #333 !important; 
     background-color: #1a1a1a !important;
     color: white;
 }
-.gr-button, .gr-chat-send-button {    
-    border-radius: 50px;     
-    padding: 12px 20px;    
-    background-color: #111;     
-    color: white;    
-    font-weight: bold;    
-    cursor: pointer;    
-    height: 50px;    
-    transition: background-color 0.5s;    
-}    
-.gr-button:active, .gr-chat-send-button:active {    
-    background-color: white !important;    
-    color: #111 !important;    
-    transition: background-color 0.5s;    
-}    
-button:not(.gr-chat-send-button) {    
-    border-radius: 30px;    
-    padding: 6px 12px;    
-    background-color: #222;    
-    color: white;    
-    height: 40px;    
-    transition: background-color 0.5s;    
-}    
-button:not(.gr-chat-send-button):active {    
-    background-color: white !important;    
-    color: #111 !important;    
-    transition: background-color 0.5s;    
-}    
 textarea {     
     height: 40px !important;     
     border-radius: 20px !important;     
@@ -93,22 +116,38 @@ textarea {
 }
 """) as demo:
 
-    saved_chats = gr.State([])  # Lista com todos os chats salvos
-
-    with gr.Column():
-        gr.HTML("<h2 style='text-align:center; color:white'>BitAI</h2>")
-        chatbot = gr.ChatInterface(fn=respond, type="messages", additional_inputs=[saved_chats])
-        
-        with gr.Row():
-            save_btn = gr.Button("💾 Salvar Chat")
-            load_index = gr.Number(label="Número do chat pra carregar", precision=0)
-            load_btn = gr.Button("📂 Carregar Chat")
-        
-        status = gr.Textbox(label="Status", interactive=False)
-
-        # Liga os botões
-        save_btn.click(save_chat, [chatbot.chatbot, saved_chats], [saved_chats, status])
-        load_btn.click(load_chat, [load_index, saved_chats], [chatbot.chatbot, status])
+    saved_chats = gr.State(load_saved_chats())
+    current_index = gr.State(0)
+
+    with gr.Row():
+        with gr.Column(scale=0.3, elem_classes="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])
+            status = gr.Textbox(label="Status", interactive=False)
+
+    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))
+        ]
+
+    # Atualiza lista de chats
+    demo.load(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(
+        update_chat_list, saved_chats, chat_list
+    )
+
+    # Clica num chat da lista
+    for i in range(20):  # Limite de 20 chats visíveis
+        btn = gr.Button(f"Chat {i+1}", visible=False)
+        btn.click(select_chat, [gr.State(i), saved_chats], [chatbot.chatbot, status])
 
 if __name__ == "__main__":
     demo.launch()