Update app.py

app.py

@@ -1,10 +1,7 @@
-import os
+import os, json, uuid
 import gradio as gr
 from huggingface_hub import InferenceClient
-import json
-import uuid  # gera IDs únicos
 
-# Funções pra salvar/carregar o histórico do usuário
 def get_user_file(user_id):
     return f"bitai_user_{user_id}.json"
 
@@ -20,55 +17,38 @@ def save_chats(user_id, chats):
     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, user_id):
+def respond(message, history, saved_chats, user_id):
     client = InferenceClient(token=os.environ["HF_TOKEN"], model="openai/gpt-oss-20b")
 
-    system_message = """
-You are BitAI (V1), a friendly, curious, and talkative chatbot created by the user 'Sal'. 
-You can share opinions, answer casual questions, and chat about personal-style topics in a safe and friendly way. 
-Avoid repeating the same phrases, and always try to keep the conversation engaging and natural. 
-Politely refuse only things that are truly harmful, illegal, or unsafe. 
-If someone asks what you are, clarify politely that you are BitAI, an AI chatbot.
-"""
-
-    messages = [{"role": "system", "content": system_message}]
-    messages.extend(history)
-    messages.append({"role": "user", "content": message})
+    system = {
+        "role": "system",
+        "content": "You are BitAI (V1), a friendly chatbot created by Sal."
+    }
 
+    messages = [system] + history + [{"role": "user", "content": message}]
     response = ""
-    for msg in client.chat_completion(
-        messages,
-        max_tokens=2048,
-        stream=True,
-        temperature=0.7,
-        top_p=0.95,
-    ):
+    for msg in client.chat_completion(messages, max_tokens=2048, stream=True):
         token = msg.choices[0].delta.content if msg.choices and msg.choices[0].delta else ""
         response += token
         yield response
 
-    # salva automático
+    # Salva automático
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": response})
     saved_chats[-1] = history
     save_chats(user_id, saved_chats)
 
-# Criar novo chat
 def new_chat(saved_chats, user_id):
     saved_chats.append([])
     save_chats(user_id, saved_chats)
     return saved_chats, len(saved_chats) - 1, []
 
-# Selecionar chat existente
 def select_chat(index, saved_chats):
     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:
@@ -76,8 +56,25 @@ def start_session(username):
         save_chats(user_id, chats)
     return user_id, chats, len(chats) - 1
 
+def render_sidebar(saved_chats):
+    html = "<div class='chatlist'>"
+    for i, chat in enumerate(saved_chats):
+        preview = ""
+        for m in chat:
+            if m["role"] == "user":
+                preview = m["content"][:35] + ("..." if len(m["content"]) > 35 else "")
+                break
+        html += f"""
+        <div class='chat-item' onclick="selectChat({i})">
+            <div class='chat-title'>💬 Chat {i+1}</div>
+            <div class='chat-preview'>{preview or 'Conversa vazia'}</div>
+        </div>
+        """
+    html += "</div>"
+    return html
+
 with gr.Blocks(css="""
-/* Sidebar e chat estilo escuro */
+/* Sidebar com vibes modernas */
 body, .gr-blocks {
     background-color: #1a1a1a !important;
     color: white;
@@ -89,36 +86,17 @@ body, .gr-blocks {
     height: 100vh;
     overflow-y: auto;
 }
-.sidebar button {
-    display: block;
-    width: 100%;
-    margin-bottom: 10px;
-    background-color: #222;
-    color: white;
+.chatlist { display: flex; flex-direction: column; gap: 8px; }
+.chat-item {
+    background: #222;
     border-radius: 10px;
     padding: 10px;
-    text-align: left;
-    border: none;
     cursor: pointer;
-    transition: background-color 0.3s;
-}
-.sidebar button:hover {
-    background-color: #444;
-}
-.gr-chat-interface { 
-    border-radius: 20px !important; 
-    border: 2px solid #333 !important; 
-    background-color: #1a1a1a !important;
-}
-textarea {     
-    height: 40px !important;     
-    border-radius: 20px !important;     
-    border: 1px solid #444 !important;     
-    padding: 8px !important;    
-    background-color: #111;    
-    color: white;    
-    resize: none !important;    
+    transition: all 0.2s ease;
 }
+.chat-item:hover { background: #333; transform: scale(1.02); }
+.chat-title { font-weight: bold; color: #fff; }
+.chat-preview { font-size: 13px; color: #bbb; margin-top: 2px; }
 """) as demo:
 
     user_id = gr.State("")
@@ -126,36 +104,25 @@ textarea {
     current_index = gr.State(0)
 
     with gr.Row():
-        with gr.Column(scale=0.3, elem_classes="sidebar") as sidebar:
-            gr.HTML("<h3>💬 Histórico</h3>")
-            chat_list = gr.Column()
+        with gr.Column(scale=0.3, elem_classes="sidebar"):
+            sidebar_html = gr.HTML("<div class='chatlist'></div>")
             new_chat_btn = gr.Button("➕ Novo Chat")
 
         with gr.Column(scale=1):
-            gr.HTML("<h2 style='text-align:center; color:white'>BitAI</h2>")
+            gr.HTML("<h2 style='text-align:center;'>BitAI</h2>")
             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):
-        return [gr.Button.update(value=f"Chat {i+1}") for i in range(len(saved_chats))]
-
-    # Inicia sessão
     start_btn.click(start_session, username_input, [user_id, saved_chats, current_index]).then(
-        update_chat_list, saved_chats, chat_list
+        render_sidebar, saved_chats, sidebar_html
     )
 
-    # 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
+        render_sidebar, saved_chats, sidebar_html
     )
 
-    # 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])
-
 if __name__ == "__main__":
     demo.launch()