Update app.py

app.py

@@ -1,128 +1,100 @@
-import os, json, uuid
-import gradio as gr
+import os, json, uuid, gradio as gr
 from huggingface_hub import InferenceClient
 
-def get_user_file(user_id):
-    return f"bitai_user_{user_id}.json"
+# Usa o modelo certo
+client = InferenceClient("openai/gpt-oss-20b", token=os.getenv("HF_TOKEN"))
 
-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:
+os.makedirs("bitai_data", exist_ok=True)
+
+def get_user_file(username):
+    if not username:
+        username = str(uuid.uuid4())[:8]
+    return f"bitai_data/{username}.json"
+
+def load_history(username):
+    file = get_user_file(username)
+    if os.path.exists(file):
+        with open(file, "r") as f:
             return json.load(f)
     return []
 
-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)
+def save_history(username, history):
+    with open(get_user_file(username), "w") as f:
+        json.dump(history, f)
 
-def respond(message, history, saved_chats, user_id):
-    client = InferenceClient(token=os.environ["HF_TOKEN"], model="openai/gpt-oss-20b")
+def list_chats():
+    files = os.listdir("bitai_data")
+    return [f.replace(".json", "") for f in files]
+
+def respond(message, history, username):
+    history.append({"role": "user", "content": message})
+    save_history(username, history)
 
-    system = {
-        "role": "system",
-        "content": "You are BitAI (V1), a friendly chatbot created by Sal."
-    }
+    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.
+Politely refuse only things that are truly harmful, illegal, or unsafe.
+"""
+
+    messages = [{"role": "system", "content": system_message}]
+    messages.extend(history)
 
-    messages = [system] + history + [{"role": "user", "content": message}]
     response = ""
-    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 ""
+    for msg in client.chat_completion(messages, max_tokens=1024, stream=True):
+        choices = msg.choices
+        token = ""
+        if len(choices) and choices[0].delta.content:
+            token = choices[0].delta.content
         response += token
-        yield response
 
-    # 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)
-
-def new_chat(saved_chats, user_id):
-    saved_chats.append([])
-    save_chats(user_id, saved_chats)
-    return saved_chats, len(saved_chats) - 1, []
-
-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."
-
-def start_session(username):
-    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
-
-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 com vibes modernas */
-body, .gr-blocks {
-    background-color: #1a1a1a !important;
-    color: white;
-}
+    save_history(username, history)
+    return history, history
+
+def select_chat(username):
+    return load_history(username), username
+
+with gr.Blocks(theme="soft", css="""
 .sidebar {
-    background-color: #111;
-    padding: 10px;
-    border-right: 2px solid #333;
-    height: 100vh;
-    overflow-y: auto;
+  width: 220px;
+  background-color: #1f1f1f;
+  color: white;
+  height: 100%;
+  padding: 10px;
+  border-right: 2px solid #333;
+  overflow-y: auto;
 }
-.chatlist { display: flex; flex-direction: column; gap: 8px; }
-.chat-item {
-    background: #222;
-    border-radius: 10px;
-    padding: 10px;
-    cursor: pointer;
-    transition: all 0.2s ease;
+.chat-btn {
+  display: block;
+  width: 100%;
+  margin-bottom: 6px;
+  padding: 8px;
+  text-align: left;
+  background: #2c2c2c;
+  color: white;
+  border-radius: 6px;
+  border: none;
+  cursor: pointer;
+  transition: 0.2s;
 }
-.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("")
-    saved_chats = gr.State([])
-    current_index = gr.State(0)
+.chat-btn:hover {
+  background: #444;
+}
+""") as app:
 
     with gr.Row():
-        with gr.Column(scale=0.3, elem_classes="sidebar"):
-            sidebar_html = gr.HTML("<div class='chatlist'></div>")
+        with gr.Column(elem_classes="sidebar"):
+            gr.Markdown("### 💬 Chats")
+            chat_list = gr.Dropdown(choices=list_chats(), label="Selecionar Chat", interactive=True)
             new_chat_btn = gr.Button("➕ Novo Chat")
+        with gr.Column(scale=3):
+            username = gr.Textbox(label="Usuário", placeholder="Digite seu nome (ou deixe vazio)")
+            chatbot = gr.Chatbot(label="GPT OSS 20B", height=500)
+            msg = gr.Textbox(placeholder="Mensagem...")
+            send = gr.Button("Enviar")
 
-        with gr.Column(scale=1):
-            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")
-
-    start_btn.click(start_session, username_input, [user_id, saved_chats, current_index]).then(
-        render_sidebar, saved_chats, sidebar_html
-    )
-
-    new_chat_btn.click(new_chat, [saved_chats, user_id], [saved_chats, current_index, chatbot.chatbot]).then(
-        render_sidebar, saved_chats, sidebar_html
-    )
+    send.click(respond, [msg, chatbot, username], [chatbot, chatbot])
+    new_chat_btn.click(lambda: ([], ""), None, [chatbot, username])
+    chat_list.change(select_chat, [chat_list], [chatbot, username])
 
-if __name__ == "__main__":
-    demo.launch()
+app.launch()