Update app.py

app.py

@@ -1,45 +1,45 @@
 """
 🎨 Basic Image Adjustments
-App 2 do Photoshop AI Ecosystem - Ajustes básicos de imagem
+Ajustes básicos de imagem
+Otimizado para Hugging Face Spaces
 """
+
 import gradio as gr
 from PIL import Image, ImageEnhance, ImageOps
 import tempfile
 import numpy as np
 
+# ========== FUNÇÕES PRINCIPAIS ==========
+
 def process_image(image, brightness=1.0, contrast=1.0, saturation=1.0, sharpness=1.0, auto_contrast=False):
     """Processa imagem com ajustes básicos"""
     if image is None:
         return None
     
     try:
-        # Converter para PIL Image se for numpy array
+        # Converter para PIL Image
         if isinstance(image, np.ndarray):
-            img = Image.fromarray(image.astype('uint8'), 'RGB')
+            img = Image.fromarray(image)
         else:
             img = image
         
-        # Aplicar contraste automático se solicitado
+        # Aplicar contraste automático
         if auto_contrast:
             img = ImageOps.autocontrast(img, cutoff=2)
         
-        # Aplicar ajustes na ordem correta
-        # 1. Brilho
+        # Aplicar ajustes
         if brightness != 1.0:
             enhancer = ImageEnhance.Brightness(img)
             img = enhancer.enhance(brightness)
         
-        # 2. Contraste
         if contrast != 1.0:
             enhancer = ImageEnhance.Contrast(img)
             img = enhancer.enhance(contrast)
         
-        # 3. Saturação
         if saturation != 1.0:
             enhancer = ImageEnhance.Color(img)
             img = enhancer.enhance(saturation)
         
-        # 4. Nitidez
         if sharpness != 1.0:
             enhancer = ImageEnhance.Sharpness(img)
             img = enhancer.enhance(sharpness)
@@ -47,177 +47,121 @@ def process_image(image, brightness=1.0, contrast=1.0, saturation=1.0, sharpness
         return img
     
     except Exception as e:
-        print(f"Erro no processamento: {e}")
+        print(f"Erro: {e}")
         return image
 
-def get_preset(preset_name):
+def get_preset_values(preset_name):
     """Retorna valores de preset"""
     presets = {
-        'vibrant': [1.2, 1.3, 1.4, 1.1, False],
-        'soft': [1.05, 0.9, 0.8, 0.9, False],
-        'dramatic': [0.9, 1.4, 0.9, 1.3, False],
-        'retro': [1.0, 1.1, 0.6, 1.0, False],
-        'bw': [1.0, 1.2, 0.0, 1.1, False],
-        'reset': [1.0, 1.0, 1.0, 1.0, False]
+        'vibrant': {'brightness': 1.2, 'contrast': 1.3, 'saturation': 1.4, 'sharpness': 1.1, 'auto_contrast': False},
+        'soft': {'brightness': 1.05, 'contrast': 0.9, 'saturation': 0.8, 'sharpness': 0.9, 'auto_contrast': False},
+        'dramatic': {'brightness': 0.9, 'contrast': 1.4, 'saturation': 0.9, 'sharpness': 1.3, 'auto_contrast': False},
+        'retro': {'brightness': 1.0, 'contrast': 1.1, 'saturation': 0.6, 'sharpness': 1.0, 'auto_contrast': False},
+        'bw': {'brightness': 1.0, 'contrast': 1.2, 'saturation': 0.0, 'sharpness': 1.1, 'auto_contrast': False},
+        'reset': {'brightness': 1.0, 'contrast': 1.0, 'saturation': 1.0, 'sharpness': 1.0, 'auto_contrast': False}
     }
-    return presets.get(preset_name, [1.0, 1.0, 1.0, 1.0, False])
+    return presets.get(preset_name, presets['reset'])
 
-def save_temp_image(image):
-    """Salva imagem em arquivo temporário para download"""
+def save_image_for_download(image):
+    """Salva imagem para download"""
     if image is None:
         return None
     
-    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png", prefix="adjusted_")
-    image.save(temp_file.name, "PNG", optimize=True)
-    return temp_file.name
+    try:
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png", prefix="adjusted_")
+        image.save(temp_file.name, "PNG", optimize=True)
+        return temp_file.name
+    except:
+        return None
 
-# Interface Gradio
-with gr.Blocks(
-    title="🎨 Basic Image Adjustments",
-    theme=gr.themes.Soft(),
-    css="""
-    .slider-container { margin: 15px 0; }
-    .preset-btn { margin: 5px; width: 100%; }
+# ========== INTERFACE GRADIO ==========
+
+# Criar interface sem parâmetros no construtor (compatível com Gradio 6+)
+with gr.Blocks(title="🎨 Basic Image Adjustments") as demo:
+    
+    # CSS customizado
+    demo.css = """
     .header { text-align: center; padding: 20px; }
-    .image-container { border: 2px solid #e0e0e0; border-radius: 10px; padding: 10px; }
+    .slider-label { font-weight: bold; margin-bottom: 5px; }
+    .preset-grid { display: grid; grid-template-columns: repeat(3, 1fr); gap: 10px; }
+    .image-box { border: 1px solid #ddd; border-radius: 8px; padding: 10px; }
+    .download-btn { width: 100%; margin-top: 20px; }
     """
-) as demo:
     
     # Cabeçalho
     gr.HTML("""
     <div class="header">
-        <h1>🎨 Basic Image Adjustments</h1>
-        <p><strong>App 2 do Photoshop AI Ecosystem</strong></p>
-        <p>Ajuste brilho, contraste, saturação e nitidez de suas imagens</p>
+        <h1 style="color: #4a6fa5;">🎨 Basic Image Adjustments</h1>
+        <p style="font-size: 1.1em;"><strong>App 2 do Photoshop AI Ecosystem</strong></p>
+        <p style="color: #666;">Ajuste brilho, contraste, saturação e nitidez</p>
     </div>
     """)
     
     with gr.Row():
-        # Coluna da esquerda - Upload e Controles
+        # Coluna 1: Upload
         with gr.Column(scale=1):
             gr.Markdown("### 📤 Upload da Imagem")
+            image_input = gr.Image(type="pil", label="Clique ou arraste uma imagem", height=150)
             
-            image_input = gr.Image(
-                label="Arraste ou clique para selecionar",
-                type="pil",
-                height=200
-            )
-            
-            gr.Markdown("### ⚙️ Controles de Ajuste")
+            gr.Markdown("### ⚙️ Controles")
             
-            brightness = gr.Slider(
-                minimum=0.0,
-                maximum=2.0,
-                value=1.0,
-                step=0.1,
-                label="☀️ Brilho",
-                info="0.0 = Escuro | 1.0 = Normal | 2.0 = Brilhante"
-            )
+            brightness = gr.Slider(0.0, 2.0, 1.0, step=0.1, label="☀️ Brilho")
+            contrast = gr.Slider(0.0, 2.0, 1.0, step=0.1, label="🎭 Contraste")
+            saturation = gr.Slider(0.0, 2.0, 1.0, step=0.1, label="🌈 Saturação")
+            sharpness = gr.Slider(0.0, 2.0, 1.0, step=0.1, label="🔍 Nitidez")
             
-            contrast = gr.Slider(
-                minimum=0.0,
-                maximum=2.0,
-                value=1.0,
-                step=0.1,
-                label="🎭 Contraste",
-                info="0.0 = Baixo | 1.0 = Normal | 2.0 = Alto"
-            )
-            
-            saturation = gr.Slider(
-                minimum=0.0,
-                maximum=2.0,
-                value=1.0,
-                step=0.1,
-                label="🌈 Saturação",
-                info="0.0 = P&B | 1.0 = Normal | 2.0 = Intenso"
-            )
-            
-            sharpness = gr.Slider(
-                minimum=0.0,
-                maximum=2.0,
-                value=1.0,
-                step=0.1,
-                label="🔍 Nitidez",
-                info="0.0 = Borrado | 1.0 = Normal | 2.0 = Nítido"
-            )
-            
-            auto_contrast = gr.Checkbox(
-                label="⚡ Contraste Automático",
-                value=False,
-                info="Otimiza automaticamente o contraste"
-            )
-            
-        # Coluna da direita - Presets e Resultado
+            auto_contrast = gr.Checkbox(label="⚡ Contraste Automático", value=False)
+        
+        # Coluna 2: Presets
         with gr.Column(scale=1):
             gr.Markdown("### 🎨 Presets Rápidos")
             
             with gr.Row():
-                vibrant_btn = gr.Button("Vibrante", variant="secondary", size="sm", min_width=100)
-                soft_btn = gr.Button("Suave", variant="secondary", size="sm", min_width=100)
-            
-            with gr.Row():
-                dramatic_btn = gr.Button("Dramático", variant="secondary", size="sm", min_width=100)
-                retro_btn = gr.Button("Retrô", variant="secondary", size="sm", min_width=100)
+                vibrant_btn = gr.Button("Vibrante", variant="secondary")
+                soft_btn = gr.Button("Suave", variant="secondary")
+                dramatic_btn = gr.Button("Dramático", variant="secondary")
             
             with gr.Row():
-                bw_btn = gr.Button("Preto & Branco", variant="secondary", size="sm", min_width=100)
-                reset_btn = gr.Button("🔄 Resetar", variant="stop", size="sm", min_width=100)
+                retro_btn = gr.Button("Retrô", variant="secondary")
+                bw_btn = gr.Button("P&B", variant="secondary")
+                reset_btn = gr.Button("Resetar", variant="stop")
     
     # Visualização
     with gr.Row():
         with gr.Column():
             gr.Markdown("### 👁️ Original")
-            original_display = gr.Image(
-                label="Imagem Original",
-                type="pil",
-                interactive=False,
-                height=300
-            )
+            original_display = gr.Image(type="pil", interactive=False, height=250, label="Original")
         
         with gr.Column():
             gr.Markdown("### ✨ Ajustada")
-            processed_display = gr.Image(
-                label="Imagem Ajustada",
-                type="pil",
-                interactive=False,
-                height=300
-            )
+            processed_display = gr.Image(type="pil", interactive=False, height=250, label="Ajustada")
     
-    # Exportação
+    # Download
     with gr.Row():
         with gr.Column():
-            gr.Markdown("### 💾 Exportar Imagem")
-            
-            export_btn = gr.Button(
-                "📥 Download Imagem Ajustada",
-                variant="primary",
-                size="lg"
-            )
-            
-            download_output = gr.File(
-                label="Faça download da imagem processada",
-                interactive=False
-            )
+            gr.Markdown("### 💾 Download")
+            download_btn = gr.Button("📥 Baixar Imagem Ajustada", variant="primary", size="lg")
+            download_file = gr.File(label="Arquivo para download", interactive=False)
     
-    # Event Handlers
+    # ========== EVENT HANDLERS ==========
     
-    # Atualizar imagem original quando upload for feito
-    def update_original(image):
-        return image
+    # Quando imagem é carregada
+    def update_displays(image):
+        if image is None:
+            return None, None
+        return image, process_image(image, 1.0, 1.0, 1.0, 1.0, False)
     
     image_input.change(
-        fn=update_original,
+        fn=update_displays,
         inputs=[image_input],
-        outputs=[original_display]
+        outputs=[original_display, processed_display]
     )
     
-    # Processar imagem em tempo real quando controles mudarem
-    def update_processed(image, bright, cont, sat, sharp, auto):
-        if image is None:
-            return None
-        return process_image(image, bright, cont, sat, sharp, auto)
+    # Processar quando controles mudam
+    def update_processed(img, b, c, s, sh, ac):
+        return process_image(img, b, c, s, sh, ac)
     
-    # Conectar todos os controles à função de processamento
+    # Conectar todos os controles
     controls = [brightness, contrast, saturation, sharpness, auto_contrast]
     
     for control in controls:
@@ -227,82 +171,84 @@ with gr.Blocks(
             outputs=[processed_display]
         )
     
-    # Também processar quando imagem for carregada
-    image_input.change(
-        fn=lambda img: process_image(img, 1.0, 1.0, 1.0, 1.0, False) if img else None,
-        inputs=[image_input],
-        outputs=[processed_display]
-    )
-    
     # Presets
+    def apply_preset(preset_name):
+        preset = get_preset_values(preset_name)
+        return [
+            preset['brightness'],
+            preset['contrast'],
+            preset['saturation'],
+            preset['sharpness'],
+            preset['auto_contrast']
+        ]
+    
     vibrant_btn.click(
-        fn=lambda: get_preset('vibrant'),
+        fn=lambda: apply_preset('vibrant'),
         outputs=[brightness, contrast, saturation, sharpness, auto_contrast]
     )
     
     soft_btn.click(
-        fn=lambda: get_preset('soft'),
+        fn=lambda: apply_preset('soft'),
         outputs=[brightness, contrast, saturation, sharpness, auto_contrast]
     )
     
     dramatic_btn.click(
-        fn=lambda: get_preset('dramatic'),
+        fn=lambda: apply_preset('dramatic'),
         outputs=[brightness, contrast, saturation, sharpness, auto_contrast]
     )
     
     retro_btn.click(
-        fn=lambda: get_preset('retro'),
+        fn=lambda: apply_preset('retro'),
         outputs=[brightness, contrast, saturation, sharpness, auto_contrast]
     )
     
     bw_btn.click(
-        fn=lambda: get_preset('bw'),
+        fn=lambda: apply_preset('bw'),
         outputs=[brightness, contrast, saturation, sharpness, auto_contrast]
     )
     
     reset_btn.click(
-        fn=lambda: get_preset('reset'),
+        fn=lambda: apply_preset('reset'),
         outputs=[brightness, contrast, saturation, sharpness, auto_contrast]
     )
     
-    # Exportação
-    export_btn.click(
-        fn=save_temp_image,
+    # Download
+    download_btn.click(
+        fn=save_image_for_download,
         inputs=[processed_display],
-        outputs=[download_output]
+        outputs=[download_file]
     )
     
-    # Informações e guia
-    with gr.Accordion("📖 Guia de Uso", open=False):
+    # Guia de uso
+    with gr.Accordion("📖 Como usar", open=False):
         gr.Markdown("""
-        ## Como usar este app:
-        
-        1. **Carregue uma imagem** clicando na área de upload
-        2. **Ajuste os controles** deslizantes para modificar:
-           - **Brilho**: Controla a luminosidade geral
-           - **Contraste**: Aumenta/diminui diferença entre cores
-           - **Saturação**: Controla intensidade das cores (0 = P&B)
-           - **Nitidez**: Melhora definição dos detalhes
+        ## Instruções:
+        1. **Faça upload** de uma imagem
+        2. **Ajuste os controles** deslizantes
         3. **Use presets** para efeitos rápidos
         4. **Compare** original vs. ajustada
-        5. **Exporte** com o botão de download
+        5. **Baixe** o resultado
         
         ## Dicas:
         - **Retratos**: Brilho 1.05, Contraste 1.1, Saturação 0.9
         - **Paisagens**: Brilho 1.1, Contraste 1.2, Saturação 1.2
         - **Produtos**: Brilho 1.0, Contraste 1.3, Saturação 1.0
+        
+        ## Valores:
+        - **0.0** = Mínimo efeito
+        - **1.0** = Normal (padrão)
+        - **2.0** = Máximo efeito
         """)
     
     # Rodapé
     gr.HTML("""
-    <div style="text-align: center; margin-top: 30px; padding: 20px; background: #f5f5f5; border-radius: 10px;">
+    <div style="text-align: center; margin-top: 30px; padding: 15px; background: #f8f9fa; border-radius: 8px;">
         <p><strong>🎨 Basic Image Adjustments</strong> - App 2 do Photoshop AI Ecosystem</p>
-        <p style="font-size: 0.9em; color: #666;">
-        Cada app faz uma coisa específica, e faz bem!<br>
-        Este app apenas ajusta brilho, contraste, saturação e nitidez.
+        <p style="color: #666; font-size: 0.9em;">
+        Cada app tem uma função específica • Este app apenas ajusta brilho, contraste, saturação e nitidez
         </p>
     </div>
     """)
 
-# Para Hugging Face Spaces, não precisa chamar demo.launch()
-# O Spaces detecta automaticamente a variável 'demo'
+# ========== FIM DO APLICATIVO ==========
+# Não chame demo.launch() - O Hugging Face cuida disso