Update app.py

app.py

@@ -21,12 +21,31 @@ st.markdown("""
     [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;}
-    .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; }
+    
+    /* Botões 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; }
     [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) */
+    [data-testid="stMain"] [data-testid="stHorizontalBlock"] button {
+        background-color: transparent !important;
+        border: none !important;
+        box-shadow: none !important;
+        font-size: 24px !important;
+        color: #a0a0a0 !important;
+        padding: 0px !important;
+        display: flex;
+        justify-content: center;
+        align-items: center;
+    }
+    [data-testid="stMain"] [data-testid="stHorizontalBlock"] button:hover {
+        color: #ffffff !important;
+        transform: scale(1.1);
+        transition: 0.2s;
+    }
     </style>
     """, unsafe_allow_html=True)
 
@@ -82,10 +101,7 @@ def analisar_intencao_gerente(prompt, api_key):
     
     payload = {
         "model": "nousresearch/hermes-2-pro-llama-3-8b", 
-        "messages": [
-            {"role": "system", "content": prompt_gerente}, 
-            {"role": "user", "content": prompt}
-        ], 
+        "messages": [{"role": "system", "content": prompt_gerente}, {"role": "user", "content": prompt}], 
         "temperature": 0.1
     }
     try: 
@@ -97,32 +113,25 @@ def analisar_intencao_gerente(prompt, api_key):
 def load_db():
     if os.path.exists(DB_FILE):
         try:
-            with open(DB_FILE, "r", encoding="utf-8") as f: 
-                return json.load(f)
-        except: 
-            pass
+            with open(DB_FILE, "r", encoding="utf-8") as f: return json.load(f)
+        except: pass
     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)
+    with open(DB_FILE, "w", encoding="utf-8") as f: json.dump(db, f, ensure_ascii=False, indent=4)
 
-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 "personalidade_ativa" not in st.session_state: 
-    st.session_state.personalidade_ativa = "❄️ Yukina (Companheira Obsessiva)"
+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 "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
 
 # ═══════════════════════════════════════════════════════════════════════════
 # 3. SIDEBAR E INTERFACE PRINCIPAL
 # ═══════════════════════════════════════════════════════════════════════════
 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')}"
@@ -137,12 +146,10 @@ with st.sidebar:
             arquivo_import = st.file_uploader("Importar conversa:", type=["json"])
             if arquivo_import:
                 try: 
-                    dados_importados = json.load(arquivo_import)
-                    st.session_state.db.update(dados_importados)
+                    st.session_state.db.update(json.load(arquivo_import))
                     save_db(st.session_state.db)
                     st.success("Sincronizado!")
-                except: 
-                    st.error("Erro no ficheiro.")
+                except: st.error("Erro no ficheiro.")
 
     st.markdown("<br>", unsafe_allow_html=True)
     busca = st.text_input("Pesquisar", placeholder="🔍 Pesquisar...", label_visibility="collapsed")
@@ -161,14 +168,12 @@ with st.sidebar:
             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()
+                    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()
+                        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"]
@@ -178,46 +183,61 @@ if len(mensagens) == 0:
 else:
     for m in mensagens:
         with st.chat_message(m["role"]):
-            if "image_url" in m: 
-                st.image(m["image_url"])
-            elif "video_url" in m: 
-                st.video(m["video_url"])
-            else: 
-                st.markdown(m["content"])
+            if "image_url" in m: st.image(m["image_url"])
+            elif "video_url" in m: st.video(m["video_url"])
+            else: st.markdown(m["content"])
 
 # ═══════════════════════════════════════════════════════════════════════════
-# 4. TOOLBAR INFERIOR E PROCESSAMENTO DE AGENTES
+# 4. TOOLBAR INFERIOR (Ícones Horizontais)
 # ═══════════════════════════════════════════════════════════════════════════
 st.markdown("<br>", unsafe_allow_html=True)
-t_col1, t_col2, t_space = st.columns([1.5, 1.5, 7])
+t_col1, t_col2, t_col3, t_col4, t_space = st.columns([1, 1, 1, 1, 6])
+
 with t_col1:
-    with st.popover("＋"): 
-        upload_file = st.file_uploader("Mídia ou Dados", type=["png", "jpg", "jpeg", "txt", "csv", "json"])
+    with st.popover("➕", help="Upload de Mídia ou Dados"): 
+        upload_file = st.file_uploader("Upload", type=["png", "jpg", "jpeg", "txt", "csv", "json"], label_visibility="collapsed")
 with t_col2:
-    with st.popover("⚙"): 
+    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))
         st.markdown("---")
         st.session_state.personalidade_ativa = st.radio("ALMA (Personalidade):", list(PERSONALIDADES.keys()), index=list(PERSONALIDADES.keys()).index(st.session_state.personalidade_ativa))
-
-with t_space:
-    if len(mensagens) > 0 and st.button("🗑️ Desfazer Última"):
+with t_col3:
+    if st.button("🗑️", help="Apagar a ú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_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 prompt := st.chat_input("Peça à Yukina..."):
-    conteudo_arquivo = ""
-    mensagem_display = prompt
-    if upload_file is not None and upload_file.name.endswith(('.txt', '.csv', '.json')):
-        conteudo_arquivo = upload_file.getvalue().decode("utf-8")
-        mensagem_display = f"📄 **Arquivo enviado:** `{upload_file.name}`\n\n{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)
+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"]
+        st.session_state.regerar = False
+    else:
+        mensagem_display = prompt
+        if conteudo_arquivo:
+            mensagem_display = f"📄 **Arquivo enviado:** `{upload_file.name}`\n\n{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("assistant"):
         or_key = os.getenv("OPENROUTER_API_KEY")
@@ -320,26 +340,4 @@ if prompt := st.chat_input("Peça à Yukina..."):
                     else:
                         matches = re.findall(r'(https?://[^\s)]+)', msg_obj.get('content', ''))
                         if matches:
-                            url = matches[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})
-                    else:
-                        st.error("Erro ao gerar a imagem.")
-                except Exception as e: 
-                    st.error(f"Erro: {e}")
-
-        # 4. Visão (MiMo V2)
-        elif "Visão" in motor_real and upload_file is not None and upload_file.name.endswith(('.png', '.jpg', '.jpeg')):
-            with st.spinner("MiMo a analisar..."):
-                img_b64 = base64.b64encode(upload_file.read()).decode()
-                payload_visao = {
-                    "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
-                }
-              
+                            url = matches