Update app.py

app.py

@@ -1,46 +1,82 @@
-import requests
-import json
-from live_preview_helpers import preview_image
-
-# Defina a URL da API do Hugging Face
-API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
-
-# Defina o cabeçalho da requisição com o token de autenticação
-headers = {
-    "Authorization": "Bearer YOUR_HUGGINGFACE_API_TOKEN"
-}
-
-def query(payload):
-    response = requests.post(API_URL, headers=headers, json=payload)
-    return response
-
-def generate_image(text_prompt, output_file="generated_image.png"):
-    data = {
-        "inputs": text_prompt,
-        "options": {
-            "wait_for_model": True,
-            "use_cpu": True,
-            "cpu_basic": {
-                "vCPU": 2,
-                "RAM": 16
-            }
-        }
-    }
-
-    response = query(data)
-
-    if response.status_code == 200:
-        with open(output_file, "wb") as f:
-            f.write(response.content)
-        print(f"Imagem gerada com sucesso e salva como '{output_file}'")
-        preview_image(output_file)  # Chama a função de visualização ao vivo
+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:
-        print(f"Erro na requisição: {response.status_code}")
-        print(f"Mensagem de erro: {response.text}")
+        frame = frame.convert("RGBA")
+    
+    # Desenha o pixel
+    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
+    )
+    return frame
 
-if __name__ == "__main__":
-    # Defina o texto que será usado para gerar a imagem
-    text_prompt = "A beautiful sunset over the mountains"
+# Função para criar uma animação GIF
+def create_animation(frames, duration):
+    if not frames:
+        return None
+    output_gif = io.BytesIO()
+    frames[0].save(
+        output_gif,
+        format="GIF",
+        save_all=True,
+        append_images=frames[1:],
+        duration=duration,
+        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
 
-    # Gere a imagem
-    generate_image(text_prompt)
+# Interface Gradio
+with gr.Blocks() as demo:
+    gr.Markdown("# Editor de Sprites e Animações")
+
+    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)
+
+    with gr.Row():
+        output = gr.Image(label="Pré-visualização do Frame Atual", tool="editor")
+        gif_output = gr.File(label="Animação Final (GIF)")
+
+    generate_btn = gr.Button("Gerar Animação")
+    generate_btn.click(
+        editor,
+        inputs=[frames_input, color_picker, x_input, y_input, pixel_size, duration],
+        outputs=[gif_output]
+    )
+
+# Executa o aplicativo
+if __name__ == "__main__":
+    demo.launch()