Update app.py

Mudado o perfil "Rápida" para usar steps: 10 (que é um valor seguro e rápido para o FLUX.1-schnell) e adicionado uma lógica de "grampo" (clamp) dentro da função gerar_imagem_robusta. Agora, antes de chamar a API, o código verifica se o modelo é o FLUX.1-schnell e, em caso afirmativo, força o número de passos a ser no máximo 16, mesmo que o perfil "Balanceada" (25) ou "Alta" (30) tenha sido selecionado

app.py

@@ -795,19 +795,19 @@ def gerar_imagem_robusta(descricao_pt, estilo_escolhido, qualidade, filtro_escol
     Retorna: (PIL.Image, str_mensagem_status)
     """
     
-    # 1. Configs de Qualidade
+    # 1. Configs de Qualidade (CORRIGIDO)
     configs_qualidade = {
         "Rápida": {
             "modelos": [MODELOS_IMAGEM[0]],  # Só FLUX-schnell
-            "steps": 20
+            "steps": 10 # CORRIGIDO: Era 20, o máximo é 16. 10 é seguro e rápido.
         },
         "Balanceada": {
             "modelos": MODELOS_IMAGEM[:2],  # FLUX schnell + dev
-            "steps": 25
+            "steps": 25 # Mantém 25 (para o FLUX.1-dev)
         },
         "Alta": {
-            "modelos": MODELOS_IMAGEM,  # Todos os 3, em ordem
-            "steps": 30
+            "modelos": MODELOS_IMAGEM,  # Todos os 3
+            "steps": 30 # Mantém 30 (para SDXL e FLUX.1-dev)
         }
     }
     config = configs_qualidade.get(qualidade, configs_qualidade["Balanceada"])
@@ -827,13 +827,26 @@ def gerar_imagem_robusta(descricao_pt, estilo_escolhido, qualidade, filtro_escol
             
             print(f"Tentando gerar imagem com {modelo_config['nome']}...")
 
+            # --- INÍCIO DA CORREÇÃO ---
+            # Pega o 'steps' do perfil de qualidade
+            steps_para_usar = config['steps'] 
+            
+            # FLUX.1-schnell tem um limite MÁXIMO de 16 steps.
+            # Se este modelo for tentado, devemos forçar os steps para um valor seguro.
+            if modelo_config['id'] == "black-forest-labs/FLUX.1-schnell":
+                # Se o perfil for "Rápida", ele já é 10.
+                # Se for "Balanceada" (25) ou "Alta" (30), reduzimos para 16 (limite máx).
+                steps_para_usar = min(config['steps'], 16)
+                print(f"Ajustando steps para {steps_para_usar} para o modelo FLUX.1-schnell.")
+            # --- FIM DA CORREÇÃO ---
+
             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']
+                num_inference_steps=steps_para_usar # CORRIGIDO
             )
             
             print(f"✅ Imagem gerada com {modelo_config['nome']}")