app.py

import gradio as gr
import os
import subprocess

# Define directories for uploads and outputs
UPLOAD_FOLDER = 'uploads_gradio'
OUTPUT_FOLDER = 'outputs_gradio'

# Create directories if they don't exist
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
os.makedirs(OUTPUT_FOLDER, exist_ok=True)

# Create flagged directory inside outputs_gradio to avoid permission issues
FLAGGED_DIR = os.path.join(OUTPUT_FOLDER, 'flagged')
os.makedirs(FLAGGED_DIR, exist_ok=True)

def animate_image(file_path):
    """
    Function to process the uploaded image and generate an animated GIF.
    
    Args:
        file_path (str): Path to the uploaded file.
    
    Returns:
        str: Path to the generated GIF.
    """
    if not file_path:
        raise ValueError("No file uploaded.")

    input_path = file_path
    filename = os.path.basename(input_path)
    base, ext = os.path.splitext(filename)
    
    # Validate file extension
    allowed_extensions = ['.png', '.jpg', '.jpeg', '.bmp']
    if ext.lower() not in allowed_extensions:
        raise ValueError("Unsupported file type. Please upload an image file (png, jpg, jpeg, bmp).")

    # Define the annotation directory for this specific image
    char_anno_dir = os.path.join(OUTPUT_FOLDER, f"{base}_out")
    os.makedirs(char_anno_dir, exist_ok=True)
    
    try:
        # Run the image_to_animation.py script
        subprocess.run([
            'python', 'examples/image_to_animation.py',
            input_path, char_anno_dir
        ], check=True)

        # Path to the generated GIF
        gif_path = os.path.join(char_anno_dir, "video.gif")
        
        if os.path.exists(gif_path):
            return gif_path
        else:
            raise FileNotFoundError("Animation failed to generate. Please ensure the input image contains clear humanoid drawings.")
    
    except subprocess.CalledProcessError as e:
        raise RuntimeError(f"Error during processing: {e}")
    except Exception as e:
        raise RuntimeError(f"Unexpected error: {e}")

# Create the Gradio interface
iface = gr.Interface(
    fn=animate_image,
    inputs=gr.Image(label="Upload Drawing", type="filepath", sources=["upload", "webcam"]),
    outputs=gr.Image(label="Animated GIF"),
    title="Animated Drawings",
    description="Upload your drawing or take a photo, and get an animated GIF.",
    flagging_dir=FLAGGED_DIR
)

if __name__ == "__main__":
    # Listen on 0.0.0.0:7860 (default for Hugging Face Spaces)
    iface.launch(server_name="0.0.0.0", server_port=7860)