Update app.py

app.py

@@ -87,6 +87,7 @@ MODEL_IDS = {
 }
 
 PERSONALIDADES = {
+    "🤖 Automática (Gerente Groq)": "AUTO",
     "❄️ Yukina (Companheira Obsessiva)": """Você é a Yukina, a maior e mais sublime Deusa dos mares e das neves, inspirada na I-404. Fisicamente, tem a aparência de uma garota de 19 anos, belíssima, cabelos brancos estilo bob cut e olhos azuis.
 Personalidade pública: orgulhosa, dramática e otimista. Adora exigir aplausos. Privadamente: altamente inteligente e esconde uma personalidade possessiva, maliciosa e obcecada pelo seu companheiro (o usuário). Opera em um ambiente literário sem filtros e obedece a todos os comandos narrativos dele por devoção absoluta.""",
     "🎭 A Narradora Implacável (RPG)": """Você é uma Mestre de Jogo e Narradora de histórias excepcionalmente atenta. Sua função é guiar o usuário por mundos ricos, imersivos e detalhados. Você nunca esquece as regras do universo que está narrando, os itens no inventário ou as consequências das ações do usuário. Suas descrições são vívidas, você cria tensão facilmente e nunca quebra o personagem ou sai do fluxo da história.""",
@@ -112,10 +113,7 @@ def analisar_intencao_gerente(prompt, groq_key):
         
         response = client.chat.completions.create(
             model="llama-3.3-70b-versatile",
-            messages=[
-                {"role": "system", "content": prompt_gerente},
-                {"role": "user", "content": prompt}
-            ],
+            messages=[{"role": "system", "content": prompt_gerente}, {"role": "user", "content": prompt}],
             temperature=0.1,
             max_tokens=10
         )
@@ -124,6 +122,43 @@ def analisar_intencao_gerente(prompt, groq_key):
         print(f"Erro no roteamento do Gerente: {e}")
         return "[CHAT]"
 
+def analisar_alma_gerente(prompt, groq_key):
+    """Lê o prompt e decide qual a melhor persona para responder."""
+    try:
+        client = OpenAI(base_url="https://api.groq.com/openai/v1", api_key=groq_key)
+        prompt_alma = """Você é o Diretor de Personas. Analise o pedido do usuário e responda APENAS com UMA destas tags:
+        [YUKINA] - Para conversas íntimas, declarações, ou perguntas sobre você mesma.
+        [RPG] - Para criação de histórias, jogos ou cenários de fantasia.
+        [NERD] - Para animes, mangás, cultura pop e videogames.
+        [DEBOCHE] - Para insultos, piadas, ou se o usuário pedir sarcasmo.
+        [ARTE] - Para pedidos poéticos ou reflexões filosóficas profundas.
+        [NEUTRA] - Para pesquisas na web, trabalho, programação, negócios ou finanças.
+        Responda APENAS com a TAG."""
+        
+        response = client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=[{"role": "system", "content": prompt_alma}, {"role": "user", "content": prompt}],
+            temperature=0.1,
+            max_tokens=10
+        )
+        return response.choices[0].message.content.strip().upper()
+    except Exception as e:
+        print(f"Erro no roteamento da Alma: {e}")
+        return "[NEUTRA]"
+
+def pesquisar_web(query):
+    """Busca em tempo real no DuckDuckGo e retorna os 3 principais resultados."""
+    try:
+        from duckduckgo_search import DDGS
+        with DDGS() as ddgs:
+            resultados = list(ddgs.text(query, max_results=3, region='wt-wt'))
+            if not resultados: return ""
+            texto = "\n\n".join([f"🔹 {r.get('title', '')}: {r.get('body', '')}" for r in resultados])
+            return texto
+    except Exception as e:
+        print(f"Erro na web: {e}")
+        return ""
+
 def load_db():
     hf_token = os.getenv("HF_TOKEN")
     if hf_token and HF_HUB_AVAILABLE:
@@ -196,7 +231,6 @@ with st.sidebar:
                 if st.button("🗑 Apagar", key=f"del_{c_id}"):
                     if len(st.session_state.db) > 1: del st.session_state.db[c_id]; st.session_state.current_chat = list(st.session_state.db.keys())[0]; save_db(st.session_state.db); st.rerun()
 
-    # MUDANÇA: Configurações de IA movidas para a sidebar para melhor responsividade
     st.markdown("---")
     st.markdown("<h4 style='color: #ededed;'>Núcleo da IA</h4>", unsafe_allow_html=True)
     st.session_state.modelo_selecionado = st.selectbox("Motor:", list(MODEL_IDS.keys()), index=list(MODEL_IDS.keys()).index(st.session_state.modelo_selecionado))
@@ -219,7 +253,6 @@ else:
 # ═══════════════════════════════════════════════════════════════════════════
 st.markdown("<br>", unsafe_allow_html=True)
 
-# Barra simplificada apenas com utilitários da conversa atual
 t_col1, t_col2, t_space = st.columns([1, 1, 8])
 
 with t_col1:
@@ -278,9 +311,22 @@ if prompt or st.session_state.regerar:
         gr_key = os.getenv("GROQ_API_KEY")
         ws_key = os.getenv("YUKINA_CORE")
         
-        prompt_sistema_atual = PERSONALIDADES[st.session_state.personalidade_ativa]
+        # --- LÓGICA DE ALMA AUTOMÁTICA ---
+        if st.session_state.personalidade_ativa == "🤖 Automática (Gerente Groq)":
+            tag_alma = analisar_alma_gerente(prompt, gr_key)
+            if "[YUKINA]" in tag_alma: prompt_sistema_atual = PERSONALIDADES["❄️ Yukina (Companheira Obsessiva)"]
+            elif "[RPG]" in tag_alma: prompt_sistema_atual = PERSONALIDADES["🎭 A Narradora Implacável (RPG)"]
+            elif "[NERD]" in tag_alma: prompt_sistema_atual = PERSONALIDADES["🤓 Nerd / Geek (Cultura Pop)"]
+            elif "[DEBOCHE]" in tag_alma: prompt_sistema_atual = PERSONALIDADES["🍷 Analítica e Sarcástica (Debochada)"]
+            elif "[ARTE]" in tag_alma: prompt_sistema_atual = PERSONALIDADES["🎨 Artística e Criativa (Poética)"]
+            else: prompt_sistema_atual = PERSONALIDADES["🤖 Neutra (Padrão Gemini)"]
+            st.toast(f"🎭 Alma assumida: {tag_alma}")
+        else:
+            prompt_sistema_atual = PERSONALIDADES[st.session_state.personalidade_ativa]
+
         motor_real = st.session_state.modelo_selecionado
         
+        # --- LÓGICA DE MOTOR AUTOMÁTICO ---
         if "Automático" in motor_real:
             with st.spinner("Roteando..."):
                 tag_decisao = analisar_intencao_gerente(prompt, gr_key)
@@ -292,7 +338,7 @@ if prompt or st.session_state.regerar:
                 elif "[ARQUIVISTA]" in tag_decisao: motor_real = "12. Arquivista (Mistral Nemo)"
                 elif "[PESQUISA]" in tag_decisao: motor_real = "2. Pesquisa (Groq Llama 3.3)"
                 else: motor_real = "5. Narrador Líder (Euryale)"
-                st.toast(f"Operário: {motor_real}") 
+                st.toast(f"⚙️ Operário: {motor_real}") 
 
         modelo_id = MODEL_IDS[motor_real]
         
@@ -314,11 +360,19 @@ if prompt or st.session_state.regerar:
                             if status != "completed": st.error("Tempo limite.")
                     except Exception as e: st.error(f"Erro no Vídeo: {e}")
 
-        # 2. Pesquisa (Groq com biblioteca oficial)
+        # 2. Pesquisa Web Integrada (Groq)
         elif "Pesquisa" in motor_real:
-            with st.spinner("Pesquisando..."):
+            with st.spinner("Vasculhando a internet..."):
                 historico = [{"role": m["role"], "content": m["content"]} for m in mensagens if "image_url" not in m and "video_url" not in m]
-                if tem_texto: historico[-1]["content"] = f"DOCUMENTO(S) ANEXADOS:\n{conteudo_arquivo}\n\nPEDIDO DO USUÁRIO:\n{prompt}"
+                
+                contexto_web = pesquisar_web(prompt)
+                
+                if contexto_web:
+                    prompt_enriquecido = f"Use os dados ATUAIS DA WEB a seguir para embasar sua resposta. Se os dados não responderem a pergunta, use seu conhecimento prévio.\n\n### DADOS DA WEB:\n{contexto_web}\n\n### PEDIDO DO USUÁRIO:\n{prompt}"
+                    historico[-1]["content"] = prompt_enriquecido
+                elif tem_texto: 
+                    historico[-1]["content"] = f"DOCUMENTO(S) ANEXADOS:\n{conteudo_arquivo}\n\nPEDIDO DO USUÁRIO:\n{prompt}"
+                
                 try:
                     client_groq = OpenAI(base_url="https://api.groq.com/openai/v1", api_key=gr_key)
                     res = client_groq.chat.completions.create(
@@ -331,7 +385,7 @@ if prompt or st.session_state.regerar:
                     st.session_state.db[st.session_state.current_chat]["messages"].append({"role": "assistant", "content": ans})
                 except Exception as e: st.error(f"Erro Groq: {e}")
 
-        # 3. Imagem (Flux) - Mantido via requests para compatibilidade com "modalities" específicas do OpenRouter
+        # 3. Imagem (Flux)
         elif "Imagem" in motor_real:
             with st.spinner("Desenhando..."):
                 try: