Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Função principal da IA
 def respond(message, history: list[dict[str, str]], hf_token: gr.OAuthToken):
     client = InferenceClient(token=hf_token.token, model="openai/gpt-oss-20b")
 
@@ -15,7 +14,6 @@ Keep a simple, approachable, and friendly tone."""
     messages.append({"role": "user", "content": message})
 
     response = ""
-    # Stream das respostas
     for message_chunk in client.chat_completion(
         messages,
         max_tokens=512,
@@ -28,9 +26,8 @@ Keep a simple, approachable, and friendly tone."""
         if len(choices) and choices[0].delta.content:
             token = choices[0].delta.content
         response += token
-        yield response  # só retorna a resposta
+        yield response
 
-# HTML e JS do ícone BitAI embaixo do chat
 bit_icon_html = """
 <div style="text-align:center; margin-top:10px;">
   <div class="wrap" id="bit" style="width:50px; height:50px; margin:auto;">
@@ -44,13 +41,17 @@ bit_icon_html = """
 </div>
 
 <script>
+function initBitAI() {
   const wrap = document.getElementById('bit');
+  
+  // Espera chat existir
+  const chatContainer = document.querySelector('.chat-interface');
+  if(!chatContainer) return setTimeout(initBitAI, 500);
 
-  // Observa novas mensagens do bot e dispara animação thinking
   const observer = new MutationObserver((mutationsList) => {
     mutationsList.forEach(mutation => {
       mutation.addedNodes.forEach(node => {
-        if (node.classList && node.classList.contains('bot')) {
+        if(node.classList && node.classList.contains('bot')) {
           wrap.classList.add('thinking');
           setTimeout(()=>{ wrap.classList.remove('thinking'); }, 1200);
         }
@@ -58,8 +59,12 @@ bit_icon_html = """
     });
   });
 
-  const chatContainer = document.querySelector('.gradio-chatbot-container');
-  if(chatContainer) observer.observe(chatContainer, { childList: true, subtree: true });
+  observer.observe(chatContainer, { childList: true, subtree: true });
+}
+
+// Inicia observador após o DOM carregar
+document.addEventListener('DOMContentLoaded', initBitAI);
+initBitAI();
 </script>
 
 <style>
@@ -92,7 +97,6 @@ with gr.Blocks(css="""
         gr.LoginButton()
     
     chatbot.render()
-    # Ícone animado no final do chat
     gr.HTML(bit_icon_html)
 
 if __name__ == "__main__":