Update app.py

adicionado a nova aba de chatbot e atualizado os textos de apoio

app.py

@@ -1,6 +1,7 @@
-# GERADOR DE POSTS PARA REDES SOCIAIS
+# CHATBOT E GERADOR DE POSTS PARA REDES SOCIAIS
 # Aula 9: Versão Web com Hugging Face Spaces + Gradio
 # Versão com persistência de dados via Firebase Admin
+# ATUALIZADO: Adicionada funcionalidade de Chatbot
 
 import gradio as gr
 import requests
@@ -27,7 +28,7 @@ if not HUGGINGFACE_API_KEY:
 BASE_URL = "https://router.huggingface.co/v1"
 MODELO_TEXTO = "meta-llama/Llama-3.1-8B-Instruct"
 MODELO_IMAGEM = "black-forest-labs/FLUX.1-schnell"
-MODELO_TRADUCA = "Helsinki-NLP/opus-mt-pt-en" 
+MODELO_TRADUCA = "Helsinki-NLP/opus-mt-pt-en"
 
 # Headers para requisições
 headers = {
@@ -67,7 +68,7 @@ FORMATO_CONFIGS = {
 # Variável global para armazenar o histórico de sessão (agora é um cache)
 post_history = []
 # Variável para a instância do Firestore
-db = None 
+db = None
 
 
 # ============================================
@@ -197,7 +198,7 @@ def _formatar_historico_para_df(history_list):
     return formatted_list
 
 # ============================================
-# FUNÇÕES DE GERAÇÃO (Sem alterações)
+# FUNÇÕES DE GERAÇÃO (POSTS)
 # ============================================
 
 def gerar_texto(tema, nicho, estilo, formato):
@@ -247,7 +248,7 @@ Escreva apenas o conteúdo, sem introduções ou explicações."""
                 texto = resultado['choices'][0]['message']['content'].strip()
                 return texto
             else:
-                 return f"❌ Erro na resposta da API: Resposta vazia ou inesperada.\n{resultado}"
+                return f"❌ Erro na resposta da API: Resposta vazia ou inesperada.\n{resultado}"
         else:
             return f"❌ Erro ao gerar texto: {response.status_code}\n{response.text}"
 
@@ -307,13 +308,61 @@ def gerar_imagem(descricao):
         print(f"Erro ao gerar imagem: {str(e)}")
         return None
 
+# ============================================
+# NOVA FUNÇÃO DO CHATBOT
+# ============================================
+def responder_chat(message, chat_history):
+    """
+    Processa uma mensagem do usuário e retorna uma resposta do LLM para o chatbot.
+    """
+    if not HUGGINGFACE_API_KEY:
+        return "❌ Erro de Configuração: API Key não está definida."
+
+    url = f"{BASE_URL}/chat/completions"
+    
+    # 1. Definir o System Prompt para o assistente
+    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."
+    
+    # 2. Construir o histórico no formato da API
+    messages = [{"role": "system", "content": system_prompt}]
+    
+    for user_msg, bot_msg in chat_history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
+        
+    # 3. Adicionar a nova mensagem do usuário
+    messages.append({"role": "user", "content": message})
+
+    payload = {
+        "model": MODELO_TEXTO, # Reutiliza o mesmo modelo
+        "messages": messages,
+        "max_tokens": 1500, # Mais espaço para uma conversa fluida
+        "temperature": 0.7,
+        "stream": False # ChatInterface do Gradio lida melhor com respostas completas
+    }
+
+    try:
+        response = requests.post(url, headers=headers, json=payload, timeout=60)
+
+        if response.status_code == 200:
+            resultado = response.json()
+            if 'choices' in resultado and resultado['choices']:
+                texto = resultado['choices'][0]['message']['content'].strip()
+                return texto
+            else:
+                return f"❌ Erro na resposta da API: Resposta vazia ou inesperada.\n{resultado}"
+        else:
+            return f"❌ Erro ao gerar resposta: {response.status_code}\n{response.text}"
+
+    except Exception as e:
+        return f"❌ Erro de Conexão: {str(e)}"
 
 # ============================================
-# FUNÇÃO PRINCIPAL PARA O GRADIO
+# FUNÇÃO PRINCIPAL PARA O GERADOR DE POSTS
 # ============================================
 def gerar_post_interface(tema, nicho, estilo, formato, descricao_imagem, gerar_img, progress=gr.Progress()):
     """
-    Função principal que o Gradio chama, agora com barra de progresso e persistência.
+    Função principal que o Gradio chama para GERAR POSTS, com barra de progresso e persistência.
     """
     progress(0, desc="Iniciando validação...")
 
@@ -382,7 +431,7 @@ def gerar_post_interface(tema, nicho, estilo, formato, descricao_imagem, gerar_i
 
 
 # ============================================
-# INTERFACE GRADIO (Sem alterações)
+# INTERFACE GRADIO (Com Chatbot)
 # ============================================
 
 custom_theme = gr.themes.Glass(
@@ -393,11 +442,11 @@ custom_theme = gr.themes.Glass(
 )
 
 
-with gr.Blocks(theme=custom_theme, title="Gerador de Posts") as demo:
+with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot") as demo:
 
     gr.Markdown("""
-    # 🚀 Gerador de Posts para Redes Sociais
-    ### Powered by Hugging Face & Gradio
+    # 🚀 Gerador de Posts e Assistente de Mídias Sociais
+    ### Powered by Hugging Face, Gradio e Llama 3.1
     """)
     
     theme_mode = gr.Radio(
@@ -554,15 +603,40 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts") as demo:
                 queue=True # Necessário para gr.Progress
             )
         
+        # ============================================
+        # NOVA ABA: CHATBOT
+        # ============================================
+        with gr.TabItem("💬 Chatbot Assistente"):
+            gr.Markdown("### 🤖 Assistente Virtual")
+            gr.Markdown("Faça perguntas sobre mídias sociais, IA, peça ideias rápidas ou qualquer outro tópico.")
+            
+            gr.ChatInterface(
+                fn=responder_chat,
+                title="Assistente Virtual",
+                description="Converse com o Llama 3.1 para obter ajuda e insights.",
+                examples=[
+                    "O que é um 'gancho' para Instagram?", 
+                    "Me dê 3 ideias de post para um nicho de 'Fitness'", 
+                    "Como o Llama 3.1 funciona?",
+                    "Qual a diferença entre um post para Instagram e um para LinkedIn?"
+                ],
+                chatbot=gr.Chatbot(height=500), # Define uma altura
+                textbox=gr.Textbox(placeholder="Digite sua mensagem aqui...", scale=7),
+                submit_btn="Enviar",
+                clear_btn="Limpar Conversa",
+                retry_btn=None, # Oculta botões desnecessários
+                undo_btn=None,
+            )
+
         # ABA 2: Histórico (Exibe o Dataframe)
-        with gr.TabItem("📚 Histórico"):
+        with gr.TabItem("📚 Histórico de Posts"):
             gr.Markdown("### Posts Gerados Recentemente")
             
             # Dataframe de exibição que é atualizado pelo output do botão
             historico_display = gr.Dataframe(
                 headers=["Data/Hora", "Tema", "Nicho", "Estilo", "Formato", "Texto (Preview)", "Status"],
                 interactive=False,
-                value=_formatar_historico_para_df(carregar_historico_inicial()), # Carrega histórico (DB ou vazio) E FORMATA                
+                value=_formatar_historico_para_df(carregar_historico_inicial()), # Carrega histórico (DB ou vazio) E FORMATA
             )
             # Garantir que o historico_display seja atualizado sempre que o historico_dataframe (invisível) for atualizado
             historico_dataframe.change(lambda x: x, historico_dataframe, historico_display)
@@ -571,29 +645,25 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts") as demo:
             ---
             *Nota sobre Persistência:*
             
-            Esta aba agora exibe o histórico persistente salvo no Google Firestore.
+            Esta aba agora exibe o histórico persistente (dos **posts gerados**) salvo no Google Firestore.
             O histórico é compartilhado entre todas as sessões e usuários.
             """)
 
         # ABA 3: Configurações
         with gr.TabItem("⚙️ Configurações"):
             gr.Markdown("### Configurações do Gerador")
-            gr.Markdown("**Modelo de Texto:** Llama 3.1 8B")
+            gr.Markdown("**Modelo de Texto (LLM):** Llama 3.1 8B (Usado para Posts e Chatbot)")
             gr.Markdown("**Modelo de Imagem:** FLUX.1 Schnell")
             gr.Markdown("**Modelo de Tradução (PT -> EN):** Helsinki-NLP/opus-mt-pt-en")
             gr.Markdown("**API Provider:** Hugging Face Inference")
             gr.Markdown("**Database:** Google Firestore (via Firebase Admin)")
             gr.Markdown("---")
-            gr.Markdown("#### Nichos Adicionais:")
-            gr.Markdown("- Finanças Pessoais e Investimentos")
-            gr.Markdown("- Cultura Pop e Entretenimento")
-            gr.Markdown("- Meio Ambiente e Sustentabilidade")
-            gr.Markdown("#### Estilos Adicionais:")
-            gr.Markdown("- Controverso e de Debate")
-            gr.Markdown("- Curiosidades (Fatos Rápidos)")
-            gr.Markdown("- Tutorial/Passo a Passo")
+            gr.Markdown("#### Funcionalidades:")
+            gr.Markdown("- **Gerador de Posts:** Cria posts completos com texto e imagem.")
+            gr.Markdown("- **Chatbot Assistente:** Converse com a IA para ideias e perguntas rápidas.")
+            gr.Markdown("- **Histórico Persistente:** Salva os *posts gerados* no Firestore.")
             gr.Markdown("---")
-            gr.Markdown("#### Formatos de Saída:")
+            gr.Markdown("#### Formatos de Saída (Gerador de Posts):")
             gr.Markdown("- Instagram (Post): 100-150 palavras")
             gr.Markdown("- Twitter/X (Curto): Até 280 caracteres")
             gr.Markdown("- LinkedIn (Artigo): 250-400 palavras")
@@ -609,16 +679,16 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts") as demo:
             **Tecnologias:**
             - Hugging Face Spaces (hospedagem)
             - Gradio (interface web)
-            - Llama 3.1 8B (geração de texto)
+            - **Llama 3.1 8B (geração de texto e chatbot)**
             - FLUX.1 (geração de imagens)
             - Opus-MT (tradução)
             - **Firebase Firestore (Banco de Dados)**
 
             **Como funciona:**
-            1. Você escolhe nicho, tema e **formato de saída**.
-            2. O sistema ajusta o prompt do LLM ao formato escolhido.
-            3. Se a imagem estiver marcada, a descrição é **automaticamente traduzida para Inglês**.
-            4. Os resultados são exibidos e o post é salvo permanentemente no **Firestore**, aparecendo na aba "Histórico".
+            1. **Gerar Post:** Você escolhe nicho, tema e **formato de saída**.
+            2. **Chatbot:** Você pode conversar diretamente com a IA na aba 'Chatbot Assistente' para tirar dúvidas.
+            3. Se a imagem estiver marcada (no Gerador), a descrição é **automaticamente traduzida para Inglês**.
+            4. Os resultados são exibidos e o post é salvo permanentemente no **Firestore**, aparecendo na aba "Histórico de Posts".
 
             **Desenvolvido por:** Wilder Paz
             """)
@@ -626,11 +696,9 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts") as demo:
     # Footer
     gr.Markdown("""
     ---
-    **Curso de Python com IA** | 🤖 Powered by Llama 3.1 & FLUX | ⚡ Hugging Face Spaces + Gradio
+    **Curso de Python com IA** | 🤖 Powered by Llama 3.1 & FLUX | ⚡ Hugging Face Spaces + Gradio + Firestore
     """)
 
 # Lançar aplicação
 if __name__ == "__main__":
     demo.launch()
-
-