Spaces:
Running
Running
File size: 4,242 Bytes
846540c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
import gradio as gr
import tempfile
import os
from PIL import Image, ImageDraw
def generate_animated_sprite(character_description, num_frames, character_action, viewing_direction):
"""
Funzione per generare un'immagine GIF animata.
"""
print(f"Generazione sprite con questi parametri:")
print(f"- Descrizione: {character_description}")
print(f"- Frames: {num_frames}")
print(f"- Azione: {character_action}")
print(f"- Direzione: {viewing_direction}")
# Crea delle immagini di esempio per simulare un'animazione
frames = []
for i in range(int(num_frames)): # Converti in int per sicurezza
# Crea un'immagine di esempio
img = Image.new('RGB', (100, 100), color=(73, 109, 137))
d = ImageDraw.Draw(img)
# Disegna qualcosa di diverso in ogni frame per simulare l'animazione
x_offset = 10 + (i * 10) % 50
d.ellipse((x_offset, 20, x_offset + 60, 80), fill=(255, 255, 0))
d.text((10, 10), f"{character_action} - {i+1}", fill=(255, 255, 255))
frames.append(img)
# Crea una directory temporanea se non esiste
os.makedirs("tmp", exist_ok=True)
output_path = os.path.join("tmp", f"sprite_{hash(character_description)}.gif")
# Salva i frame come GIF animata
frames[0].save(
output_path,
format='GIF',
append_images=frames[1:],
save_all=True,
duration=200, # 200 ms tra i frame
loop=0 # 0 = loop infinito
)
return output_path
# Definisci l'interfaccia Gradio
def create_interface():
with gr.Blocks(title="Animated Sprite Generator") as demo:
gr.Markdown("# 🎮 AI Animated Sprite Generator")
gr.Markdown("""
This tool uses an AI model to generate animated sprites based on text descriptions.
Enter the character description, number of frames, character action, and viewing direction to generate your animated sprite.
""")
with gr.Row():
with gr.Column():
# Input components
char_desc = gr.Textbox(label="Character Description",
placeholder="Ex: a knight with golden armor and a fire sword",
lines=3)
num_frames = gr.Slider(minimum=1, maximum=8, step=1, value=4,
label="Number of Animation Frames")
char_action = gr.Dropdown(
choices=["idle", "walk", "run", "attack", "jump", "die", "cast spell", "dance"],
label="Character Action",
value="idle"
)
view_direction = gr.Dropdown(
choices=["front", "back", "left", "right", "front-left", "front-right", "back-left", "back-right"],
label="Viewing Direction",
value="front"
)
generate_btn = gr.Button("Generate Animated Sprite")
with gr.Column():
# Output component
animated_output = gr.Image(label="Animated Sprite (GIF)")
# Connect the button to the function
generate_btn.click(
fn=generate_animated_sprite,
inputs=[char_desc, num_frames, char_action, view_direction],
outputs=animated_output
)
# Predefined examples
gr.Examples(
[
["A wizard with blue cloak and pointed hat", 4, "cast spell", "front"],
["A warrior with heavy armor and axe", 6, "attack", "right"],
["A ninja with black clothes and throwing stars", 8, "run", "front-left"],
["A princess with golden crown and pink dress", 4, "dance", "front"]
],
inputs=[char_desc, num_frames, char_action, view_direction]
)
return demo
# Crea ed avvia l'interfaccia
demo = create_interface()
# Launch the Gradio interface
if __name__ == "__main__":
demo.launch()
else:
# Questa è la parte importante per Hugging Face Spaces
# Non chiamare .launch() qui, lo Space lo farà automaticamente
app = demo |