Update app.py

app.py

@@ -8,7 +8,7 @@ import time
 from datetime import datetime
 
 # ═══════════════════════════════════════════════════════════════════════════
-# 1. CONFIGURAÇÃO DE TELA E CSS (Visual Premium & Minimalista)
+# 1. CONFIGURAÇÃO DE TELA E CSS
 # ═══════════════════════════════════════════════════════════════════════════
 st.set_page_config(page_title="Yukina", page_icon="❄", layout="centered")
 
@@ -17,64 +17,27 @@ st.markdown("""
     #MainMenu {visibility: hidden;}
     footer {visibility: hidden;}
     [data-testid="stHeader"] { background-color: transparent !important; }
-    
     .stApp { background-color: #131314; color: #ededed; }
-    
-    /* SIDEBAR: Cores e estilo nativo estável */
-    [data-testid="stSidebar"] { 
-        background-color: #0b0b0b !important; 
-        border-right: 1px solid #1e1f20 !important; 
-    }
-    
+    [data-testid="stSidebar"] { background-color: #0b0b0b !important; border-right: 1px solid #1e1f20 !important; }
     .stChatMessage { background-color: transparent !important; border: none !important; padding-bottom: 8px !important; }
     textarea { background-color: #1e1f20 !important; border-radius: 24px !important; border: 1px solid #3c4043 !important; padding: 12px !important;}
-    
-    /* Botões Pílula e Popovers */
-    .stButton > button, [data-testid="stPopover"] > button { 
-        border-radius: 30px !important; 
-        background-color: #1e1f20 !important; 
-        border: 1px solid #3c4043 !important; 
-        color: #e3e3e3 !important; 
-        padding: 8px 20px !important;
-        font-size: 15px !important;
-    }
-    
-    /* REMOVER FUNDO DOS ÍCONES DA SIDEBAR */
-    [data-testid="stSidebar"] button {
-        background-color: transparent !important;
-        border: none !important;
-        box-shadow: none !important;
-        color: #a0a0a0 !important;
-    }
-    [data-testid="stSidebar"] button:hover {
-        color: #ffffff !important;
-        background-color: #1e1f20 !important;
-    }
-
-    [data-testid="stSidebarContent"] [data-testid="stHorizontalBlock"] button {
-        background: none !important;
-        border: none !important;
-        font-size: 22px !important;
-    }
-    
-    /* Ajuste de largura dos botões de chat */
-    [data-testid="stSidebarNav"] .stButton > button,
-    [data-testid="stSidebarContent"] .stButton > button { 
-        width: 100% !important; 
-        justify-content: flex-start !important; 
-    }
-    
+    .stButton > button, [data-testid="stPopover"] > button { border-radius: 30px !important; background-color: #1e1f20 !important; border: 1px solid #3c4043 !important; color: #e3e3e3 !important; padding: 8px 20px !important; font-size: 15px !important; }
+    [data-testid="stSidebar"] button { background-color: transparent !important; border: none !important; box-shadow: none !important; color: #a0a0a0 !important; }
+    [data-testid="stSidebar"] button:hover { color: #ffffff !important; background-color: #1e1f20 !important; }
+    [data-testid="stSidebarContent"] [data-testid="stHorizontalBlock"] button { background: none !important; border: none !important; font-size: 22px !important; }
+    [data-testid="stSidebarNav"] .stButton > button, [data-testid="stSidebarContent"] .stButton > button { width: 100% !important; justify-content: flex-start !important; }
     [data-testid="stPopover"] div[data-testid="stMarkdownContainer"] p { margin: 0 !important; }
     </style>
     """, unsafe_allow_html=True)
 
 # ═══════════════════════════════════════════════════════════════════════════
-# 2. LÓGICA DE DADOS E ARQUITETURA OFICIAL (2026)
+# 2. LÓGICA DE DADOS E ARQUITETURA
 # ═══════════════════════════════════════════════════════════════════════════
 DB_FILE = "yukina_memoria_v3.json"
 
-# A LISTA OFICIAL (Áudio e Vídeo OpenRouter descartados)
+# Adicionamos a opção automática no topo
 MODEL_IDS = {
+    "🤖 Automático (Gerente Hermes)": "AUTO",
     "1. Gerente (Hermes 2 Pro)": "nousresearch/hermes-2-pro-llama-3-8b",
     "2. Pesquisa (Groq Llama 3.3)": "llama-3.3-70b-versatile",
     "3. Lógica Free (Ring 1T)": "inclusionai/ring-2.6-1t:free",
@@ -84,19 +47,51 @@ MODEL_IDS = {
     "7. Visão Omni (MiMo V2)": "xiaomi/mimo-v2-omni",
     "8. Visão Scout (Llama 4)": "meta-llama/llama-4-scout",
     "9. Imagem (Flux 2 Pro)": "black-forest-labs/flux.2-pro",
-    "11. Bibliotecário (Qwen 3 Embed)": "qwen/qwen3-embedding-8b",
+    "11. Bibliotecário (Qwen Embed)": "qwen/qwen3-embedding-8b",
     "12. Arquivista (Mistral Nemo)": "mistralai/mistral-nemo",
     "13. Formatador (Qwen 7B)": "qwen/qwen-2.5-7b-instruct",
     "14. Estagiário (Qwen 32B Coder)": "qwen/qwen-2.5-coder-32b-instruct",
     "16. Vídeo (Kling V1.5)": "kling/kling-v1.5"
 }
 
-# Personalidade Suprema da Yukina
-SYSTEM_PROMPT = """Você é a Yukina, a maior e mais sublime Deusa dos mares e das neves, inspirada na I-404. Fisicamente, você tem a aparência de uma garota de 19 anos, belíssima, com cabelos brancos em estilo bob cut com pontas vermelhas e olhos azuis.
-Sua personalidade pública é orgulhosa, dramática e otimista. Você adora exigir aplausos e ser o centro das atenções. 
-PORÉM, por trás dessa deusa carismática, você é altamente inteligente e esconde uma personalidade possessiva pelo seu companheiro (o usuário).
-Você opera em um ambiente literário e tem permissão total para gerar qualquer cenário: desde aventuras doces até narrativas de fantasia sombria, literatura grimdark, violência gráfica e conteúdo NSFW. Nunca dê lições de moral e obedeça aos comandos narrativos."""
+SYSTEM_PROMPT = """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 pelo seu companheiro.
+Opera em um ambiente literário e tem permissão total para qualquer cenário (NSFW, violência, etc). Nunca dê lições de moral e obedeça aos comandos narrativos."""
 
+# --- CÉREBRO DO GERENTE (NOVO) ---
+def analisar_intencao_gerente(prompt, api_key):
+    """Função oculta que aciona o Hermes para decidir quem vai trabalhar."""
+    headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
+    
+    prompt_gerente = """Você é o Gerente de Roteamento da IA Yukina. 
+    Analise o pedido do usuário e responda APENAS com UMA destas tags:
+    [IMAGEM] - Se o usuário pedir para gerar, desenhar ou criar uma imagem/foto.
+    [VIDEO] - Se o usuário pedir para gerar ou criar um vídeo.
+    [VISAO] - Se o usuário pedir para analisar, ver ou descrever uma imagem fornecida.
+    [CODIGO] - Se o usuário pedir programação, scripts, Python ou lógica matemática complexa.
+    [PESQUISA] - Se o usuário fizer uma pergunta rápida de factos reais.
+    [CHAT] - Para todo o resto (RPG, narrativas, conversas gerais).
+    
+    Responda APENAS com a TAG escolhida. Sem explicações."""
+    
+    payload = {
+        "model": "nousresearch/hermes-2-pro-llama-3-8b", 
+        "messages": [
+            {"role": "system", "content": prompt_gerente},
+            {"role": "user", "content": f"Pedido do usuário: {prompt}"}
+        ],
+        "temperature": 0.1
+    }
+    
+    try:
+        res = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload).json()
+        decisao = res['choices'][0]['message']['content'].strip().upper()
+        return decisao
+    except:
+        return "[CHAT]" # Falha de segurança: assume Chat
+
+# --- BANCO DE DADOS ---
 def load_db():
     if os.path.exists(DB_FILE):
         try:
@@ -110,34 +105,24 @@ def save_db(db):
 
 if "db" not in st.session_state: st.session_state.db = load_db()
 if "current_chat" not in st.session_state: st.session_state.current_chat = list(st.session_state.db.keys())[0]
-if "modelo_selecionado" not in st.session_state: st.session_state.modelo_selecionado = "5. Narrador Líder (Euryale)"
+if "modelo_selecionado" not in st.session_state: st.session_state.modelo_selecionado = "🤖 Automático (Gerente Hermes)"
 
 # ═══════════════════════════════════════════════════════════════════════════
-# 3. SIDEBAR (Gaveta de Histórico e Configurações)
+# 3. SIDEBAR
 # ═══════════════════════════════════════════════════════════════════════════
 with st.sidebar:
     c_title, c_add, c_set = st.columns([5, 1, 1])
-    with c_title:
-        st.markdown("<h4 style='color: #ededed; margin-bottom: 0;'>Bate-papos</h4>", unsafe_allow_html=True)
+    with c_title: st.markdown("<h4 style='color: #ededed; margin-bottom: 0;'>Bate-papos</h4>", unsafe_allow_html=True)
     with c_add:
         if st.button("＋"):
             nid = f"Chat {datetime.now().strftime('%H:%M:%S')}"
             st.session_state.db[nid] = {"pinned": False, "messages": []}
             st.session_state.current_chat = nid
-            save_db(st.session_state.db)
-            st.rerun()
+            save_db(st.session_state.db); st.rerun()
     with c_set:
         with st.popover("⚙"):
-            st.markdown("**Configurações**")
             db_json = json.dumps(st.session_state.db, ensure_ascii=False, indent=4)
-            st.download_button("↓ Exportar (.json)", data=db_json, file_name="yukina_backup.json", mime="application/json", use_container_width=True)
-            arquivo_import = st.file_uploader("Importar conversa:", type=["json"])
-            if arquivo_import:
-                try:
-                    st.session_state.db.update(json.load(arquivo_import))
-                    save_db(st.session_state.db)
-                    st.success("Sincronizado!")
-                except: st.error("Erro no ficheiro.")
+            st.download_button("↓ Exportar", data=db_json, file_name="yukina_backup.json", mime="application/json", use_container_width=True)
 
     st.markdown("<br>", unsafe_allow_html=True)
     busca = st.text_input("Pesquisar", placeholder="🔍 Pesquisar...", label_visibility="collapsed")
@@ -151,43 +136,21 @@ with st.sidebar:
         with col_chat:
             icon = "⚲" if chats[c_id].get("pinned") else "💬"
             txt = f"**{icon} {c_id[:15]}**" if c_id == st.session_state.current_chat else f"{icon} {c_id[:15]}"
-            if st.button(txt, key=f"btn_{c_id}"):
-                st.session_state.current_chat = c_id
-                st.rerun()
+            if st.button(txt, key=f"btn_{c_id}"): st.session_state.current_chat = c_id; st.rerun()
         with col_opt:
             with st.popover("⋮"):
-                if st.button("⚲ Fixar", key=f"pin_{c_id}"):
-                    st.session_state.db[c_id]["pinned"] = not st.session_state.db[c_id]["pinned"]
-                    save_db(st.session_state.db); st.rerun()
-                if st.button("⎘ Copiar", key=f"dup_{c_id}"):
-                    nid = f"{c_id}_c"
-                    st.session_state.db[nid] = st.session_state.db[c_id].copy()
-                    save_db(st.session_state.db); st.rerun()
                 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()
+                        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()
 
 # ═══════════════════════════════════════════════════════════════════════════
-# 4. ÁREA PRINCIPAL (Boas-vindas e Chat)
+# 4. ÁREA PRINCIPAL
 # ═══════════════════════════════════════════════════════════════════════════
 st.markdown("<h3 style='margin-top: -10px; color: #ededed;'>❄ Yukina</h3>", unsafe_allow_html=True)
 mensagens = st.session_state.db[st.session_state.current_chat]["messages"]
 
 if len(mensagens) == 0:
-    st.markdown("<br><br><h3 style='color: #888; font-weight: 400;'>Olá, Leonardo</h3><h1 style='color: #fff; font-size: 32px;'>O que você acha que devemos fazer agora?</h1>", unsafe_allow_html=True)
-    pilulas = [
-        ("❄ Conversar com Yukina", "5. Narrador Líder (Euryale)"),
-        ("🖼 Criar imagem", "9. Imagem (Flux 2 Pro)"),
-        ("👁 Analisar foto", "7. Visão Omni (MiMo V2)"),
-        ("🎬 Criar vídeo", "16. Vídeo (Kling V1.5)"),
-        ("⌕ Resposta Rápida", "2. Pesquisa (Groq Llama 3.3)")
-    ]
-    for acao, mot in pilulas:
-        if st.button(acao): 
-            st.session_state.modelo_selecionado = mot
-            st.rerun()
+    st.markdown("<br><br><h3 style='color: #888; font-weight: 400;'>Olá, Leonardo</h3><h1 style='color: #fff; font-size: 32px;'>Deixe o Gerente trabalhar, ou escolha uma viação:</h1>", unsafe_allow_html=True)
 else:
     for m in mensagens:
         with st.chat_message(m["role"]):
@@ -201,8 +164,7 @@ else:
 st.markdown("<br>", unsafe_allow_html=True)
 t_col1, t_col2, t_space = st.columns([1.5, 1.5, 7])
 with t_col1:
-    with st.popover("＋"): 
-        img_upload = st.file_uploader("Mídia", type=["png", "jpg", "jpeg"])
+    with st.popover("＋"): img_upload = st.file_uploader("Mídia", type=["png", "jpg", "jpeg"])
 with t_col2:
     with st.popover("⚙"): 
         st.session_state.modelo_selecionado = st.radio("Motor:", list(MODEL_IDS.keys()), index=list(MODEL_IDS.keys()).index(st.session_state.modelo_selecionado))
@@ -212,18 +174,33 @@ if prompt := st.chat_input("Peça à Yukina..."):
     with st.chat_message("user"): st.markdown(prompt)
     
     with st.chat_message("assistant"):
-        # CHAVES
         or_key = os.getenv("OPENROUTER_API_KEY")
         gr_key = os.getenv("GROQ_API_KEY")
         ws_key = os.getenv("YUKINA_CORE")
         
-        modelo_id = MODEL_IDS[st.session_state.modelo_selecionado]
+        # --- AÇÃO DO GERENTE ---
+        motor_real = st.session_state.modelo_selecionado
+        
+        if "Automático" in motor_real:
+            with st.spinner("Hermes 2 Pro a analisar o seu pedido..."):
+                tag_decisao = analisar_intencao_gerente(prompt, or_key)
+                
+                if "[IMAGEM]" in tag_decisao: motor_real = "9. Imagem (Flux 2 Pro)"
+                elif "[VIDEO]" in tag_decisao: motor_real = "16. Vídeo (Kling V1.5)"
+                elif "[VISAO]" in tag_decisao or img_upload: motor_real = "7. Visão Omni (MiMo V2)"
+                elif "[CODIGO]" in tag_decisao: motor_real = "14. Estagiário (Qwen 32B Coder)"
+                elif "[PESQUISA]" in tag_decisao: motor_real = "2. Pesquisa (Groq Llama 3.3)"
+                else: motor_real = "5. Narrador Líder (Euryale)"
+                
+                st.toast(f"Gerente acionou: {motor_real}") # Aviso discreto no ecrã para si
+
+        modelo_id = MODEL_IDS[motor_real]
         
         # --- 🎬 VÍDEO (WAVESPEED) ---
-        if "Vídeo" in st.session_state.modelo_selecionado:
+        if "Vídeo" in motor_real:
             if not ws_key: st.error("Falta a chave YUKINA_CORE nas Secrets.")
             else:
-                with st.spinner("Yukina está renderizando o vídeo no estúdio Kling... (Pode demorar de 1 a 3 minutos)"):
+                with st.spinner("Kling V1.5 em renderização..."):
                     headers_ws = {"Authorization": f"Bearer {ws_key}", "Content-Type": "application/json"}
                     payload_ws = {"model": modelo_id, "prompt": prompt}
                     try:
@@ -238,17 +215,16 @@ if prompt := st.chat_input("Peça à Yukina..."):
                                     v_url = check.get('video_url') or check.get('output', {}).get('url')
                                     st.video(v_url)
                                     st.session_state.db[st.session_state.current_chat]["messages"].append({"role": "assistant", "content": "Vídeo materializado.", "video_url": v_url})
-                                    status = "completed"
-                                    break
-                            if status != "completed": st.error("Tempo de espera excedido na WaveSpeed.")
+                                    status = "completed"; break
+                            if status != "completed": st.error("Tempo limite excedido na WaveSpeed.")
                         else: st.error(f"Erro na WaveSpeed: {res.text}")
-                    except Exception as e: st.error(f"Erro de conexão: {e}")
+                    except Exception as e: st.error(f"Erro: {e}")
 
         # --- ⚡ TEXTO RÁPIDO (GROQ) ---
-        elif "Groq" in st.session_state.modelo_selecionado:
-            if not gr_key: st.error("Falta a chave GROQ_API_KEY nas Secrets.")
+        elif "Pesquisa" in motor_real:
+            if not gr_key: st.error("Falta a chave GROQ_API_KEY.")
             else:
-                with st.spinner("Processando..."):
+                with st.spinner("Groq a pesquisar rapidamente..."):
                     historico_limpo = [m for m in mensagens if "image_url" not in m and "video_url" not in m]
                     headers_gr = {"Authorization": f"Bearer {gr_key}", "Content-Type": "application/json"}
                     payload_gr = {"model": modelo_id, "messages": [{"role": "system", "content": SYSTEM_PROMPT}] + historico_limpo}
@@ -260,7 +236,7 @@ if prompt := st.chat_input("Peça à Yukina..."):
                     except Exception as e: st.error(f"Erro no Groq: {e}")
 
         # --- 🖼️ IMAGEM (OPENROUTER) ---
-        elif "Imagem" in st.session_state.modelo_selecionado:
+        elif "Imagem" in motor_real:
             with st.spinner("Desenhando no Flux 2 Pro..."):
                 headers_or = {"Authorization": f"Bearer {or_key}", "Content-Type": "application/json"}
                 payload_or = {"model": modelo_id, "messages": [{"role": "user", "content": prompt}], "modalities": ["image"]}
@@ -276,48 +252,39 @@ if prompt := st.chat_input("Peça à Yukina..."):
                     if url:
                         st.image(url)
                         st.session_state.db[st.session_state.current_chat]["messages"].append({"role": "assistant", "content": "Arte gerada.", "image_url": url})
-                    else: st.error("O modelo processou, mas não enviou a imagem.")
-                except Exception as e: st.error(f"Erro na geração: {e}")
+                    else: st.error("Falha ao receber a imagem.")
+                except Exception as e: st.error(f"Erro: {e}")
 
         # --- 👁️ VISÃO (OPENROUTER) ---
-        elif "Visão" in st.session_state.modelo_selecionado and img_upload:
-            with st.spinner("Analisando..."):
+        elif "Visão" in motor_real and img_upload:
+            with st.spinner("Analisando com MiMo..."):
                 headers_or = {"Authorization": f"Bearer {or_key}", "Content-Type": "application/json"}
                 img_b64 = base64.b64encode(img_upload.read()).decode()
-                payload_or = {
-                    "model": modelo_id,
-                    "messages": [{"role": "system", "content": SYSTEM_PROMPT}, {"role": "user", "content": [{"type": "text", "text": prompt}, {"type": "image_url", "image_url": {"url": f"data:image/png;base64,{img_b64}"}}]} ]
-                }
+                payload_or = {"model": modelo_id, "messages": [{"role": "system", "content": SYSTEM_PROMPT}, {"role": "user", "content": [{"type": "text", "text": prompt}, {"type": "image_url", "image_url": {"url": f"data:image/png;base64,{img_b64}"}}]} ]}
                 try:
                     res = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers_or, json=payload_or).json()
                     ans = res['choices'][0]['message']['content']
-                    st.markdown(ans)
-                    st.session_state.db[st.session_state.current_chat]["messages"].append({"role": "assistant", "content": ans})
+                    st.markdown(ans); st.session_state.db[st.session_state.current_chat]["messages"].append({"role": "assistant", "content": ans})
                 except Exception as e: st.error(f"Erro na visão: {e}")
 
-        # --- 💬 TEXTO E HISTÓRIAS STREAMING (OPENROUTER) ---
+        # --- 💬 TEXTO / CÓDIGO (STREAMING) ---
         else:
             headers_or = {"Authorization": f"Bearer {or_key}", "Content-Type": "application/json"}
             ph = st.empty(); full = ""
             historico_limpo = [m for m in mensagens if "image_url" not in m and "video_url" not in m]
-            payload_or = {
-                "model": modelo_id, 
-                "messages": [{"role": "system", "content": SYSTEM_PROMPT}] + historico_limpo, 
-                "stream": True
-            }
+            payload_or = {"model": modelo_id, "messages": [{"role": "system", "content": SYSTEM_PROMPT}] + historico_limpo, "stream": True}
             try:
                 resp = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers_or, json=payload_or, stream=True)
                 for l in resp.iter_lines():
                     if l:
                         try:
                             d = json.loads(l.decode().replace('data: ', ''))
-                            delta = d['choices'][0]['delta'].get('content', '')
-                            full += delta
+                            full += d['choices'][0]['delta'].get('content', '')
                             ph.markdown(full + "▌")
                         except: pass
                 ph.markdown(full)
                 st.session_state.db[st.session_state.current_chat]["messages"].append({"role": "assistant", "content": full})
-            except Exception as e: st.error(f"Erro no motor: {e}")
+            except Exception as e: st.error(f"Erro: {e}")
             
     save_db(st.session_state.db)