이미지 생성기

!pip install openai
!pip install gradio

import gradio as gr
import numpy as np
from PIL import Image

def generate_image(prompt, style, steps):
    # 실제로는 AI 모델 호출
    # 여기서는 데모용 더미 이미지 생성
    img = np.random.randint(0, 255, (512, 512, 3), dtype=np.uint8)
    return Image.fromarray(img)

with gr.Blocks(theme=gr.themes.Soft(), title="AI Art Studio") as demo:
    gr.Markdown("# 🎨 AI Art Studio")
    gr.Markdown("Create stunning artwork with AI")

    with gr.Row():
        with gr.Column(scale=1):
            prompt = gr.Textbox(
                label="Prompt",
                placeholder="Describe your artwork...",
                lines=3
            )
            style = gr.Dropdown(
                choices=["Realistic", "Anime", "Oil Painting", "Watercolor"],
                label="Art Style",
                value="Realistic"
            )
            steps = gr.Slider(20, 100, value=50, label="Quality Steps")
            generate_btn = gr.Button("Generate Art", variant="primary")

        with gr.Column(scale=2):
            output_image = gr.Image(label="Generated Artwork")

    generate_btn.click(
        generate_image,
        inputs=[prompt, style, steps],
        outputs=output_image
    )

demo.launch()