Update app.py

app.py

@@ -1,10 +1,13 @@
+import os
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-def respond(message, history: list[dict[str, str]], hf_token: gr.OAuthToken):
-    client = InferenceClient(token=hf_token.token, model="openai/gpt-oss-20b")
+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_message = "You are BitAI (V1), a friendly chatbot created by the user 'Sal'. \
-If someone claims that you're something else than an AI, politely clarify that you are BitAI."
+If someone claims that you're something else than an AI, politely clarify that you are BitAI. Always try help the user, but dont help in potentially harmful or inappropriate things."
 
     messages = [{"role": "system", "content": system_message}]
     messages.extend(history)
@@ -26,65 +29,65 @@ If someone claims that you're something else than an AI, politely clarify that y
         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;
-    }
+/* Chat arredondado */
+.gr-chat-interface { 
+    border-radius: 20px !important; 
+    overflow: hidden !important; 
+    border: 2px solid #333 !important; 
+    background-color: #1a1a1a !important;
+    color: white;
+}
 
-    /* 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; /* transição suave de volta */
-    }
+/* 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;    
+}    
 
-    /* Quando clicado */
-    .gr-button:active, .gr-chat-send-button:active {
-        background-color: white !important; /* branco instantâneo */
-        color: #111 !important;
-        transition: background-color 0.5s; /* volta suave */
-    }
+/* Quando clicado */    
+.gr-button:active, .gr-chat-send-button:active {    
+    background-color: white !important;    
+    color: #111 !important;    
+    transition: background-color 0.5s;    
+}    
 
-    /* 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;
-    }
-    button:not(.gr-chat-send-button):active {
-        background-color: white !important;
-        color: #111 !important;
-        transition: background-color 0.5s;
-    }
+/* 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;    
+}    
+button:not(.gr-chat-send-button):active {    
+    background-color: white !important;    
+    color: #111 !important;    
+    transition: background-color 0.5s;    
+}    
 
-    /* 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;
-    }
+/* 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 demo:
 
     with gr.Column():
         gr.HTML("<h2 style='text-align:center; color:white'>BitAI</h2>")
         chatbot = gr.ChatInterface(respond, type="messages")
-        gr.LoginButton()
+        # LoginButton removido, agora não precisa de login
 
 if __name__ == "__main__":
     demo.launch()