Update app.py

app.py

@@ -1,30 +1,36 @@
 import gradio as gr
 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")
+import os
+
+
+# Função para criar frames a partir de prompts (exemplo usando texto sobre imagens)
+def generate_frame(prompt, frame_number, width=256, height=256):
+    # Cria uma imagem em branco
+    img = Image.new("RGB", (width, height), color="white")
+    draw = ImageDraw.Draw(img)
+
+    # Adiciona o texto do prompt no centro
+    text = f"{prompt} ({frame_number})"
+    text_width, text_height = draw.textsize(text)
+    draw.text(
+        ((width - text_width) / 2, (height - text_height) / 2),
+        text,
+        fill="black",
+    )
+    return img
 
 
-# Função para desenhar no frame
-def draw_pixel(frame, x, y, color):
-    draw = ImageDraw.Draw(frame)
-    draw.rectangle(
-        [x, y, x + 1, y + 1], fill=color
-    )  # Cada pixel é um pequeno quadrado 1x1
-    return frame
+# Função para criar uma animação (GIF)
+def create_animation_from_prompts(prompt, frame_count=10, duration=100):
+    frames = []
 
+    # Gera os frames com base no prompt
+    for i in range(frame_count):
+        frame = generate_frame(prompt, i + 1)
+        frames.append(frame)
 
-# Função para converter frames em um GIF
-def create_animation(frames, duration):
-    gif_path = "animation.gif"
+    # Salva a animação como GIF
+    gif_path = "output.gif"
     frames[0].save(
         gif_path,
         save_all=True,
@@ -35,55 +41,43 @@ def create_animation(frames, duration):
     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 Gradio
+def animate_sprite(prompt, frame_count, duration):
+    gif_path = create_animation_from_prompts(prompt, frame_count, duration)
+    return gif_path
 
 
-# Interface com Gradio
 with gr.Blocks() as app:
-    gr.Markdown("# Editor de Sprites e Animação")
-
-    with gr.Row():
-        canvas = gr.Image(label="Canvas", type="pil", tool=None)
-
-    with gr.Row():
-        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")
-
+    gr.Markdown("# Gerador de Animações com Prompt de Texto")
+    
     with gr.Row():
-        update_button = gr.Button("Atualizar Frame")
-        save_button = gr.Button("Salvar Animação")
-
-    # Estados
-    frames_state = gr.State(value=initial_frames)
-
-    # Atualizar o frame
-    update_button.click(
-        editor,
-        inputs=[frames_state, frame_index, x, y, color, duration],
-        outputs=[animation_output, frames_state],
+        prompt = gr.Textbox(
+            label="Descrição do Sprite",
+            placeholder="Descreva o sprite ou a animação que deseja criar",
+        )
+        frame_count = gr.Slider(
+            label="Número de Frames",
+            minimum=1,
+            maximum=30,
+            value=10,
+            step=1,
+        )
+        duration = gr.Slider(
+            label="Duração do Frame (ms)",
+            minimum=50,
+            maximum=500,
+            value=100,
+            step=10,
+        )
+
+    animation_output = gr.Image(label="Animação Gerada")
+
+    generate_button = gr.Button("Gerar Animação")
+
+    generate_button.click(
+        animate_sprite,
+        inputs=[prompt, frame_count, duration],
+        outputs=animation_output,
     )
 
 app.launch()