Update app.py

app.py

@@ -6,6 +6,7 @@ import base64
 import re
 import time
 from datetime import datetime
+from openai import OpenAI
 
 try:
     from huggingface_hub import HfApi, hf_hub_download
@@ -47,15 +48,13 @@ st.markdown("""
         transform: scale(1.15);
         transition: 0.2s ease-in-out;
     }
-    [data-testid="stMain"] [data-testid="stHorizontalBlock"] div[data-testid="stPopover"] > button svg {
-        display: none !important;
-    }
     
     /* Estilização elegante para a Gaveta Nativa de Upload */
     [data-testid="stExpander"] {
         background-color: #1e1f20 !important;
         border: 1px solid #3c4043 !important;
         border-radius: 15px !important;
+        margin-bottom: 10px;
     }
     [data-testid="stExpander"] summary {
         color: #e3e3e3 !important;
@@ -70,7 +69,7 @@ DB_FILE = "yukina_memoria_v3.json"
 DATASET_ID = "Astarok/Yukina_Memoria"
 
 MODEL_IDS = {
-    "🤖 Automático (Gerente Hermes)": "AUTO",
+    "🤖 Automático (Gerente Groq)": "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",
@@ -97,23 +96,32 @@ Personalidade pública: orgulhosa, dramática e otimista. Adora exigir aplausos.
     "🤖 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 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] - Escrever programação, scripts, Python.
-    [ARQUIVISTA] - Resumir textos, planilhas, analisar dados anexados.
-    [PESQUISA] - Perguntas de conhecimentos gerais rápidas e buscas recentes.
-    [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}
-    try: 
-        res = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=payload).json()
-        return res['choices'][0]['message']['content'].strip().upper()
-    except: 
+def analisar_intencao_gerente(prompt, groq_key):
+    """Utiliza a Groq para roteamento ultra-rápido da intenção."""
+    try:
+        client = OpenAI(base_url="https://api.groq.com/openai/v1", api_key=groq_key)
+        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] - Escrever programação, scripts, Python.
+        [ARQUIVISTA] - Resumir textos, planilhas, analisar dados anexados.
+        [PESQUISA] - Perguntas de conhecimentos gerais rápidas e buscas recentes.
+        [CHAT] - Conversas, RPG ou ordens simples.
+        Responda APENAS com a TAG."""
+        
+        response = client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=[
+                {"role": "system", "content": prompt_gerente},
+                {"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 do Gerente: {e}")
         return "[CHAT]"
 
 def load_db():
@@ -124,28 +132,33 @@ def load_db():
             api.create_repo(repo_id=DATASET_ID, repo_type="dataset", private=True, exist_ok=True)
             path = hf_hub_download(repo_id=DATASET_ID, filename=DB_FILE, repo_type="dataset", token=hf_token)
             with open(path, "r", encoding="utf-8") as f: return json.load(f)
-        except Exception: pass
+        except Exception as e: 
+            print(f"Aviso HuggingFace Load: {e}")
     if os.path.exists(DB_FILE):
         try:
             with open(DB_FILE, "r", encoding="utf-8") as f: return json.load(f)
-        except: pass
+        except Exception as e: 
+            print(f"Erro ao ler DB local: {e}")
     return {"Nova Conversa": {"pinned": False, "messages": []}}
 
 def save_db(db):
-    with open(DB_FILE, "w", encoding="utf-8") as f: json.dump(db, f, ensure_ascii=False, indent=4)
-    hf_token = os.getenv("HF_TOKEN")
-    if hf_token and HF_HUB_AVAILABLE:
-        try:
+    try:
+        with open(DB_FILE, "w", encoding="utf-8") as f: json.dump(db, f, ensure_ascii=False, indent=4)
+        hf_token = os.getenv("HF_TOKEN")
+        if hf_token and HF_HUB_AVAILABLE:
             api = HfApi(token=hf_token)
             api.create_repo(repo_id=DATASET_ID, repo_type="dataset", private=True, exist_ok=True)
             api.upload_file(path_or_fileobj=DB_FILE, path_in_repo=DB_FILE, repo_id=DATASET_ID, repo_type="dataset")
-        except Exception: pass
+    except Exception as e:
+        print(f"Erro ao salvar DB: {e}")
 
+# Inicialização de Variáveis de Estado
 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 = "🤖 Automático (Gerente Hermes)"
+if "modelo_selecionado" not in st.session_state: st.session_state.modelo_selecionado = "🤖 Automático (Gerente Groq)"
 if "personalidade_ativa" not in st.session_state: st.session_state.personalidade_ativa = "❄️ Yukina (Companheira Obsessiva)"
 if "regerar" not in st.session_state: st.session_state.regerar = False
+if "uploader_key" not in st.session_state: st.session_state.uploader_key = 0
 
 # ═══════════════════════════════════════════════════════════════════════════
 # 3. SIDEBAR E INTERFACE PRINCIPAL
@@ -154,7 +167,7 @@ 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("➕", help="Nova conversa"):
             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()
     with c_set:
@@ -183,11 +196,17 @@ 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))
+    st.session_state.personalidade_ativa = st.selectbox("Alma:", list(PERSONALIDADES.keys()), index=list(PERSONALIDADES.keys()).index(st.session_state.personalidade_ativa))
+
 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;'>Como você quer que eu aja hoje?</h1>", unsafe_allow_html=True)
+    st.markdown("<br><br><h3 style='color: #888; font-weight: 400;'>Olá!</h3><h1 style='color: #fff; font-size: 32px;'>Como você quer que eu aja hoje?</h1>", unsafe_allow_html=True)
 else:
     for m in mensagens:
         with st.chat_message(m["role"]):
@@ -196,35 +215,31 @@ else:
             else: st.markdown(m["content"])
 
 # ═══════════════════════════════════════════════════════════════════════════
-# 4. TOOLBAR INFERIOR (Gaveta Nativa Blindada)
+# 4. TOOLBAR INFERIOR
 # ═══════════════════════════════════════════════════════════════════════════
 st.markdown("<br>", unsafe_allow_html=True)
 
-t_col1, t_col2, t_col3, t_space = st.columns([1, 1, 1, 7])
+# Barra simplificada apenas com utilitários da conversa atual
+t_col1, t_col2, t_space = st.columns([1, 1, 8])
 
 with t_col1:
-    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:", list(PERSONALIDADES.keys()), index=list(PERSONALIDADES.keys()).index(st.session_state.personalidade_ativa))
-with t_col2:
-    if st.button("🗑️"):
+    if st.button("🗑️", help="Apagar última mensagem"):
         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()
-with t_col3:
-    if st.button("🔄"):
+with t_col2:
+    if st.button("🔄", help="Regerar resposta"):
         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()
 
-# GAVETA NATIVA: Não sofre "crash" ao ir para a galeria em segundo plano
+# GAVETA NATIVA: Chave dinâmica para esvaziar arquivos após o envio
 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")
+    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}")
 
 prompt = st.chat_input("Peça à Yukina...")
 
@@ -268,7 +283,7 @@ if prompt or st.session_state.regerar:
         
         if "Automático" in motor_real:
             with st.spinner("Roteando..."):
-                tag_decisao = analisar_intencao_gerente(prompt, or_key)
+                tag_decisao = analisar_intencao_gerente(prompt, gr_key)
                 if tem_texto: 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)"
@@ -297,19 +312,26 @@ if prompt or st.session_state.regerar:
                                     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.")
-                    except Exception as e: st.error(f"Erro: {e}")
+                    except Exception as e: st.error(f"Erro no Vídeo: {e}")
 
-        # 2. Pesquisa (Groq)
+        # 2. Pesquisa (Groq com biblioteca oficial)
         elif "Pesquisa" in motor_real:
             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 tem_texto: historico[-1]["content"] = f"DOCUMENTO(S) ANEXADOS:\n{conteudo_arquivo}\n\nPEDIDO DO USUÁRIO:\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={"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})
+                    client_groq = OpenAI(base_url="https://api.groq.com/openai/v1", api_key=gr_key)
+                    res = client_groq.chat.completions.create(
+                        model=modelo_id,
+                        messages=[{"role": "system", "content": prompt_sistema_atual}] + historico,
+                        max_tokens=4000
+                    )
+                    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 (Flux)
+        # 3. Imagem (Flux) - Mantido via requests para compatibilidade com "modalities" específicas do OpenRouter
         elif "Imagem" in motor_real:
             with st.spinner("Desenhando..."):
                 try:
@@ -320,21 +342,33 @@ if prompt or st.session_state.regerar:
                     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}")
+                    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 na Imagem: {e}")
 
-        # 4. Visão (Análise de Múltiplas Imagens)
+        # 4. Visão (Análise de Múltiplas Imagens via biblioteca oficial)
         elif "Visão" in motor_real and tem_imagem:
             with st.spinner("Analisando imagens..."):
                 content_list = [{"type": "text", "text": prompt}]
                 for img_b64 in imagens_b64:
                     content_list.append({"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{img_b64}"}})
                 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": content_list} ], "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})
+                    client_or_vision = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=or_key)
+                    res = client_or_vision.chat.completions.create(
+                        model=modelo_id,
+                        messages=[
+                            {"role": "system", "content": prompt_sistema_atual},
+                            {"role": "user", "content": content_list}
+                        ],
+                        max_tokens=4000
+                    )
+                    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 com Múltiplos Arquivos)
+        # 5. Texto (Streaming Blindado com biblioteca oficial)
         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."
@@ -345,21 +379,28 @@ if prompt or st.session_state.regerar:
             ph = st.empty(); full = ""
             
             try:
-                resp = 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}] + historico, "stream": True, "temperature": 0.8, "max_tokens": 4096}, stream=True, timeout=60)
-                if resp.status_code == 200:
-                    for l in resp.iter_lines():
-                        if l and l.startswith(b'data: '):
-                            decoded = l.decode('utf-8')[6:]
-                            if decoded == '[DONE]': break
-                            try:
-                                d = json.loads(decoded)
-                                full += d['choices'][0]['delta'].get('content', '')
-                                ph.markdown(full + "▌")
-                            except: pass
-                    ph.markdown(full)
-            except Exception as e: st.error(f"Oscilação de rede: {e}")
+                client_stream = OpenAI(base_url="https://openrouter.ai/api/v1", api_key=or_key)
+                resp = client_stream.chat.completions.create(
+                    model=modelo_id,
+                    messages=[{"role": "system", "content": prompt_sistema_atual}] + historico,
+                    stream=True,
+                    temperature=0.8,
+                    max_tokens=4096
+                )
+                
+                for chunk in resp:
+                    if chunk.choices and chunk.choices[0].delta.content:
+                        full += chunk.choices[0].delta.content
+                        ph.markdown(full + "▌")
+                ph.markdown(full)
+                
+            except Exception as e: 
+                st.error(f"Oscilação de rede ou erro na API: {e}")
             finally:
                 if full.strip() and len(st.session_state.db[st.session_state.current_chat]["messages"]) == len(mensagens):
                     st.session_state.db[st.session_state.current_chat]["messages"].append({"role": "assistant", "content": full})
             
+    # Incrementa a chave do uploader para resetar a gaveta de arquivos
+    st.session_state.uploader_key += 1
     save_db(st.session_state.db)
+    st.rerun() # Força o rerun para limpar a gaveta visualmente