Update app.py

nova lista de estilos criativos (Studio Ghibli, Pixar, etc. e tradutor de erros da API para que as mensagens de falha (como 402, 503, etc.

app.py

@@ -84,24 +84,26 @@ ESTILOS_DISPONIVEIS = [
     "Tutorial/Passo a Passo",
 ]
 
-# NOVO: Dicionário de estilos de imagem (expandido)
-estilos_por_nicho = {
-    "Nenhum (Automático)": "standard photography, high quality",
-    "Fitness e Vida Saudável": "athletic, energetic, dynamic pose, sports photography, high contrast",
-    "Alimentação e Nutrição": "appetizing, fresh ingredients, food photography, professional styling, macro shot",
-    "Motivação e Desenvolvimento Pessoal": "inspiring, uplifting atmosphere, professional portrait, cinematic lighting",
-    "Negócios e Empreendedorismo": "professional, modern office, business casual, corporate photography, sleek",
-    "Viagens e Turismo": "scenic landscape, travel photography, vibrant destination, wanderlust, breathtaking view",
-    "Tecnologia e Inovação": "modern tech, sleek design, futuristic, innovation, clean aesthetic, glowing blue lights",
-    "Finanças Pessoais e Investimentos": "secure, wealthy, financial growth, data visualization, professional",
-    "Cultura Pop e Entretenimento": "vibrant, pop art, iconic, dynamic, colorful, entertainment photography",
-    "Meio Ambiente e Sustentabilidade": "lush greens, natural lighting, eco-friendly, sustainable, earth tones",
+# NOVO: Dicionário de estilos de imagem (baseado na Solicitação 2)
+ESTILOS_DE_IMAGEM = {
+    "Nenhum (Automático)": "standard photography, high quality, 4k",
+    "Fotografia Vintage": "vintage photography, retro style, film grain, analog",
+    "Quente (Vintage)": "warm tones, vintage filter, retro, analog film look",
+    "Frio (Moderno)": "cool tones, modern aesthetic, clean, desaturated blues",
+    "Estilo Studio Ghibli": "Studio Ghibli style, poetic, soft, pastel colors, magical atmosphere, nostalgic portrait, fantasy scene",
+    "Estilo Simpsons": "Simpsons style, iconic visual, strong black outlines, solid colors, humorous daily scene",
+    "Estilo Pixar": "Pixar style, 3D digital animation, striking expressions, friendly characters",
+    "Estilo Tim Burton": "Tim Burton style, dark aesthetic, thin lines, gothic environment, mysterious characters",
+    "Estilo Attack on Titan": "Attack on Titan anime style, intense and dark lines, action scene",
+    "Estilo RPG Clássico": "Classic RPG style, epic aesthetic, fantasy book cover",
+    "Estilo 8-bit e 16-bit": "8-bit and 16-bit retro visual, nostalgic games",
+    "Estilo Animação Anos 2000": "2000s animation style (Samurai Jack inspired), modern and stylized",
     "Arte Digital (Cinemático)": "cinematic, dramatic lighting, fantasy art, concept art",
     "Arte Digital (Neon)": "neonoir, cyberpunk, glowing lights, futuristic city",
     "Minimalista": "minimalist, clean background, simple, elegant",
-    "Fotografia Vintage": "vintage photography, retro style, film grain, analog"
 }
 
+
 # NOVO: Dicionário de filtros de imagem
 FILTROS_IMAGEM = {
     "Nenhum": "",
@@ -114,6 +116,7 @@ FILTROS_IMAGEM = {
 
 
 # CORRIGIDO: O dicionário do LinkedIn estava quebrado
+# ATUALIZADO: Funcionalidade de formato reativada (Solicitação 1)
 FORMATO_CONFIGS = {
     "Instagram (Post)": {"tamanho": "100-150 palavras", "estrutura": "gancho inicial + desenvolvimento + call-to-action", "tom_adicional": "próximo, empático e motivador", "max_tokens": 350, "limite_palavras_ia": "150 palavras", "hashtags": "Incluir 4-5 hashtags relevantes no final. Incluir no máximo 3 emojis relevantes no texto."},
     "Twitter/X (Curto)": {"tamanho": "Até 280 caracteres", "estrutura": "frase de impacto + link/hashtag", "tom_adicional": "direto e conciso, ideal para tweets", "max_tokens": 150, "limite_palavras_ia": "280 caracteres", "hashtags": "Incluir no máximo 2 hashtags."},
@@ -284,7 +287,7 @@ def copiar_feedback(texto):
     return criar_alerta('warning', '⚠️ Nada para copiar')
 
 def limpar_tudo():
-    # Retorna 10 valores para limpar todos os campos
+    # Retorna 11 valores para limpar todos os campos
     return (
         "", # texto_output
         None, # imagem_output
@@ -292,6 +295,7 @@ def limpar_tudo():
         0, # palavras_output
         0, # caracteres_output
         0, # hashtags_output
+        list(FORMATO_CONFIGS.keys())[0], # formato_input (Solicitação 1)
         "Nenhum (Automático)", # estilo_img_input
         "Balanceada", # qualidade_img_input
         "Nenhum", # filtro_img_input
@@ -305,19 +309,47 @@ def recarregar_e_formatar_historico():
     # através da chamada a 'atualizar_historico'
     return _formatar_historico_para_df(post_history)
 
+# ============================================
+# NOVA FUNÇÃO: INTERPRETAR ERROS (Solicitação 3)
+# ============================================
+def interpretar_erro_api(erro_str):
+    """Interpreta erros comuns da API para o usuário em Português."""
+    erro_str_lower = erro_str.lower()
+    print(f"Interpretando erro: {erro_str}")
+
+    if "402" in erro_str_lower or "payment required" in erro_str_lower or "exceeded your monthly included credits" in erro_str_lower:
+        return ("Erro 402: Limite de créditos excedido. Você excedeu seus créditos mensais da API do Hugging Face. "
+                "Considere assinar o plano PRO para mais créditos.")
+    
+    if "503" in erro_str_lower or "model is loading" in erro_str_lower or "service temporarily unavailable" in erro_str_lower:
+        return ("Erro 503: Modelo indisponível. O modelo está carregando ou temporariamente indisponível. "
+                "Por favor, tente novamente em alguns segundos.")
+
+    if "429" in erro_str_lower or "too many requests" in erro_str_lower:
+        return ("Erro 429: Muitas requisições. O limite de taxa foi atingido. "
+                "Por favor, aguarde um momento e tente novamente.")
+
+    if "timeout" in erro_str_lower or "timed out" in erro_str_lower:
+        return ("Erro de Timeout: A conexão expirou. O modelo demorou muito para responder. "
+                "Tente novamente.")
+    
+    if "authorization" in erro_str_lower or "401" in erro_str_lower:
+            return ("Erro 401: Autenticação falhou. A Chave da API (Secret 'Capoeira') pode estar inválida ou ausente.")
+
+    # Fallback para erros genéricos
+    return f"Erro inesperado: {erro_str[:200]}..." # Trunca o erro
 
 # ============================================
 # FUNÇÕES DE GERAÇÃO (POSTS)
 # ============================================
 
-def gerar_texto(tema, nicho, estilo):
+def gerar_texto(tema, nicho, estilo, formato): # ATUALIZADO: 'formato' re-adicionado
     """
     Gera texto usando API do Hugging Face com base no formato escolhido.
-    NOTA: 'formato' foi hardcoded para "Instagram (Post)" para
-    suportar a nova função gerar_post_interface.
     """
     
-    formato = "Instagram (Post)" # Hardcoded pois o novo fluxo removeu o input de formato
+    # ATUALIZADO: Linha hardcoded removida
+    # formato = "Instagram (Post)" 
 
     if not HUGGINGFACE_API_KEY:
         return "❌ Erro de Configuração: API Key não está definida."
@@ -363,10 +395,12 @@ Escreva apenas o conteúdo, sem introduções ou explicações."""
             else:
                 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}"
+            # ATUALIZADO: Interpretar erro
+            return f"❌ {interpretar_erro_api(f'Erro {response.status_code}: {response.text}')}"
 
     except Exception as e:
-        return f"❌ Erro de Conexão: {str(e)}"
+        # ATUALIZADO: Interpretar erro
+        return f"❌ {interpretar_erro_api(str(e))}"
 
 def traduzir_texto(texto_pt):
     """Traduz texto de Português (PT) para Inglês (EN) usando API do Hugging Face.
@@ -395,6 +429,7 @@ def traduzir_texto(texto_pt):
             return texto_pt # Fallback
 
     except Exception as e:
+        print(f"Falha na tradução (fallback para PT): {e}")
         return texto_pt # Fallback
 
 # ============================================
@@ -405,8 +440,8 @@ def otimizar_prompt_imagem(descricao_pt, estilo_escolhido, filtro_escolhido):
     """Combina as escolhas do usuário em um prompt otimizado (em Português)."""
     
     # 1. Obter estilo
-    # Usa o estilo escolhido, com fallback para "Nenhum"
-    estilo = estilos_por_nicho.get(estilo_escolhido, estilos_por_nicho["Nenhum (Automático)"])
+    # ATUALIZADO: Usar o novo dicionário ESTILOS_DE_IMAGEM (Solicitação 2)
+    estilo = ESTILOS_DE_IMAGEM.get(estilo_escolhido, ESTILOS_DE_IMAGEM["Nenhum (Automático)"])
     
     # 2. Obter filtro
     filtro = FILTROS_IMAGEM.get(filtro_escolhido, FILTROS_IMAGEM["Nenhum"])
@@ -486,7 +521,8 @@ def gerar_imagem_robusta(descricao_pt, estilo_escolhido, qualidade, filtro_escol
                 continue
             else:
                 # Último modelo falhou, retornar erro
-                mensagem = f"❌ Todos os modelos de imagem falharam. (Erro: {str(e)})"
+                # ATUALIZADO: Interpretar erro
+                mensagem = f"❌ {interpretar_erro_api(str(e))}"
                 return (None, mensagem)
 
     # Se chegou aqui, algo deu errado
@@ -537,10 +573,12 @@ def responder_chat(message, chat_history):
             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}"
+            # ATUALIZADO: Interpretar erro
+            return f"❌ {interpretar_erro_api(f'Erro {response.status_code}: {response.text}')}"
 
     except Exception as e:
-        return f"❌ Erro de Conexão: {str(e)}"
+        # ATUALIZADO: Interpretar erro
+        return f"❌ {interpretar_erro_api(str(e))}"
 
 # ============================================
 # NOVAS FUNÇÕES DE DOWNLOAD
@@ -647,7 +685,8 @@ def preparar_download(texto, imagem_pil, tema):
 # FUNÇÃO PRINCIPAL COM PROGRESSO (ATUALIZADA)
 # ============================================
 
-def gerar_post_interface(tema, nicho, estilo, descricao_imagem, gerar_img, 
+def gerar_post_interface(tema, nicho, estilo, formato, # ATUALIZADO: 'formato' adicionado
+                         descricao_imagem, gerar_img, 
                          estilo_img_input, qualidade_img_input, filtro_img_input, 
                          progress=gr.Progress()):
     """Gera post completo com feedback e lógica de imagem avançada"""
@@ -670,10 +709,11 @@ def gerar_post_interface(tema, nicho, estilo, descricao_imagem, gerar_img,
     
     # Gerando texto
     progress(0.3, desc="🤖 Gerando texto (Llama 3.1)...")
-    texto = gerar_texto(tema, nicho, estilo) 
+    # ATUALIZADO: 'formato' passado para gerar_texto
+    texto = gerar_texto(tema, nicho, estilo, formato) 
     
-    if texto.startswith("❌") or texto.startswith("⏳"):
-        status_final = criar_alerta('error', f'Erro no texto: {texto}')
+    if texto.startswith("❌"): # Erro já vem formatado
+        status_final = criar_alerta('error', f'{texto}')
         return (texto, None, status_final, 0, 0, 0)
     
     progress(0.5, desc="✅ Texto pronto!")
@@ -699,6 +739,7 @@ def gerar_post_interface(tema, nicho, estilo, descricao_imagem, gerar_img,
             status_final = criar_alerta('success', f'🎉 Post completo gerado! ({status_imagem})')
         else:
             # status_imagem conterá a mensagem de erro (ex: "Todos os modelos falharam")
+            # ATUALIZADO: status_imagem já vem formatado com o erro
             status_final = criar_alerta('warning', f'✅ Texto OK, mas imagem falhou: {status_imagem}')
     else:
         progress(0.7, desc="⏭️ Pulando geração de imagem...")
@@ -720,9 +761,8 @@ def gerar_post_interface(tema, nicho, estilo, descricao_imagem, gerar_img,
     # INÍCIO DA CORREÇÃO: Re-adicionando salvamento no histórico
     # ==================================================
     
-    # A variável 'formato' foi hardcoded como "Instagram (Post)" na função gerar_texto
-    # Vamos usar o mesmo valor aqui para consistência.
-    formato_usado = "Instagram (Post)" 
+    # ATUALIZADO: Usar o 'formato' vindo do input
+    formato_usado = formato
     
     # Usar a variável 'status' que já foi definida
     status_final_historico = status_final # Usar o status HTML (o Firestore lida bem com strings)
@@ -735,7 +775,7 @@ def gerar_post_interface(tema, nicho, estilo, descricao_imagem, gerar_img,
         "Tema": tema,
         "Nicho": nicho,
         "Estilo": estilo,
-        "Formato": formato_usado,
+        "Formato": formato_usado, # Atualizado
         "Texto (Preview)": texto_preview,
         "Status": status_imagem or "Texto Gerado", # Salva o status da imagem (ex: "Gerada com FLUX")
     }
@@ -796,13 +836,13 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot") as demo:
                         placeholder="Ex: Transforme seu corpo, transforme sua vida"
                     )
 
-                    # Ocultado, pois a lógica agora foca em "Instagram (Post)"
+                    # ATUALIZADO: Reativado (Solicitação 1)
                     formato_input = gr.Radio(
                         choices=list(FORMATO_CONFIGS.keys()),
-                        label="Escolha o Formato de Saída (Desativado)",
+                        label="Escolha o Formato de Saída",
                         value=list(FORMATO_CONFIGS.keys())[0],
-                        interactive=False, # Desativado
-                        visible=False # Oculto
+                        interactive=True,
+                        visible=True
                     )
         
                     gr.Markdown("### 🎨 2. Configurações da Imagem (Opcional)")
@@ -819,9 +859,10 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot") as demo:
                         visible=False
                     )
                     
+                    # ATUALIZADO: Usando ESTILOS_DE_IMAGEM (Solicitação 2)
                     estilo_img_input = gr.Dropdown(
                         label="Estilo da Imagem",
-                        choices=list(estilos_por_nicho.keys()),
+                        choices=list(ESTILOS_DE_IMAGEM.keys()),
                         value="Nenhum (Automático)",
                         visible=False,
                         interactive=True
@@ -909,14 +950,15 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot") as demo:
         
             gr.Markdown("### 💡 Experimente estes exemplos:")
         
+            # ATUALIZADO: Exemplos agora incluem o formato
             gr.Examples(
                 examples=[
-                    [NICHOS_DISPONIVEIS[2], ESTILOS_DISPONIVEIS[0], "Frases marcantes de pessoas importantes"],
-                    [NICHOS_DISPONIVEIS[1], ESTILOS_DISPONIVEIS[2], "Receita rápida de smoothie verde"],
-                    [NICHOS_DISPONIVEIS[5], ESTILOS_DISPONIVEIS[3], "O futuro da IA em 2025"],
-                    [NICHOS_DISPONIVEIS[4], ESTILOS_DISPONIVEIS[1], "Melhores destinos para lua de mel na Europa"],
+                    [NICHOS_DISPONIVEIS[2], ESTILOS_DISPONIVEIS[0], "Frases marcantes de pessoas importantes", "Instagram (Post)"],
+                    [NICHOS_DISPONIVEIS[1], ESTILOS_DISPONIVEIS[2], "Receita rápida de smoothie verde", "Instagram (Post)"],
+                    [NICHOS_DISPONIVEIS[5], ESTILOS_DISPONIVEIS[3], "O futuro da IA em 2025", "LinkedIn (Artigo)"],
+                    [NICHOS_DISPONIVEIS[4], ESTILOS_DISPONIVEIS[1], "Melhores destinos para lua de mel na Europa", "Twitter/X (Curto)"],
                 ],
-                inputs=[nicho_input, estilo_input, tema_input], 
+                inputs=[nicho_input, estilo_input, tema_input, formato_input], # 'formato_input' adicionado
                 outputs=[texto_output, imagem_output, status_output]
             )
 
@@ -930,6 +972,7 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot") as demo:
                 fn=gerar_post_interface,
                 inputs=[
                     tema_input, nicho_input, estilo_input, 
+                    formato_input, # ATUALIZADO: 'formato_input' adicionado
                     descricao_img_input, gerar_img_checkbox,
                     estilo_img_input, qualidade_img_input, filtro_img_input # Novos inputs
                 ],
@@ -954,6 +997,7 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot") as demo:
                 outputs=[
                     texto_output, imagem_output, status_output, 
                     palavras_output, caracteres_output, hashtags_output,
+                    formato_input, # ATUALIZADO: 'formato_input' adicionado
                     estilo_img_input, qualidade_img_input, filtro_img_input, # Limpa novos inputs
                     download_output # Limpa o arquivo de download
                 ]
@@ -992,7 +1036,11 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot") as demo:
                 ],
                 chatbot=chatbot_para_interface, # Passando o componente chatbot customizado
                 textbox=gr.Textbox(placeholder="Digite sua mensagem aqui...", scale=7),
-                submit_btn="Enviar"                
+                submit_btn="Enviar",
+                # CORREÇÃO: Botões removidos conforme solicitado (Solicitação 7)
+                clear_btn=None,
+                retry_btn=None, 
+                undo_btn=None,
             )
             
         # ABA 2: Histórico (Exibe o Dataframe)
@@ -1036,6 +1084,7 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot") as demo:
             gr.Markdown("---")
             gr.Markdown("#### Funcionalidades:")
             gr.Markdown("- **Gerador de Posts:** Cria posts completos com texto e imagem.")
+            gr.Markdown("- **Seleção de Formato:** Permite escolher o formato do texto (Instagram, Twitter, LinkedIn).")
             gr.Markdown("- **Controles Avançados:** Permite seleção de Estilo, Qualidade e Filtros para a imagem.")
             gr.Markdown("- **Download de Post:** Combina texto e imagem em um único arquivo PNG para download.")
             gr.Markdown("- **Chatbot Assistente:** Converse com a IA para ideias e perguntas rápidas.")
@@ -1059,7 +1108,7 @@ with gr.Blocks(theme=custom_theme, title="Gerador de Posts e Chatbot") as demo:
             - **PIL (Python Imaging Library) (para composição de posts)**
 
             **Como funciona:**
-            1. **Gerar Post:** Você define o tema, nicho e estilo do *texto*.
+            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.
             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.