Update app.py

Melhorias na aba Chatbot Assistente

app.py

@@ -139,12 +139,18 @@ CACHE_DIR.mkdir(exist_ok=True)
 def _inicializar_firestore():
     """
     Inicializa o Firebase Admin SDK usando as credenciais
-    armazenadas nos Secrets do Hugging Face Spaces.
+    armazenadas nos Secrets do Hugging Face Spaces ou Colab.
     """
     global db, analytics
-    
+
+    # Preferir userdata no Colab
     secret_name = "FIREBASE_SERVICE_ACCOUNT_JSON"
-    secret_json_string = os.environ.get(secret_name)
+    secret_json_string = userdata.get(secret_name)
+
+    if not secret_json_string:
+         # Fallback para ambiente Hugging Face Spaces
+        secret_json_string = os.environ.get(secret_name)
+
 
     if not secret_json_string:
         print(f"❌ Erro de Configuração do Firebase: Secret '{secret_name}' não encontrado.")
@@ -252,18 +258,18 @@ def _salvar_analytics_firestore():
 def atualizar_analytics(nicho, estilo, palavras, imagem_gerada, cache_hit, favorito):
     """Atualiza as métricas de analytics (agora salva no Firestore)."""
     global analytics
-    
+
     analytics['total_posts'] = analytics.get('total_posts', 0) + 1
     analytics['total_palavras'] = analytics.get('total_palavras', 0) + palavras
-    
+
     if imagem_gerada:
         analytics['total_imagens'] = analytics.get('total_imagens', 0) + 1
-        
+
     if cache_hit:
         analytics['cache_hits'] = analytics.get('cache_hits', 0) + 1
     else:
         analytics['cache_misses'] = analytics.get('cache_misses', 0) + 1
-        
+
     if favorito:
         analytics['total_favoritos'] = analytics.get('total_favoritos', 0) + 1
 
@@ -271,11 +277,11 @@ def atualizar_analytics(nicho, estilo, palavras, imagem_gerada, cache_hit, favor
     nicho_counts = analytics.get('posts_por_nicho', {})
     nicho_counts[nicho] = nicho_counts.get(nicho, 0) + 1
     analytics['posts_por_nicho'] = nicho_counts
-    
+
     estilo_counts = analytics.get('posts_por_estilo', {})
     estilo_counts[estilo] = estilo_counts.get(estilo, 0) + 1
     analytics['posts_por_estilo'] = estilo_counts
-    
+
     # Salvar no Firestore
     _salvar_analytics_firestore()
 
@@ -284,14 +290,14 @@ def gerar_relatorio_analytics():
     global analytics
     if not analytics or 'status' in analytics or analytics.get("total_posts", 0) == 0:
         return "📊 Nenhum post gerado ainda."
-        
+
     # Ordenar os dicionários por valor (mais usados primeiro)
     posts_por_nicho_sorted = dict(sorted(analytics.get('posts_por_nicho', {}).items(), key=lambda item: item[1], reverse=True))
     posts_por_estilo_sorted = dict(sorted(analytics.get('posts_por_estilo', {}).items(), key=lambda item: item[1], reverse=True))
 
     total_reqs = analytics.get('cache_hits', 0) + analytics.get('cache_misses', 0)
     taxa_cache_hit = (analytics.get('cache_hits', 0) / total_reqs * 100) if total_reqs > 0 else 0
-    
+
     nicho_top = max(analytics["posts_por_nicho"].items(), key=lambda x: x[1]) if analytics.get("posts_por_nicho") else ("N/A", 0)
     estilo_top = max(analytics["posts_por_estilo"].items(), key=lambda x: x[1]) if analytics.get("posts_por_estilo") else ("N/A", 0)
 
@@ -357,18 +363,18 @@ def buscar_no_cache(key):
         try:
             with open(cache_file, 'r', encoding='utf-8') as f:
                 data = json.load(f)
-            
+
             texto = data.get("texto")
             imagem_path = data.get("imagem_path")
             imagem = None
-            
+
             if imagem_path:
                 img_cache_file = CACHE_DIR / imagem_path
                 if img_cache_file.exists():
                     imagem = Image.open(img_cache_file)
                 else:
                     return None, None
-            
+
             return texto, imagem
         except Exception as e:
             print(f"Erro ao ler cache {key}: {e}")
@@ -379,7 +385,7 @@ def salvar_imagem_cache(key, imagem_pil):
     """Salva a imagem PIL no diretório de cache e retorna o nome do arquivo."""
     if not imagem_pil:
         return None
-    
+
     try:
         imagem_path = f"{key}_img.png"
         imagem_pil.save(CACHE_DIR / imagem_path)
@@ -437,6 +443,10 @@ def copiar_feedback(texto):
         return criar_alerta('success', '✅ Texto copiado!')
     return criar_alerta('warning', '⚠️ Nada para copiar')
 
+def print_like_dislike(x: gr.LikeData):
+    """Função de callback para o evento 'like' do chatbot."""
+    print(f"Mensagem {x.index} foi marcada como: {x.value}, Liked: {x.liked}")
+
 def limpar_tudo():
     """Limpa todos os inputs da UI, incluindo filtros de histórico."""
     analytics_data = gerar_relatorio_analytics() # Gerar relatório limpo
@@ -506,34 +516,34 @@ def interpretar_erro_api(erro_str):
 def filtrar_historico_local(query, nicho, estilo, formato, favoritos_apenas):
     """Filtra a lista global `post_history` com base nos inputs da UI."""
     global post_history
-    
+
     # Começa com a lista completa
     resultados = post_history
-    
+
     # Filtro de busca por texto
     if query:
         query_lower = query.lower()
         resultados = [
-            post for post in resultados 
+            post for post in resultados
             if query_lower in post.get("Tema", "").lower() or query_lower in post.get("Texto", "").lower()
         ]
-        
+
     # Filtro de Nicho
     if nicho != "Todos":
         resultados = [post for post in resultados if post.get("Nicho") == nicho]
-        
+
     # Filtro de Estilo
     if estilo != "Todos":
         resultados = [post for post in resultados if post.get("Estilo") == estilo]
-        
+
     # Filtro de Formato
     if formato != "Todos":
         resultados = [post for post in resultados if post.get("Formato") == formato]
-        
+
     # Filtro de Favoritos
     if favoritos_apenas:
         resultados = [post for post in resultados if post.get("Favorito") == True]
-        
+
     # Formata para o Dataframe
     return _formatar_historico_para_df(resultados)
 
@@ -546,7 +556,7 @@ def carregar_post_do_historico(evt: gr.SelectData):
     try:
         # Pega o post correspondente da lista global
         post_selecionado = post_history[evt.index]
-        
+
         # Extrai os dados
         tema = post_selecionado.get("Tema", "")
         nicho = post_selecionado.get("Nicho", NICHOS_DISPONIVEIS[0])
@@ -554,10 +564,10 @@ def carregar_post_do_historico(evt: gr.SelectData):
         formato = post_selecionado.get("Formato", list(FORMATO_CONFIGS.keys())[0])
         favorito = post_selecionado.get("Favorito", False)
         texto = post_selecionado.get("Texto", "")
-        
+
         # Feedback para o usuário
         status_alerta = criar_alerta('info', '✅ Post carregado do histórico! A imagem deve ser gerada novamente se desejado.')
-        
+
         # Retorna os valores para a UI, incluindo a mudança de aba
         return (
             gr.Tabs(selected=0), # Muda para a primeira aba (Gerar Post)
@@ -585,12 +595,12 @@ def gerar_texto(tema, nicho, estilo, formato):
     """
     Gera texto usando API do Hugging Face com base no formato escolhido.
     """
-    
+
     if not HUGGINGFACE_API_KEY:
         return "❌ Erro de Configuração: API Key não está definida."
 
     config = FORMATO_CONFIGS.get(formato, FORMATO_CONFIGS["Instagram (Post)"])
-    
+
     url = f"{BASE_URL}/chat/completions"
 
     payload = {
@@ -646,7 +656,7 @@ def traduzir_texto(texto_pt):
 
     try:
         response = requests.post(url, headers=headers, json=payload, timeout=30)
-        
+
         if response.status_code == 200:
             resultado = response.json()
             if resultado and isinstance(resultado, list) and 'translation_text' in resultado[0]:
@@ -662,12 +672,12 @@ def traduzir_texto(texto_pt):
 
 def otimizar_prompt_imagem(descricao_pt, estilo_escolhido, filtro_escolhido):
     """Combina as escolhas do usuário em um prompt otimizado (em Português)."""
-    
+
     estilo = ESTILOS_DE_IMAGEM.get(estilo_escolhido, ESTILOS_DE_IMAGEM["Nenhum (Automático)"])
     filtro = FILTROS_IMAGEM.get(filtro_escolhido, FILTROS_IMAGEM["Nenhum"])
-    
+
     prompt_final = f"{descricao_pt}, {estilo}, {filtro}, best quality, 4k"
-    
+
     prompt_final = prompt_final.replace(", ,", ",").replace(", ,", ",")
     return prompt_final
 
@@ -680,7 +690,7 @@ def gerar_imagem_robusta(descricao_pt, estilo_escolhido, qualidade, filtro_escol
     Gera imagem com sistema robusto de fallback e controle de qualidade.
     Retorna: (PIL.Image, str_mensagem_status)
     """
-    
+
     # 1. Configs de Qualidade
     configs_qualidade = {
         "Rápida": {
@@ -697,7 +707,7 @@ def gerar_imagem_robusta(descricao_pt, estilo_escolhido, qualidade, filtro_escol
         }
     }
     config = configs_qualidade.get(qualidade, configs_qualidade["Balanceada"])
-    
+
     # 2. Otimizar e Traduzir Prompt
     if progress: progress(0.55, desc="🌍 Otimizando e traduzindo prompt...")
     prompt_otimizado_pt = otimizar_prompt_imagem(descricao_pt, estilo_escolhido, filtro_escolhido)
@@ -710,26 +720,26 @@ def gerar_imagem_robusta(descricao_pt, estilo_escolhido, qualidade, filtro_escol
             if progress:
                 prog_val = 0.6 + (i * 0.1) # Ajustar progresso
                 progress(prog_val, desc=f"🎨 Tentando {modelo_config['nome']}...")
-            
+
             print(f"Tentando gerar imagem com {modelo_config['nome']}...")
 
             client = InferenceClient(api_key=HUGGINGFACE_API_KEY)
-            
+
             imagem = client.text_to_image(
                 prompt=prompt_final_en,
                 model=modelo_config['id'],
                 negative_prompt=negative_prompt,
                 num_inference_steps=config['steps']
             )
-            
+
             print(f"✅ Imagem gerada com {modelo_config['nome']}")
             mensagem = f"✅ Imagem gerada com {modelo_config['nome']}"
-            
+
             return (imagem, mensagem) # Retorna (PIL.Image, str)
 
         except Exception as e:
             print(f"❌ Falha com {modelo_config['nome']}: {str(e)}")
-            
+
             if i < len(config['modelos']) - 1:
                 print(f"⏭️ Tentando próximo modelo...")
                 continue
@@ -744,6 +754,10 @@ def gerar_imagem_robusta(descricao_pt, estilo_escolhido, qualidade, filtro_escol
 # FUNÇÃO DO CHATBOT
 # ============================================
 def responder_chat(message, chat_history):
+    """
+    Função principal de lógica do chatbot. Recebe a nova mensagem e o histórico,
+    retorna a string de resposta da IA.
+    """
     if not HUGGINGFACE_API_KEY:
         return "❌ Erro de Configuração: API Key não está definida."
 
@@ -751,8 +765,11 @@ def responder_chat(message, chat_history):
     
     system_prompt = "Você é um assistente virtual prestativo e amigável, especializado em marketing de mídias sociais e criação de conteúdo, mas pode responder sobre qualquer tópico. Seja direto e útil."
     
+    # Constrói o payload de mensagens
     messages = [{"role": "system", "content": system_prompt}]
+    # Adiciona o histórico existente
     messages.extend(chat_history)
+    # Adiciona a nova mensagem do usuário
     messages.append({"role": "user", "content": message})
 
     payload = {
@@ -779,6 +796,21 @@ def responder_chat(message, chat_history):
     except Exception as e:
         return f"❌ {interpretar_erro_api(str(e))}"
 
+def chatbot_respond(message, chat_history):
+    """
+    Função wrapper para a UI do Gradio.
+    Recebe a mensagem e o histórico, chama a lógica do bot,
+    e retorna o histórico atualizado.
+    """
+    # 1. Adiciona a mensagem do usuário ao histórico
+    chat_history.append({"role": "user", "content": message})
+    # 2. Obtém a resposta do bot (string)
+    bot_response_str = responder_chat(message, chat_history)
+    # 3. Adiciona a resposta do bot ao histórico
+    chat_history.append({"role": "assistant", "content": bot_response_str})
+    # 4. Retorna a caixa de texto vazia e o histórico atualizado
+    return "", chat_history
+
 # ============================================
 # FUNÇÕES DE DOWNLOAD
 # ============================================
@@ -791,14 +823,14 @@ def criar_post_completo(texto, imagem_pil, tema):
     if not imagem_pil or not texto:
         print("Erro: Texto ou Imagem ausente para criar post.")
         return None
-    
+
     try:
         LARGURA_POST = 1080
         ALTURA_IMAGEM = 1080
         PADDING = 60
         COR_FUNDO = (255, 255, 255)
         COR_TEXTO = (0, 0, 0)
-        
+
         try:
             fonte_texto = ImageFont.truetype("DejaVuSans.ttf", size=42)
             fonte_titulo = ImageFont.truetype("DejaVuSans-Bold.ttf", size=55)
@@ -808,30 +840,30 @@ def criar_post_completo(texto, imagem_pil, tema):
             fonte_titulo = ImageFont.load_default()
 
         imagem_quadrada = imagem_pil.resize((LARGURA_POST, ALTURA_IMAGEM), Image.Resampling.LANCZOS)
-        
+
         linhas_titulo = textwrap.wrap(tema.upper(), width=40)
         linhas_texto = textwrap.wrap(texto, width=50)
-        
-        altura_titulo = len(linhas_titulo) * 60  
+
+        altura_titulo = len(linhas_titulo) * 60
         altura_texto = len(linhas_texto) * 45
-        altura_total_texto = altura_titulo + 20 + altura_texto + (PADDING * 2)  
-        
+        altura_total_texto = altura_titulo + 20 + altura_texto + (PADDING * 2)
+
         altura_total = ALTURA_IMAGEM + altura_total_texto
         post_completo = Image.new('RGB', (LARGURA_POST, int(altura_total)), COR_FUNDO)
-        
+
         post_completo.paste(imagem_quadrada, (0, 0))
-        
+
         draw = ImageDraw.Draw(post_completo)
         pos_y = ALTURA_IMAGEM + PADDING
-        
+
         for linha in linhas_titulo:
             largura_linha = draw.textlength(linha, font=fonte_titulo)
             pos_x_titulo = (LARGURA_POST - largura_linha) / 2
             draw.text((pos_x_titulo, pos_y), linha, font=fonte_titulo, fill=COR_TEXTO)
             pos_y += 60
-        
+
         pos_y += 20
-        
+
         for linha in linhas_texto:
             draw.text((PADDING, pos_y), linha, font=fonte_texto, fill=COR_TEXTO)
             pos_y += 45
@@ -840,7 +872,7 @@ def criar_post_completo(texto, imagem_pil, tema):
             post_completo.save(f, 'PNG')
             print(f"Arquivo temporário salvo em: {f.name}")
             return f.name
-    
+
     except Exception as e:
         print(f"❌ Erro ao criar post completo: {e}")
         return None
@@ -857,14 +889,14 @@ def preparar_download(texto, imagem_pil, tema):
 
     if caminho_arquivo:
         return caminho_arquivo
-    
+
     return None
 
 # ============================================
 # FUNÇÃO PRINCIPAL
 # ============================================
 
-def gerar_post_interface(tema, nicho, estilo, formato, usar_cache, favorito_checkbox,
+def gerar_post_interface(tema, nicho, estilo, formato, usar_cache, favorito_checkbox_value, # Capture the boolean value
                          descricao_imagem, gerar_img,
                          estilo_img_input, qualidade_img_input, filtro_img_input,
                          progress=gr.Progress()):
@@ -872,12 +904,12 @@ def gerar_post_interface(tema, nicho, estilo, formato, usar_cache, favorito_chec
     Função principal unificada, com Cache, Analytics, Favoritos e Geração Avançada.
     Retorna 7 valores para a UI.
     """
-    
+
     analytics_display = gerar_relatorio_analytics() # Carregar estado atual
-    
+
     progress(0, desc="🚀 Iniciando...")
     time.sleep(0.3)
-    
+
     progress(0.1, desc="✅ Validando...")
     if not tema or len(tema.strip()) < 3:
         status_final = criar_alerta('error', '⚠️ Digite um tema válido!')
@@ -889,57 +921,57 @@ def gerar_post_interface(tema, nicho, estilo, formato, usar_cache, favorito_chec
     if usar_cache:
         progress(0.2, desc="🔍 Buscando no cache...")
         texto, imagem = buscar_no_cache(cache_key)
-        
+
         if texto:
             print("✅ Cache hit!")
             progress(1.0, desc="🎉 Encontrado no cache!")
             status_final = criar_alerta('success', '🎉 Post carregado do cache!')
-            
-            atualizar_analytics(nicho, estilo, len(texto.split()), (imagem is not None), cache_hit=True, favorito=favorito_checkbox)
+
+            atualizar_analytics(nicho, estilo, len(texto.split()), (imagem is not None), cache_hit=True, favorito=favorito_checkbox_value) # Pass the boolean value
             analytics_display = gerar_relatorio_analytics() # Recarregar
-            
+
             palavras = len(texto.split())
             caracteres = len(texto)
             hashtags = texto.count('#')
-            
+
             history_entry = {
                 "Data/Hora": datetime.now(ZoneInfo("America/Bahia")).strftime("%Y-%m-%d %H:%M:%S"),
                 "Tema": tema, "Nicho": nicho, "Estilo": estilo, "Formato": formato,
                 "Texto": texto,
                 "Status": "Carregado do Cache",
-                "Favorito": favorito_checkbox
+                "Favorito": favorito_checkbox_value # Pass the boolean value
             }
             atualizar_historico(history_entry)
-            
+
             return (texto, imagem, status_final, palavras, caracteres, hashtags, analytics_display)
-    
+
     print("Cache miss ou cache desativado.")
     progress(0.3, desc="🤖 Gerando texto (Llama 3.1)...")
-    
+
     # 2. Gerar Texto
-    texto = gerar_texto(tema, nicho, estilo, formato)  
-    
+    texto = gerar_texto(tema, nicho, estilo, formato)
+
     if texto.startswith("❌"):
         status_final = criar_alerta('error', f'{texto}')
         return (texto, None, status_final, 0, 0, 0, analytics_display)
-    
+
     progress(0.5, desc="✅ Texto pronto!")
     time.sleep(0.5)
-    
+
     # 3. Gerar Imagem
     imagem = None
     status_imagem = ""
     if gerar_img:
         descricao_pt = descricao_imagem or f"{tema} imagem"
-        
+
         (imagem, status_imagem) = gerar_imagem_robusta(
-            descricao_pt,  
-            estilo_img_input,  
-            qualidade_img_input,  
-            filtro_img_input,  
+            descricao_pt,
+            estilo_img_input,
+            qualidade_img_input,
+            filtro_img_input,
             progress
         )
-        
+
         if imagem:
             status_final = criar_alerta('success', f'🎉 Post completo gerado! ({status_imagem})')
         else:
@@ -947,16 +979,16 @@ def gerar_post_interface(tema, nicho, estilo, formato, usar_cache, favorito_chec
     else:
         progress(0.7, desc="⏭️ Pulando geração de imagem...")
         status_final = criar_alerta('success', '✅ Texto gerado (sem imagem)!')
-    
+
     time.sleep(0.5)
-    
+
     # 4. Estatísticas
     progress(0.9, desc="📊 Calculando estatísticas...")
     palavras = len(texto.split())
     caracteres = len(texto)
     hashtags = texto.count('#')
     time.sleep(0.3)
-    
+
     # 5. Salvar no Cache
     if usar_cache:
         progress(0.95, desc="💾 Salvando no cache...")
@@ -966,23 +998,23 @@ def gerar_post_interface(tema, nicho, estilo, formato, usar_cache, favorito_chec
             "imagem_path": imagem_path_cache
         }
         salvar_no_cache(cache_key, cache_data)
-        
+
     # 6. Atualizar Histórico (Firestore)
     history_entry = {
         "Data/Hora": datetime.now(ZoneInfo("America/Bahia")).strftime("%Y-%m-%d %H:%M:%S"),
         "Tema": tema, "Nicho": nicho, "Estilo": estilo, "Formato": formato,
         "Texto": texto, # Salva o texto completo
         "Status": status_imagem or "Texto Gerado",
-        "Favorito": favorito_checkbox
+        "Favorito": favorito_checkbox_value # Pass the boolean value
     }
     atualizar_historico(history_entry)
-    
+
     # 7. Atualizar Analytics (Firestore)
-    atualizar_analytics(nicho, estilo, palavras, (imagem is not None), cache_hit=False, favorito=favorito_checkbox)
+    atualizar_analytics(nicho, estilo, palavras, (imagem is not None), cache_hit=False, favorito=favorito_checkbox_value) # Pass the boolean value
     analytics_display = gerar_relatorio_analytics() # Recarregar
-    
+
     progress(1.0, desc="🎉 Pronto!")
-    
+
     return (texto, imagem, status_final, palavras, caracteres, hashtags, analytics_display)
 
 
@@ -1006,27 +1038,27 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
     # 🚀 Gerador de Posts e Assistente de Mídias Sociais (Versão Completa)
     ### Powered by Hugging Face, Gradio, Llama 3.1 e Firebase
     """)
-    
+
     with gr.Tabs() as main_tabs: # Adicionado 'as main_tabs' para controle
         with gr.TabItem("✨ Gerar Post", id=0):
             with gr.Row():
                 with gr.Column(scale=1):
                     gr.Markdown("### ⚙️ 1. Configurações do Texto")
-        
+
                     nicho_input = gr.Dropdown(
                         choices=NICHOS_DISPONIVEIS,
                         label="Escolha um nicho (para o texto)",
                         value=NICHOS_DISPONIVEIS[0],
                         interactive=True
                     )
-        
+
                     estilo_input = gr.Radio(
                         choices=ESTILOS_DISPONIVEIS,
                         label="Escolha um estilo (para o texto)",
                         value=ESTILOS_DISPONIVEIS[0],
                         interactive=True
                     )
-        
+
                     tema_input = gr.Textbox(
                         label="Tema do Post",
                         placeholder="Ex: Transforme seu corpo, transforme sua vida"
@@ -1038,25 +1070,25 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
                         value=list(FORMATO_CONFIGS.keys())[0],
                         interactive=True
                     )
-                    
+
                     usar_cache_checkbox = gr.Checkbox(
                         label="Usar Cache? (Acelera posts repetidos)",
                         value=True
                     )
-        
+
                     gr.Markdown("### 🎨 2. Configurações da Imagem (Opcional)")
-        
+
                     gerar_img_checkbox = gr.Checkbox(
                         label="Gerar imagem para o post?",
                         value=False
                     )
-        
+
                     descricao_img_input = gr.Textbox(
                         label="Descrição da imagem (em Português)",
                         placeholder="Ex: Pessoa correndo ao nascer do sol",
                         visible=False
                     )
-                    
+
                     estilo_img_input = gr.Dropdown(
                         label="Estilo da Imagem",
                         choices=list(ESTILOS_DE_IMAGEM.keys()),
@@ -1064,7 +1096,7 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
                         visible=False,
                         interactive=True
                     )
-                    
+
                     qualidade_img_input = gr.Radio(
                         label="Qualidade da Imagem (Velocidade vs Qualidade)",
                         choices=["Rápida", "Balanceada", "Alta"],
@@ -1072,7 +1104,7 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
                         visible=False,
                         interactive=True
                     )
-                    
+
                     filtro_img_input = gr.Dropdown(
                         label="Filtro da Imagem",
                         choices=list(FILTROS_IMAGEM.keys()),
@@ -1080,7 +1112,7 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
                         visible=False,
                         interactive=True
                     )
-        
+
                     def toggle_descricao_img(gerar):
                         return (
                             gr.Textbox(visible=gerar),
@@ -1088,41 +1120,41 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
                             gr.Radio(visible=gerar),
                             gr.Dropdown(visible=gerar)
                         )
-        
+
                     gerar_img_checkbox.change(
                         toggle_descricao_img,
                         inputs=[gerar_img_checkbox],
                         outputs=[descricao_img_input, estilo_img_input, qualidade_img_input, filtro_img_input]
                     )
-                    
+
                     favorito_checkbox = gr.Checkbox(label="⭐ Favoritar este post?", value=False)
-        
+
                     gerar_btn = gr.Button("✨ Gerar Post", variant="primary")
-                
+
                 with gr.Column(scale=1):
                     gr.Markdown("### 📋 3. Resultado")
-        
+
                     status_output = gr.HTML(
                         label="Status",
                         value=criar_alerta('info', 'Pronto para gerar!')
                     )
-        
+
                     texto_output = gr.Textbox(
                         label="Texto Gerado",
                         lines=10,
                         interactive=True, # Alterado para True para carregar do histórico
                         show_copy_button=True
                     )
-                    
+
                     with gr.Row():
                         copiar_btn = gr.Button("📋 Copiar Texto", variant="secondary")
                         limpar_btn = gr.Button("🧹 Limpar Tudo", variant="stop")
-        
+
                     imagem_output = gr.Image(
                         label="Imagem Gerada",
                         type="pil"
                     )
-                    
+
                     gr.Markdown("### 📥 4. Download")
                     download_btn = gr.Button(
                         "Baixar Post Completo (Imagem + Texto)",
@@ -1132,7 +1164,7 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
                         label="Clique para baixar",
                         visible=True
                     )
-                    
+
                     gr.Markdown("### 📊 Estatísticas do Texto")
                     with gr.Row():
                         palavras_output = gr.Number(label="Palavras", value=0, interactive=False)
@@ -1140,7 +1172,7 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
                         hashtags_output = gr.Number(label="Hashtags", value=0, interactive=False)
 
             gr.Markdown("### 💡 Experimente estes exemplos:")
-            
+
             gr.Examples(
                 examples=[
                     [NICHOS_DISPONIVEIS[2], ESTILOS_DISPONIVEIS[0], "Frases marcantes de pessoas importantes", "Instagram (Post)"],
@@ -1151,68 +1183,96 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
                 inputs=[nicho_input, estilo_input, tema_input, formato_input],
                 outputs=[texto_output, imagem_output, status_output]
             )
-            
+
         with gr.TabItem("💬 Chatbot Assistente", id=1):
             gr.Markdown("### 🤖 Assistente Virtual")
             gr.Markdown("Faça perguntas sobre mídias sociais, IA, peça ideias rápidas ou qualquer outro tópico.")
             
             chatbot_para_interface = gr.Chatbot(
                 height=500,  
-                type="messages"
+                type="messages"                
             )
+            
+            with gr.Column():
+                with gr.Row():
+                    chat_input = gr.Textbox(
+                        show_label=False,
+                        placeholder="Digite sua mensagem aqui...",
+                        scale=7
+                    )
+                    submit_btn = gr.Button("Enviar", variant="primary", scale=1)
+                
+                clear_btn = gr.ClearButton(
+                    [chat_input, chatbot_para_interface],
+                    value="🧹 Limpar Chat"
+                )
 
-            gr.ChatInterface(
-                fn=responder_chat,
-                title="Assistente Virtual",
-                description="Converse com o Llama 3.1 para obter ajuda e insights.",
+            gr.Examples(
                 examples=[
                     "O que é um 'gancho' para Instagram?",  
                     "Me dê 3 ideias de post para um nicho de 'Fitness'",  
                     "Qual a diferença entre um post para Instagram e um para LinkedIn?"
                 ],
-                chatbot=chatbot_para_interface,
-                textbox=gr.Textbox(placeholder="Digite sua mensagem aqui...", scale=7),
-                submit_btn="Enviar"
+                inputs=[chat_input]
+            )
+
+            # Conectar eventos do chatbot
+            submit_btn.click(
+                fn=chatbot_respond,
+                inputs=[chat_input, chatbot_para_interface],
+                outputs=[chat_input, chatbot_para_interface]
+            )
+            
+            chat_input.submit(
+                fn=chatbot_respond,
+                inputs=[chat_input, chatbot_para_interface],
+                outputs=[chat_input, chatbot_para_interface]
             )
             
+            chatbot_para_interface.like(
+                fn=print_like_dislike, 
+                inputs=None, 
+                outputs=None
+            )
+
         with gr.TabItem("📚 Histórico de Posts", id=2):
             gr.Markdown("### 🔍 Buscar e Filtrar Histórico")
             gr.Markdown("Carregue posts antigos clicando em uma linha da tabela.")
 
             with gr.Row():
                 busca_query_input = gr.Textbox(
-                    label="Buscar por Tema/Texto", 
-                    placeholder="Digite para buscar...", 
+                    label="Buscar por Tema/Texto",
+                    placeholder="Digite para buscar...",
                     scale=3,
                     interactive=True
                 )
                 filtro_nicho_hist = gr.Dropdown(
-                    label="Nicho", 
-                    choices=["Todos"] + NICHOS_DISPONIVEIS, 
+                    label="Nicho",
+                    choices=["Todos"] + NICHOS_DISPONIVEIS,
                     value="Todos",
                     interactive=True
                 )
             with gr.Row():
                 filtro_estilo_hist = gr.Dropdown(
-                    label="Estilo", 
-                    choices=["Todos"] + ESTILOS_DISPONIVEIS, 
+                    label="Estilo",
+                    choices=["Todos"] + ESTILOS_DISPONIVEIS,
                     value="Todos",
                     interactive=True
                 )
                 filtro_formato_hist = gr.Dropdown(
-                    label="Formato", 
-                    choices=["Todos"] + list(FORMATO_CONFIGS.keys()), 
+                    label="Formato",
+                    choices=["Todos"] + list(FORMATO_CONFIGS.keys()),
                     value="Todos",
                     interactive=True
                 )
                 filtro_favoritos_hist = gr.Checkbox(
-                    label="⭐ Apenas Favoritos", 
+                    label="⭐ Apenas Favoritos",
                     value=False,
                     interactive=True
                 )
-            
+
             buscar_hist_btn = gr.Button("Buscar", variant="primary")
-            
+
             historico_display = gr.Dataframe(
                 headers=["⭐", "Data/Hora", "Tema", "Nicho", "Estilo", "Formato", "Texto (Preview)", "Status"],
                 interactive=True, # Habilitado para seleção
@@ -1222,11 +1282,11 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
         with gr.TabItem("📊 Analytics", id=3):
             gr.Markdown("### Análise de Uso da Ferramenta")
             gr.Markdown("Estes dados são salvos no Firestore e agregam o uso de todos os usuários.")
-            
+
             analytics_display = gr.Markdown(
                 value=gerar_relatorio_analytics()
             )
-            
+
             with gr.Row():
                 gerar_relatorio_btn = gr.Button("Atualizar Relatório", variant="secondary")
                 resetar_analytics_btn = gr.Button("Resetar Analytics (CUIDADO)", variant="stop")
@@ -1270,10 +1330,10 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
             **Como funciona:**
             1. **Gerar Post:** Você define o tema, nicho, estilo e **formato** do *texto*.
             2. **Imagem (Opcional):** Você ativa, descreve a imagem e seleciona *Estilo*, *Qualidade* e *Filtro*.
-            3. O sistema otimiza o prompt, traduz para inglês e usa o sistema de *fallback* de modelos (baseado na *Qualidade*) para gerar a imagem.
+            3. O sistema otimiza o prompt, traduz para inglês e usa o sistema de *fallback* de modelos (based on *Qualidade*) para gerar a imagem.
             4. **Download:** Após a geração, você pode clicar em "Baixar Post Completo" para salvar um PNG com a imagem e o texto formatado.
             5. **Chatbot:** Você pode conversar diretamente com a IA na aba 'Chatbot Assistente' para tirar dúvidas.
-            6. **Histórico & Analytics:** Os posts gerados são salvos no Firestore e as métricas de uso são atualizadas.
+            6. **Histórico & Analytics:** Os posts gerados são salvos no Firestore and as métricas de uso são atualizadas.
 
             **Desenvolvido por:** Wilder Paz
             """)
@@ -1287,22 +1347,22 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
     # ============================================
     # CONECTAR EVENTOS
     # ============================================
-    
+
     # Lista de inputs para o botão Gerar
     gerar_inputs = [
-        tema_input, nicho_input, estilo_input, 
+        tema_input, nicho_input, estilo_input,
         formato_input, usar_cache_checkbox, favorito_checkbox,
         descricao_img_input, gerar_img_checkbox,
         estilo_img_input, qualidade_img_input, filtro_img_input
     ]
-    
+
     # Lista de outputs do botão Gerar
     gerar_outputs = [
-        texto_output, imagem_output, status_output,  
+        texto_output, imagem_output, status_output,
         palavras_output, caracteres_output, hashtags_output,
         analytics_display
     ]
-    
+
     # Botão principal
     click_event = gerar_btn.click(
         fn=gerar_post_interface,
@@ -1310,20 +1370,20 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
         outputs=gerar_outputs,
         show_progress="full"
     )
-    
+
     # Botão copiar
     copiar_btn.click(
         fn=copiar_feedback,
         inputs=[texto_output],
         outputs=[status_output]
     )
-    
+
     # Lista de outputs para o botão Limpar
     limpar_outputs = [
         # Aba Gerador
-        texto_output, imagem_output, status_output,  
+        texto_output, imagem_output, status_output,
         palavras_output, caracteres_output, hashtags_output,
-        formato_input, 
+        formato_input,
         estilo_img_input, qualidade_img_input, filtro_img_input,
         download_output,
         usar_cache_checkbox,
@@ -1336,46 +1396,46 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
         filtro_formato_hist,
         filtro_favoritos_hist
     ]
-    
+
     # Botão limpar
     limpar_btn.click(
         fn=limpar_tudo,
         inputs=[],
         outputs=limpar_outputs
     )
-    
+
     # Botão de Download
     download_btn.click(
         fn=preparar_download,
         inputs=[texto_output, imagem_output, tema_input],
         outputs=[download_output]
     )
-    
+
     # --- Eventos da Aba Histórico ---
-    
+
     # Lista de inputs para os filtros de histórico
     hist_filter_inputs = [
-        busca_query_input, 
-        filtro_nicho_hist, 
-        filtro_estilo_hist, 
-        filtro_formato_hist, 
+        busca_query_input,
+        filtro_nicho_hist,
+        filtro_estilo_hist,
+        filtro_formato_hist,
         filtro_favoritos_hist
     ]
-    
+
     # Botão de buscar no histórico
     buscar_hist_btn.click(
         fn=filtrar_historico_local,
         inputs=hist_filter_inputs,
         outputs=[historico_display]
     )
-    
+
     # Atualizar o histórico (mantendo filtros) após gerar um novo post
     click_event.then(
         fn=recarregar_e_formatar_historico,
         inputs=hist_filter_inputs,
         outputs=[historico_display]
     )
-    
+
     # Clicar em uma linha do histórico para carregar
     historico_display.select(
         fn=carregar_post_do_historico,
@@ -1392,22 +1452,44 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot (Completo)"
         ],
         show_progress="minimal"
     )
-    
+
     # --- Eventos da Aba Analytics ---
-    
+
     gerar_relatorio_btn.click(
         fn=gerar_relatorio_analytics,
         inputs=None,
         outputs=[analytics_display]
     )
-    
+
     resetar_analytics_btn.click(
         fn=resetar_analytics,
         inputs=None,
-        outputs=[analytics_display]
+        outputs=[analytics_display],
+        js="() => confirm('Tem certeza que deseja resetar TODOS os dados de analytics e cache? Esta ação não pode ser desfeita.')"
     )
 
+# Definir a função resetar_analytics
+# Renomeada para evitar conflito com a função que será chamada pelo evento
+def resetar_analytics():
+    """Reseta os dados de analytics no Firestore e localmente."""
+    global analytics
+    analytics = {
+        "total_posts": 0,
+        "posts_por_nicho": {},
+        "posts_por_estilo": {},
+        "total_palavras": 0,
+        "total_imagens": 0,
+        "cache_hits": 0,
+        "cache_misses": 0,
+        "total_favoritos": 0
+    }
+    _salvar_analytics_firestore()
+    # Limpar cache local
+    for f in CACHE_DIR.glob('*'):
+        f.unlink()
+    print("Analytics e Cache resetados.")
+    return gerar_relatorio_analytics()
+
 # Lançar aplicação
 if __name__ == "__main__":
-    demo.launch()
-
+    demo.launch()