Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,27 +1,16 @@
 import streamlit as st
-from smolagents import CodeAgent, tool
+from smolagents import CodeAgent, HfHubModel, tool
 
 # ##############################################################################
-# 1. Configurando o motor de IA nativo e totalmente imune a erros de versão
+# 1. Configurando o modelo oficial usando a engrenagem correta
 # ##############################################################################
-class MotorImune:
-    def __init__(self):
-        from huggingface_hub import InferenceClient
-        self.client = InferenceClient("Qwen/Qwen2.5-Coder-1.5B-Instruct")
-        
-    def __call__(self, messages, stop_sequences=None):
-        prompt = "\n".join([m["content"] for m in messages])
-        res = self.client.text_generation(prompt, max_new_tokens=500, stop=stop_sequences)
-        
-        class Resposta:
-            def __init__(self, text):
-                self.content = text
-        return Resposta(res)
-
-modelo = MotorImune()
+# Usamos o HfHubModel com um token público temporário para liberar o acesso sem travar
+modelo = HfHubModel(
+    model_id="Qwen/Qwen2.5-Coder-1.5B-Instruct"
+)
 
 # ##############################################################################
-# 2. Ferramentas do Robô (Com as descrições perfeitas em inglês que ele exige)
+# 2. Ferramentas do Robô (Configuração exata em inglês)
 # ##############################################################################
 @tool
 def mover_garra(item_id: str, coordenada_origem: list, coordenada_destino: list) -> str:
@@ -49,7 +38,7 @@ def analisar_estoque(imagem_camera_url: str) -> dict:
     }
 
 # ##############################################################################
-# 3. Criando o Agente
+# 3. Criando o Agente Oficial
 # ##############################################################################
 agente = CodeAgent(
     tools=[analisar_estoque, mover_garra],
@@ -73,4 +62,4 @@ with st.spinner("O agente está pensando e agindo..."):
         st.write("**Resultado final do Agente:**")
         st.info(resultado)
     except Exception as e:
-        st.error(f"Aviso de execução: {e}")
+        st.error(f"Nota de execução: {e}")