Upload app.py

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 from livro import processar_e_mostrar, verificar_contraste, gerar_preview_kdp
 
@@ -21,12 +22,17 @@ with gr.Blocks() as demo:
 
     def gerar(imagem_array, mask_array, idade):
         global resultado
+        if imagem_array is None:
+            return "⚠️ Nenhuma imagem enviada.", None
         aviso = verificar_contraste(imagem_array)
-        resultado = processar_e_mostrar(imagem_array, mask_array, idade)
-        return aviso, resultado["preview"]
+        try:
+            resultado = processar_e_mostrar(imagem_array, mask_array, idade)
+            return aviso, resultado["preview"]
+        except Exception as e:
+            return f"Erro: {str(e)}", None
 
     def salvar_pdf():
-        return resultado.get("pdf")
+        return resultado.get("pdf") if resultado else None
 
     def gerar_capa():
         return gerar_preview_kdp()