Update app.py

app.py

@@ -1,82 +1,89 @@
 import gradio as gr
-from PIL import Image, ImageDraw, ImageSequence
-import io
-
-# Função para criar ou editar frames
-def edit_frame(frame, color, x, y, pixel_size):
-    if frame is None:
-        # Cria um frame vazio
-        frame = Image.new("RGBA", (64, 64), (255, 255, 255, 0))
-    else:
-        frame = frame.convert("RGBA")
-    
-    # Desenha o pixel
+from PIL import Image, ImageDraw
+
+# Configurações gerais
+GRID_SIZE = 32  # Tamanho da grade
+PIXEL_SIZE = 20  # Tamanho de cada pixel no desenho
+CANVAS_SIZE = GRID_SIZE * PIXEL_SIZE  # Tamanho total do canvas
+FRAME_COUNT = 4  # Número de frames padrão na animação
+
+
+# Função para criar uma imagem em branco
+def create_blank_frame():
+    return Image.new("RGB", (GRID_SIZE, GRID_SIZE), color="white")
+
+
+# Função para desenhar no frame
+def draw_pixel(frame, x, y, color):
     draw = ImageDraw.Draw(frame)
     draw.rectangle(
-        [x * pixel_size, y * pixel_size, (x + 1) * pixel_size - 1, (y + 1) * pixel_size - 1],
-        fill=color,
-        outline=color
-    )
+        [x, y, x + 1, y + 1], fill=color
+    )  # Cada pixel é um pequeno quadrado 1x1
     return frame
 
-# Função para criar uma animação GIF
+
+# Função para converter frames em um GIF
 def create_animation(frames, duration):
-    if not frames:
-        return None
-    output_gif = io.BytesIO()
+    gif_path = "animation.gif"
     frames[0].save(
-        output_gif,
-        format="GIF",
+        gif_path,
         save_all=True,
         append_images=frames[1:],
         duration=duration,
-        loop=0
+        loop=0,
     )
-    output_gif.seek(0)
-    return output_gif
-
-# Função principal para o Gradio
-def editor(frames, color, x, y, pixel_size, duration):
-    updated_frames = []
-    for frame_data in frames:
-        if frame_data is None:
-            updated_frames.append(None)
-        else:
-            # Converte string de bytes para imagem
-            frame = Image.open(io.BytesIO(frame_data))
-            updated_frames.append(frame)
-
-    # Atualiza o frame atual
-    updated_frame = edit_frame(updated_frames[-1], color, x, y, pixel_size)
-    updated_frames[-1] = updated_frame
-
-    # Cria o GIF
-    gif = create_animation(updated_frames, duration)
-    return gif
-
-# Interface Gradio
-with gr.Blocks() as demo:
-    gr.Markdown("# Editor de Sprites e Animações")
+    return gif_path
+
+
+# Função principal do editor
+def editor(frames, frame_index, x, y, color, duration):
+    frame = frames[frame_index]
+    frame = draw_pixel(frame, x, y, color)
+    frames[frame_index] = frame
+
+    gif_path = create_animation(frames, duration)
+
+    return gif_path, frames
+
+
+# Inicializa frames em branco
+initial_frames = [create_blank_frame() for _ in range(FRAME_COUNT)]
+
+
+# Interface com Gradio
+with gr.Blocks() as app:
+    gr.Markdown("# Editor de Sprites e Animação")
 
     with gr.Row():
-        frames_input = gr.File(label="Frames Existentes (opcional)", file_types=["image"], file_count="multiple")
-        color_picker = gr.ColorPicker(label="Cor")
-        x_input = gr.Number(label="X", value=0, precision=0)
-        y_input = gr.Number(label="Y", value=0, precision=0)
-        pixel_size = gr.Slider(label="Tamanho do Pixel", minimum=1, maximum=10, value=1)
-        duration = gr.Slider(label="Duração do Frame (ms)", minimum=50, maximum=500, value=100)
+        canvas = gr.Image(label="Canvas", type="pil", tool=None)
 
     with gr.Row():
-        output = gr.Image(label="Pré-visualização do Frame Atual", tool="editor")
-        gif_output = gr.File(label="Animação Final (GIF)")
+        with gr.Column():
+            gr.Markdown("### Ferramentas de Desenho")
+            x = gr.Slider(0, GRID_SIZE - 1, step=1, label="Posição X")
+            y = gr.Slider(0, GRID_SIZE - 1, step=1, label="Posição Y")
+            color = gr.ColorPicker(label="Cor", value="#ff0000")
+            frame_index = gr.Slider(
+                0, FRAME_COUNT - 1, step=1, label="Frame Atual"
+            )
+            duration = gr.Slider(50, 500, step=10, label="Duração (ms)", value=100)
+
+        with gr.Column():
+            gr.Markdown("### Animação")
+            animation_output = gr.Image(label="Prévia da Animação")
+
+    with gr.Row():
+        update_button = gr.Button("Atualizar Frame")
+        save_button = gr.Button("Salvar Animação")
+
+    # Estados
+    frames_state = gr.State(value=initial_frames)
 
-    generate_btn = gr.Button("Gerar Animação")
-    generate_btn.click(
+    # Atualizar o frame
+    update_button.click(
         editor,
-        inputs=[frames_input, color_picker, x_input, y_input, pixel_size, duration],
-        outputs=[gif_output]
+        inputs=[frames_state, frame_index, x, y, color, duration],
+        outputs=[animation_output, frames_state],
     )
 
-# Executa o aplicativo
-if __name__ == "__main__":
-    demo.launch()
+app.launch()