Update app.py

app.py

@@ -1,83 +1,34 @@
+from transformers import pipeline
 import gradio as gr
-from PIL import Image, ImageDraw
-import os
+from PIL import Image
 
+# Carrega o pipeline de geração de imagens
+generator = pipeline("image-generation", model="stabilityai/stable-diffusion-2")
 
-# 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)
+def generate_frame_with_model(prompt, width=256, height=256):
+    result = generator(prompt, height=height, width=width, num_inference_steps=50)
+    image = Image.open(result[0]["generated_image"]).convert("RGB")
+    return image
 
-    # 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 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)
-
-    # Salva a animação como GIF
+def create_animation_with_model(prompt, frame_count, duration):
+    frames = [generate_frame_with_model(f"{prompt}, frame {i}") for i in range(frame_count)]
     gif_path = "output.gif"
-    frames[0].save(
-        gif_path,
-        save_all=True,
-        append_images=frames[1:],
-        duration=duration,
-        loop=0,
-    )
+    frames[0].save(gif_path, save_all=True, append_images=frames[1:], duration=duration, loop=0)
     return gif_path
 
-
-# Interface Gradio
-def animate_sprite(prompt, frame_count, duration):
-    gif_path = create_animation_from_prompts(prompt, frame_count, duration)
+def animate_with_model(prompt, frame_count, duration):
+    gif_path = create_animation_with_model(prompt, frame_count, duration)
     return gif_path
 
-
+# Interface Gradio
 with gr.Blocks() as app:
     gr.Markdown("# Gerador de Animações com Prompt de Texto")
-    
-    with gr.Row():
-        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,
-        )
-
+    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=10, value=5, 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,
-    )
+    generate_button.click(animate_with_model, inputs=[prompt, frame_count, duration], outputs=animation_output)
 
 app.launch()