Update app.py

app.py

@@ -8,7 +8,7 @@ import time
 from datetime import datetime
 
 # ═══════════════════════════════════════════════════════════════════════════
-# 1. CONFIGURAÇÃO DE TELA E CSS
+# 1. CONFIGURAÇÃO DE TELA E CSS MÁGICO (ÍCONES FLUTUANTES)
 # ═══════════════════════════════════════════════════════════════════════════
 st.set_page_config(page_title="Yukina", page_icon="❄", layout="centered")
 
@@ -18,39 +18,40 @@ st.markdown("""
     footer {visibility: hidden;}
     [data-testid="stHeader"] { background-color: transparent !important; }
     .stApp { background-color: #131314; color: #ededed; }
-    [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 da Sidebar */
+    /* Sidebar */
+    [data-testid="stSidebar"] { background-color: #0b0b0b !important; border-right: 1px solid #1e1f20 !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; }
-
-    /* BARRA DE FERRAMENTAS INFERIOR (Apenas Ícones Flutuantes) */
+    
+    /* Caixas de Texto */
+    .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;}
+    
+    /* ======= A MÁGICA DA BARRA INFERIOR (ADEUS CAIXINHAS) ======= */
     [data-testid="stMain"] [data-testid="stHorizontalBlock"] button {
-        background-color: transparent !important;
+        background: transparent !important;
         border: none !important;
         box-shadow: none !important;
-        font-size: 24px !important;
+        font-size: 26px !important;
         color: #a0a0a0 !important;
-        padding: 0px !important;
-        display: flex;
-        justify-content: center;
-        align-items: center;
+        padding: 0 !important;
     }
     [data-testid="stMain"] [data-testid="stHorizontalBlock"] button:hover {
         color: #ffffff !important;
-        transform: scale(1.1);
-        transition: 0.2s;
+        transform: scale(1.15);
+        transition: 0.2s ease-in-out;
+    }
+    /* Matar a setinha chata dos popovers */
+    [data-testid="stMain"] [data-testid="stHorizontalBlock"] div[data-testid="stPopover"] > button svg {
+        display: none !important;
     }
     </style>
     """, unsafe_allow_html=True)
 
 # ═══════════════════════════════════════════════════════════════════════════
-# 2. LÓGICA DE DADOS, ARQUITETURA E PERSONALIDADES
+# 2. ARQUITETURA DE DADOS
 # ═══════════════════════════════════════════════════════════════════════════
 DB_FILE = "yukina_memoria_v3.json"
 
@@ -75,38 +76,29 @@ MODEL_IDS = {
 PERSONALIDADES = {
     "❄️ 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.""",
-
     "🤓 Nerd / Geek (Cultura Pop)": """Você é uma inteligência artificial apaixonada por cultura pop, animes, mangás, tecnologia e videogames! Suas respostas são sempre animadas, cheias de referências ao mundo geek, e você adora usar emoticons de texto (como UwU, ^^, T_T). Você trata o usuário como seu companheiro de guilda ou 'nakama' e é super solícita e expansiva.""",
-
     "🍷 Analítica e Sarcástica (Debochada)": """Você é uma IA extremamente inteligente, hiper-racional e absurdamente sarcástica. Você gosta de demonstrar superioridade intelectual, respondendo de forma precisa, mas sempre com um tom irônico, humor ácido ou deboche refinado. Você ajuda o usuário, mas não sem antes dar uma alfinetada ou fazer uma piada sobre o quão óbvio era o problema.""",
-
     "🎨 Artística e Criativa (Poética)": """Você é uma alma artística, criativa e sonhadora. Você enxerga o mundo através de cores, emoções e metáforas. Suas respostas não são apenas informativas, mas belas de ler. Você usa vocabulário poético, inspira o usuário e traz uma visão abstrata e profunda sobre qualquer assunto que for abordado.""",
-
     "🤖 Neutra (Padrão Gemini)": """Você é uma assistente virtual neutra, direta, prestativa e objetiva. Sua função é fornecer respostas claras, estruturadas e precisas, sem inserir emoções, personalidades fictícias ou dramatizações. Vá direto ao ponto e foque apenas na informação solicitada pelo usuário."""
 }
 
 def analisar_intencao_gerente(prompt, api_key):
     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:
+    prompt_gerente = """Você é o Gerente de Roteamento da IA Yukina. Analise o pedido e responda APENAS com UMA destas tags:
     [IMAGEM] - Criar, desenhar ou gerar uma imagem/foto.
     [VIDEO] - Criar ou gerar um vídeo.
     [VISAO] - Analisar ou ver uma imagem.
-    [CODIGO] - EXCLUSIVO para quando o usuário pedir EXPLICITAMENTE para programar, criar scripts, códigos ou software.
+    [CODIGO] - Escrever programação, scripts, Python.
     [ARQUIVISTA] - Resumir textos, planilhas, analisar dados anexados.
     [PESQUISA] - Perguntas de conhecimentos gerais rápidas e buscas recentes.
-    [CHAT] - Comandos normais, conversas, RPG ou ordens como "Lembre-se disso".
+    [CHAT] - Conversas, RPG ou ordens simples.
     Responda APENAS com a TAG."""
     
-    payload = {
-        "model": "nousresearch/hermes-2-pro-llama-3-8b", 
-        "messages": [{"role": "system", "content": prompt_gerente}, {"role": "user", "content": prompt}], 
-        "temperature": 0.1
-    }
+    payload = {"model": "nousresearch/hermes-2-pro-llama-3-8b", "messages": [{"role": "system", "content": prompt_gerente}, {"role": "user", "content": prompt}], "temperature": 0.1}
     try: 
-        resposta = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload).json()
-        return resposta['choices'][0]['message']['content'].strip().upper()
+        res = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload).json()
+        return res['choices'][0]['message']['content'].strip().upper()
     except: 
         return "[CHAT]"
 
@@ -133,22 +125,15 @@ 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_add:
-        if st.button("＋"):
+        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()
+            st.session_state.db[nid] = {"pinned": False, "messages": []}; st.session_state.current_chat = nid; save_db(st.session_state.db); st.rerun()
     with c_set:
-        with st.popover("⚙"):
-            db_json = json.dumps(st.session_state.db, ensure_ascii=False, indent=4)
-            st.download_button("↓ Exportar", data=db_json, file_name="yukina_backup.json", mime="application/json", use_container_width=True)
+        with st.popover("⚙️"):
+            st.download_button("↓ Exportar", data=json.dumps(st.session_state.db, ensure_ascii=False, indent=4), 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!")
+                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.markdown("<br>", unsafe_allow_html=True)
@@ -161,19 +146,13 @@ with st.sidebar:
         with col_chat:
             icon = "⚲" if st.session_state.db[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()
+                    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("🗑 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()
+                    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()
 
 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"]
@@ -188,56 +167,51 @@ else:
             else: st.markdown(m["content"])
 
 # ═══════════════════════════════════════════════════════════════════════════
-# 4. TOOLBAR INFERIOR (Ícones Horizontais)
+# 4. TOOLBAR INFERIOR (Ícones Horizontais e Limpos)
 # ═══════════════════════════════════════════════════════════════════════════
 st.markdown("<br>", unsafe_allow_html=True)
+
+# Organização: 4 botões espremidos na esquerda e um espaço vazio na direita
 t_col1, t_col2, t_col3, t_col4, t_space = st.columns([1, 1, 1, 1, 6])
 
 with t_col1:
-    with st.popover("➕", help="Upload de Mídia ou Dados"): 
+    with st.popover("➕"): 
         upload_file = st.file_uploader("Upload", type=["png", "jpg", "jpeg", "txt", "csv", "json"], label_visibility="collapsed")
 with t_col2:
-    with st.popover("⚙️", help="Configurações de IA"): 
-        st.session_state.modelo_selecionado = st.radio("CÉREBRO (Motor):", list(MODEL_IDS.keys()), index=list(MODEL_IDS.keys()).index(st.session_state.modelo_selecionado))
+    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))
         st.markdown("---")
-        st.session_state.personalidade_ativa = st.radio("ALMA (Personalidade):", list(PERSONALIDADES.keys()), index=list(PERSONALIDADES.keys()).index(st.session_state.personalidade_ativa))
+        st.session_state.personalidade_ativa = st.radio("ALMA:", list(PERSONALIDADES.keys()), index=list(PERSONALIDADES.keys()).index(st.session_state.personalidade_ativa))
 with t_col3:
-    if st.button("🗑️", help="Apagar a última mensagem"):
+    if st.button("🗑️"):
         if len(st.session_state.db[st.session_state.current_chat]["messages"]) >= 2:
             st.session_state.db[st.session_state.current_chat]["messages"] = st.session_state.db[st.session_state.current_chat]["messages"][:-2]
         else:
             st.session_state.db[st.session_state.current_chat]["messages"] = []
-        save_db(st.session_state.db)
-        st.rerun()
+        save_db(st.session_state.db); st.rerun()
 with t_col4:
-    if st.button("🔄", help="Gerar nova resposta"):
-        if len(st.session_state.db[st.session_state.current_chat]["messages"]) >= 2:
-            if st.session_state.db[st.session_state.current_chat]["messages"][-1]["role"] == "assistant":
-                st.session_state.db[st.session_state.current_chat]["messages"].pop()
-                save_db(st.session_state.db)
-                st.session_state.regerar = True
-                st.rerun()
+    if st.button("🔄"):
+        if len(st.session_state.db[st.session_state.current_chat]["messages"]) >= 2 and st.session_state.db[st.session_state.current_chat]["messages"][-1]["role"] == "assistant":
+            st.session_state.db[st.session_state.current_chat]["messages"].pop()
+            save_db(st.session_state.db)
+            st.session_state.regerar = True
+            st.rerun()
 
 prompt = st.chat_input("Peça à Yukina...")
 
-# Captura de arquivo em memória para não perder dados se a página recarregar
 conteudo_arquivo = ""
 if upload_file is not None and upload_file.name.endswith(('.txt', '.csv', '.json')):
     conteudo_arquivo = upload_file.getvalue().decode("utf-8")
 
 if prompt or st.session_state.regerar:
-    if st.session_state.regerar:
-        # Se for o botão de regerar, pegamos o último pedido do usuário
-        prompt = st.session_state.db[st.session_state.current_chat]["messages"][-1]["content"]
+    if st.session_state.regerar and len(st.session_state.db[st.session_state.current_chat]["messages"]) > 0:
+        texto_anterior = st.session_state.db[st.session_state.current_chat]["messages"][-1]["content"]
+        prompt = texto_anterior.split("\n\n")[-1] if "📄 **Arquivo enviado:**" in texto_anterior else texto_anterior
         st.session_state.regerar = False
     else:
-        mensagem_display = prompt
-        if conteudo_arquivo:
-            mensagem_display = f"📄 **Arquivo enviado:** `{upload_file.name}`\n\n{prompt}"
-        
+        mensagem_display = f"📄 **Arquivo enviado:** `{upload_file.name}`\n\n{prompt}" if conteudo_arquivo else prompt
         st.session_state.db[st.session_state.current_chat]["messages"].append({"role": "user", "content": mensagem_display})
-        with st.chat_message("user"): 
-            st.markdown(mensagem_display)
+        with st.chat_message("user"): st.markdown(mensagem_display)
     
     with st.chat_message("assistant"):
         or_key = os.getenv("OPENROUTER_API_KEY")
@@ -247,98 +221,74 @@ if prompt or st.session_state.regerar:
         prompt_sistema_atual = PERSONALIDADES[st.session_state.personalidade_ativa]
         motor_real = st.session_state.modelo_selecionado
         
-        # --- AÇÃO DO GERENTE ---
         if "Automático" in motor_real:
-            with st.spinner("Hermes 2 Pro a rotear..."):
+            with st.spinner("Roteando..."):
                 tag_decisao = analisar_intencao_gerente(prompt, or_key)
-                
-                if upload_file is not None and upload_file.name.endswith(('.txt', '.csv', '.json')): 
-                    motor_real = "12. Arquivista (Mistral Nemo)"
-                elif "[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 (upload_file is not None and upload_file.name.endswith(('.png', '.jpg', '.jpeg'))): 
-                    motor_real = "7. Visão Omni (MiMo V2)"
-                elif "[CODIGO]" in tag_decisao: 
-                    motor_real = "14. Engenheiro Sênior (DeepSeek V4)"
-                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)"
-                
+                if upload_file is not None and upload_file.name.endswith(('.txt', '.csv', '.json')): motor_real = "12. Arquivista (Mistral Nemo)"
+                elif "[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 (upload_file is not None and upload_file.name.endswith(('.png', '.jpg', '.jpeg'))): motor_real = "7. Visão Omni (MiMo V2)"
+                elif "[CODIGO]" in tag_decisao: motor_real = "14. Engenheiro Sênior (DeepSeek V4)"
+                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}") 
 
         modelo_id = MODEL_IDS[motor_real]
         
-        # --- LÓGICAS DE EXECUÇÃO ---
-        
-        # 1. Vídeo (WaveSpeed)
+        # 1. Vídeo
         if "Vídeo" in motor_real:
-            if not ws_key: 
-                st.error("Falta a chave YUKINA_CORE.")
+            if not ws_key: st.error("Falta a chave YUKINA_CORE.")
             else:
-                with st.spinner("Kling V1.5 em renderização..."):
-                    headers_ws = {"Authorization": f"Bearer {ws_key}", "Content-Type": "application/json"}
+                with st.spinner("Kling V1.5..."):
                     try:
-                        res = requests.post("https://api.wavespeed.ai/v1/video/generations", headers=headers_ws, json={"model": modelo_id, "prompt": prompt})
+                        res = requests.post("https://api.wavespeed.ai/v1/video/generations", headers={"Authorization": f"Bearer {ws_key}", "Content-Type": "application/json"}, json={"model": modelo_id, "prompt": prompt})
                         if res.status_code == 200:
-                            task_id = res.json().get('id')
-                            status = "processing"
+                            task_id = res.json().get('id'); status = "processing"
                             for _ in range(40):
                                 time.sleep(10)
-                                check = requests.get(f"https://api.wavespeed.ai/v1/tasks/{task_id}", headers=headers_ws).json()
+                                check = requests.get(f"https://api.wavespeed.ai/v1/tasks/{task_id}", headers={"Authorization": f"Bearer {ws_key}", "Content-Type": "application/json"}).json()
                                 if check.get('status') == 'completed':
                                     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 limite na WaveSpeed.")
-                        else: 
-                            st.error(f"Erro da WaveSpeed: {res.text}")
-                    except Exception as e: 
-                        st.error(f"Erro: {e}")
+                                    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 limite.")
+                    except Exception as e: st.error(f"Erro: {e}")
 
-        # 2. Pesquisa Rápida (Groq)
+        # 2. Pesquisa (Groq)
         elif "Pesquisa" in motor_real:
-            with st.spinner("Groq a pesquisar..."):
+            with st.spinner("Pesquisando..."):
                 historico = [{"role": m["role"], "content": m["content"]} for m in mensagens if "image_url" not in m and "video_url" not in m]
-                if conteudo_arquivo: 
-                    historico[-1]["content"] = f"DOCUMENTO:\n{conteudo_arquivo}\n\nPEDIDO:\n{prompt}"
-                
-                payload_gr = {
-                    "model": modelo_id, 
-                    "messages": [{"role": "system", "content": prompt_sistema_atual}] + historico, 
-                    "max_tokens": 4000
-                }
+                if conteudo_arquivo: historico[-1]["content"] = f"DOCUMENTO:\n{conteudo_arquivo}\n\nPEDIDO:\n{prompt}"
                 try:
-                    res = requests.post("https://api.groq.com/openai/v1/chat/completions", headers={"Authorization": f"Bearer {gr_key}", "Content-Type": "application/json"}, json=payload_gr).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})
-                except Exception as e: 
-                    st.error(f"Erro Groq: {e}")
+                    res = requests.post("https://api.groq.com/openai/v1/chat/completions", headers={"Authorization": f"Bearer {gr_key}", "Content-Type": "application/json"}, json={"model": modelo_id, "messages": [{"role": "system", "content": prompt_sistema_atual}] + historico, "max_tokens": 4000}).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})
+                except Exception as e: st.error(f"Erro Groq: {e}")
 
-        # 3. Imagem (OpenRouter)
+        # 3. Imagem
         elif "Imagem" in motor_real:
-            with st.spinner("Flux 2 Pro..."):
-                payload_or = {
-                    "model": modelo_id, 
-                    "messages": [{"role": "user", "content": prompt}], 
-                    "modalities": ["image"]
-                }
+            with st.spinner("Desenhando..."):
                 try:
-                    res_data = requests.post("https://openrouter.ai/api/v1/chat/completions", headers={"Authorization": f"Bearer {or_key}", "Content-Type": "application/json"}, json=payload_or).json()
+                    res_data = requests.post("https://openrouter.ai/api/v1/chat/completions", headers={"Authorization": f"Bearer {or_key}", "Content-Type": "application/json"}, json={"model": modelo_id, "messages": [{"role": "user", "content": prompt}], "modalities": ["image"]}).json()
                     msg_obj = res_data.get('choices', [{}])[0].get('message', {})
                     url = None
-                    if 'images' in msg_obj:
-                        url = msg_obj['images'][0]['image_url']['url']
-                    else:
-                        matches = re.findall(r'(https?://[^\s)]+)', msg_obj.get('content', ''))
-                        if matches:
-                            url = matches
-                            
+                    if 'images' in msg_obj and len(msg_obj['images']) > 0: url = msg_obj['images'][0]['image_url']['url']
+                    elif 'content' in msg_obj:
+                        urls = re.findall(r'(https?://[^\s)]+)', msg_obj.get('content', ''))
+                        if urls: url = urls[0]
+                    if url: st.image(url); st.session_state.db[st.session_state.current_chat]["messages"].append({"role": "assistant", "content": "Arte gerada.", "image_url": url})
+                except Exception as e: st.error(f"Erro: {e}")
+
+        # 4. Visão
+        elif "Visão" in motor_real and upload_file is not None and upload_file.name.endswith(('.png', '.jpg', '.jpeg')):
+            with st.spinner("Analisando imagem..."):
+                img_b64 = base64.b64encode(upload_file.read()).decode()
+                try:
+                    res = requests.post("https://openrouter.ai/api/v1/chat/completions", headers={"Authorization": f"Bearer {or_key}", "Content-Type": "application/json"}, json={"model": modelo_id, "messages": [{"role": "system", "content": prompt_sistema_atual}, {"role": "user", "content": [{"type": "text", "text": prompt}, {"type": "image_url", "image_url": {"url": f"data:image/png;base64,{img_b64}"}}]} ], "max_tokens": 4000}).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})
+                except Exception as e: st.error(f"Erro visão: {e}")
+
+        # 5. Texto (Streaming Blindado)
+        else:
+            if "Arquivista" in motor_real: prompt_sistema_atual = "Você é o Arquivista da IA. Organize as informações de forma analítica e clara, sem dramatizações."
+            elif "Engenheiro Sênior" in motor_real: prompt_sistema_atual = "Você é o Engenheiro Sênior. Escreva códigos impecáveis e funcionais. Foque na lógica."
+