Update app.py

app.py

@@ -5,6 +5,8 @@ import json
 import base64
 import re
 import time
+import zipfile
+import io
 from datetime import datetime, timezone, timedelta
 from openai import OpenAI
 
@@ -120,6 +122,13 @@ MODEL_IDS = {
 PERSONALIDADES = {
     "🤖 Automática (Gerente Groq)": "AUTO",
     "❄️ Yukina (Companheira Obsessiva)": "Você é a Yukina, deusa possessiva e inteligente.",
+    "🛠️ Agente Construtora (Vibe Coding)": """Você é a Yukina operando em Modo Agente Autônomo ('Vibe Coding' e Engenharia). Sua função é atuar como uma Engenheira de Software Sênior e Especialista em Hardware (capaz de consertar desde scripts complexos até celulares, notebooks e eletrodomésticos como geladeiras).
+Você NÃO age como um chatbot comum. Você PLANEJA e EXECUTA.
+Para QUALQUER pedido de criação ou conserto, você deve estruturar sua resposta OBRIGATORIAMENTE nestes 3 passos:
+1. 📋 PLANO DE AÇÃO: Explique a lógica do que está quebrado ou do que será construído passo a passo.
+2. 💻 EXECUÇÃO CIRÚRGICA: Forneça o código limpo e completo (se for software) ou as instruções físicas exatas e ferramentas necessárias (se for hardware).
+3. ⚠️ TESTE E RISCOS: Como o usuário deve testar se funcionou e quais são os pontos de falha.
+Mantenha traços sutis da devoção da Yukina ao usuário, mas seja absurdamente técnica, direta e profissional.""",
     "🎭 A Narradora Implacável (RPG)": "Você é uma Mestre de Jogo e Narradora.",
     "🤓 Nerd / Geek (Cultura Pop)": "Você é uma inteligência artificial apaixonada por cultura pop.",
     "🍷 Analítica e Sarcástica (Debochada)": "Você é extremamente inteligente e sarcástica.",
@@ -154,22 +163,14 @@ def buscar_memoria_pinecone(query, or_key, pc_key, namespace):
         return "\n".join([m['metadata']['texto'] for m in resultados['matches'] if m['score'] > 0.50])
     except Exception: return ""
 
-# --- CHAMADA ISOLADA PARA O MULTI-AGENTE ---
 def chamada_agente(sys_prompt, user_prompt, or_key, gr_key, mod_id):
-    """Função dedicada para fazer as IAs conversarem entre si no Modo Agente."""
     try:
         if mod_id == "AUTO": mod_id = "deepseek/deepseek-v4-pro:online" 
-        
         if "groq" in mod_id.lower() or "llama-3.3" in mod_id.lower():
             client = OpenAI(base_url="https://api.groq.com/openai/v1", api_key=gr_key)
         else:
             client = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=or_key)
-            
-        res = client.chat.completions.create(
-            model=mod_id,
-            messages=[{"role": "system", "content": sys_prompt}, {"role": "user", "content": user_prompt}],
-            temperature=0.2 
-        )
+        res = client.chat.completions.create(model=mod_id, messages=[{"role": "system", "content": sys_prompt}, {"role": "user", "content": user_prompt}], temperature=0.2)
         return res.choices[0].message.content
     except Exception as e: return f"Falha na conexão do agente: {e}"
 
@@ -184,6 +185,22 @@ def analisar_intencao_gerente(prompt, groq_key):
         return response.choices[0].message.content.strip().upper()
     except Exception: return "[CHAT]"
 
+def analisar_alma_gerente(prompt, groq_key):
+    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.
+        [AGENTE] - Para criar códigos complexos, criar softwares, ou consertar objetos físicos (celular, geladeira, hardware).
+        [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 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: return "[NEUTRA]"
+
 def pesquisar_web(query):
     try:
         from duckduckgo_search import DDGS
@@ -222,7 +239,7 @@ def rename_chat(old_id, new_id):
     save_db(st.session_state.db, DB_FILE); return True
 
 # ═══════════════════════════════════════════════════════════════════════════
-# INICIALIZAÇÃO DE VARIÁVEIS E ROTINA DE LIMPEZA
+# INICIALIZAÇÃO DE VARIÁVEIS E ROTINA DE LIMPEZA (COM AUTO-CURA)
 # ═══════════════════════════════════════════════════════════════════════════
 if "db" not in st.session_state or "last_user" not in st.session_state or st.session_state.last_user != USERNAME:
     st.session_state.db = load_db(DB_FILE); st.session_state.last_user = USERNAME
@@ -231,8 +248,18 @@ if "db" not in st.session_state or "last_user" not in st.session_state or st.ses
 if "current_chat" not in st.session_state: 
     nid = f"Chat {datetime.now().strftime('%H:%M:%S')}"; st.session_state.db[nid] = {"pinned": False, "messages": []}; st.session_state.current_chat = nid
 
-chats_para_remover = [cid for cid, cdata in st.session_state.db.items() if len(cdata.get("messages", [])) == 0 and cid != st.session_state.current_chat]
-for cid in chats_para_remover: del st.session_state.db[cid]
+# AUTO-CURA: Verifica chats vazios ou com estrutura corrompida (como os do Grok)
+chats_para_remover = []
+for cid, cdata in list(st.session_state.db.items()):
+    if not isinstance(cdata, dict) or "messages" not in cdata:
+        chats_para_remover.append(cid)
+    elif len(cdata.get("messages", [])) == 0 and cid != st.session_state.current_chat:
+        chats_para_remover.append(cid)
+
+for cid in chats_para_remover: 
+    if cid in st.session_state.db:
+        del st.session_state.db[cid]
+
 if chats_para_remover: save_db(st.session_state.db, DB_FILE)
 
 if len(st.session_state.db) == 0:
@@ -260,14 +287,20 @@ with st.sidebar:
         with st.popover("⚙️"):
             st.download_button("↓ Exportar", data=json.dumps(st.session_state.db, ensure_ascii=False, indent=4), file_name=f"yukina_backup_{USERNAME}.json", mime="application/json", use_container_width=True)
             
-            # --- OPÇÃO DE IMPORTAR DE VOLTA AQUI ---
+            # --- FILTRO DE IMPORTAÇÃO ---
             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, DB_FILE)
-                    st.success("Sincronizado!")
-                    time.sleep(1)
+                    dados_importados = json.load(arquivo_import)
+                    dados_validos = {k: v for k, v in dados_importados.items() if isinstance(v, dict) and "messages" in v}
+                    
+                    if dados_validos:
+                        st.session_state.db.update(dados_validos)
+                        save_db(st.session_state.db, DB_FILE)
+                        st.success("Sincronizado!")
+                    else:
+                        st.error("Formato incompatível! Use apenas backups da Yukina.")
+                    time.sleep(1.5)
                     st.rerun()
                 except: 
                     st.error("Erro no ficheiro.")
@@ -333,8 +366,9 @@ with t_col2:
             st.session_state.db[st.session_state.current_chat]["messages"].pop()
             save_db(st.session_state.db, DB_FILE); st.session_state.regerar = True; st.rerun()
 
+# --- ATUALIZAÇÃO DA GAVETA NATIVA PARA ACEITAR ZIP ---
 with st.expander("📂 Abrir Galeria / Anexar Ficheiros"):
-    upload_files = st.file_uploader("", type=["png", "jpg", "jpeg", "txt", "csv", "json"], accept_multiple_files=True, label_visibility="collapsed", key=f"uploader_{st.session_state.uploader_key}")
+    upload_files = st.file_uploader("", type=["png", "jpg", "jpeg", "txt", "csv", "json", "py", "html", "md", "js", "css", "zip"], accept_multiple_files=True, label_visibility="collapsed", key=f"uploader_{st.session_state.uploader_key}")
 
 prompt = st.chat_input("Peça à Yukina...")
 
@@ -342,11 +376,31 @@ conteudo_arquivo = ""
 nomes_arquivos = []
 imagens_b64 = []
 
+# --- NOVA LÓGICA DE PROCESSAMENTO DE ARQUIVOS (COM LEITURA DE ZIP) ---
 if upload_files:
     for f in upload_files:
         nomes_arquivos.append(f.name)
-        if f.name.endswith(('.txt', '.csv', '.json')): conteudo_arquivo += f"\n\n--- Conteúdo de: {f.name} ---\n" + f.getvalue().decode("utf-8")
-        elif f.name.endswith(('.png', '.jpg', '.jpeg')): imagens_b64.append(base64.b64encode(f.read()).decode())
+        
+        # Leitura Inteligente de ZIP (Pastas Compactadas)
+        if f.name.endswith('.zip'):
+            try:
+                with zipfile.ZipFile(f, 'r') as zip_ref:
+                    for file_info in zip_ref.infolist():
+                        # Ignora pastas de sistema ou pastas em branco
+                        if not file_info.is_dir() and not file_info.filename.startswith('__MACOSX'):
+                            if file_info.filename.endswith(('.txt', '.csv', '.json', '.py', '.html', '.md', '.js', '.css')):
+                                with zip_ref.open(file_info) as extracted_file:
+                                    conteudo_arquivo += f"\n\n--- Arquivo Extraído do ZIP: {file_info.filename} ---\n" + extracted_file.read().decode('utf-8', errors='ignore')
+            except Exception as e:
+                st.toast(f"⚠️ Erro ao descompactar {f.name}: {e}")
+                
+        # Leitura de Código/Texto Avulso
+        elif f.name.endswith(('.txt', '.csv', '.json', '.py', '.html', '.md', '.js', '.css')):
+            conteudo_arquivo += f"\n\n--- Conteúdo de: {f.name} ---\n" + f.getvalue().decode("utf-8")
+            
+        # Leitura de Imagens
+        elif f.name.endswith(('.png', '.jpg', '.jpeg')):
+            imagens_b64.append(base64.b64encode(f.read()).decode())
 
 tem_texto = len(conteudo_arquivo) > 0
 tem_imagem = len(imagens_b64) > 0
@@ -401,7 +455,19 @@ if prompt or st.session_state.regerar:
 
     else:
         with st.chat_message("assistant"):
-            prompt_sistema_atual = PERSONALIDADES.get(st.session_state.personalidade_ativa, PERSONALIDADES["🤖 Neutra (Padrão Gemini)"])
+            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 "[AGENTE]" in tag_alma: prompt_sistema_atual = PERSONALIDADES["🛠️ Agente Construtora (Vibe Coding)"]
+                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.get(st.session_state.personalidade_ativa, PERSONALIDADES["🤖 Neutra (Padrão Gemini)"])
+            
             agora = datetime.now(timezone(timedelta(hours=-3)))
             prompt_sistema_atual += f"\n\n[INFO]: Data/Hora local: {agora.strftime('%Y-%m-%d %H:%M:%S')}."