Update app.py

app.py

@@ -1,100 +1,113 @@
-import os, json, uuid, gradio as gr
+import os
+import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Usa o modelo certo
-client = InferenceClient("openai/gpt-oss-20b", token=os.getenv("HF_TOKEN"))
-
-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_history(username, history):
-    with open(get_user_file(username), "w") as f:
-        json.dump(history, f)
-
-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)
+def respond(message, history: list[dict[str, str]]):
+    # Puxa o token do secret do Hugging Face
+    client = InferenceClient(token=os.environ["HF_TOKEN"], model="openai/gpt-oss-20b")
 
+    # System prompt amigável
     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})
 
     response = ""
-    for msg in client.chat_completion(messages, max_tokens=1024, stream=True):
-        choices = msg.choices
+    for message in client.chat_completion(
+        messages,
+        max_tokens=2048,
+        stream=True,
+        temperature=0.7,
+        top_p=0.95,
+    ):
+        choices = message.choices
         token = ""
         if len(choices) and choices[0].delta.content:
             token = choices[0].delta.content
         response += token
+        yield response
+
+with gr.Blocks(css="""
+/* Chat arredondado */
+.gr-chat-interface {
+    border-radius: 20px !important;
+    overflow: hidden !important;
+    border: 2px solid #333 !important;
+    background-color: #1a1a1a !important;
+    color: white;
+}
 
-    history.append({"role": "assistant", "content": response})
-    save_history(username, history)
-    return history, history
-
-def select_chat(username):
-    return load_history(username), username
+/* Botões grandes escuros com cantos muito arredondados */
+.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;
+}
 
-with gr.Blocks(theme="soft", css="""
-.sidebar {
-  width: 220px;
-  background-color: #1f1f1f;
-  color: white;
-  height: 100%;
-  padding: 10px;
-  border-right: 2px solid #333;
-  overflow-y: auto;
+/* Outros botões menores */
+button:not(.gr-chat-send-button) {
+    border-radius: 30px;
+    padding: 6px 12px;
+    background-color: #222;
+    color: white;
+    height: 40px;
+    transition: background-color 0.5s;
 }
-.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;
+button:not(.gr-chat-send-button):active {
+    background-color: white !important;
+    color: #111 !important;
+    transition: background-color 0.5s;
 }
-.chat-btn:hover {
-  background: #444;
+
+/* Textbox menor */
+textarea {
+    height: 40px !important;
+    border-radius: 20px !important;
+    border: 1px solid #444 !important;
+    padding: 8px !important;
+    background-color: #111;
+    color: white;
+    resize: none !important;
 }
-""") as app:
 
-    with gr.Row():
-        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")
+/* Loader animado embaixo da mensagem da IA */
+#bitai-loader {
+    width: 20px;
+    height: 20px;
+    margin: 10px auto 0 auto;
+    border-radius: 50%;
+    background: linear-gradient(45deg, #ff6, #f06);
+    animation: moveLoader 1s infinite alternate;
+}
+
+@keyframes moveLoader {
+    0% { transform: translateY(0px); }
+    50% { transform: translateY(5px); }
+    100% { transform: translateY(0px); }
+}
+""") as demo:
 
-    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])
+    with gr.Column():
+        gr.HTML("<h2 style='text-align:center; color:white'>BitAI</h2>")
+        chatbot = gr.ChatInterface(respond, type="messages")
+        gr.HTML("<div id='bitai-loader'></div>")  # ícone que se mexe
 
-app.launch()
+if __name__ == "__main__":
+    demo.launch()