|
|
import gradio as gr |
|
|
import os |
|
|
from PIL import Image, ImageDraw, ImageFont |
|
|
import requests |
|
|
import io |
|
|
|
|
|
|
|
|
ART_STYLES = { |
|
|
"Renaissance": { |
|
|
"prompt_prefix": "ultra-detailed renaissance style artwork, masterful anatomical details, dramatic chiaroscuro lighting", |
|
|
"guidance": 15.0, |
|
|
"steps": 80, |
|
|
"negative_prompt": "blurry, low quality, flat lighting, cartoon, modern" |
|
|
}, |
|
|
"Impressionnisme": { |
|
|
"prompt_prefix": "impressionist art style, vibrant brushstrokes, atmospheric lighting, monet-like technique", |
|
|
"guidance": 12.0, |
|
|
"steps": 65, |
|
|
"negative_prompt": "sharp details, photorealistic, digital art" |
|
|
}, |
|
|
"Art Moderne": { |
|
|
"prompt_prefix": "modern art composition, bold geometric shapes, abstract elements, strong visual impact", |
|
|
"guidance": 11.0, |
|
|
"steps": 60, |
|
|
"negative_prompt": "realistic, traditional, detailed texture" |
|
|
}, |
|
|
"Illustration Pro": { |
|
|
"prompt_prefix": "professional editorial illustration, sophisticated art style, high-end magazine quality", |
|
|
"guidance": 13.5, |
|
|
"steps": 70, |
|
|
"negative_prompt": "amateur, basic, photographic" |
|
|
}, |
|
|
"Néo Vintage": { |
|
|
"prompt_prefix": "vintage poster design, retro aesthetic, limited color palette, screen printing style", |
|
|
"guidance": 12.5, |
|
|
"steps": 65, |
|
|
"negative_prompt": "modern, digital, photorealistic" |
|
|
}, |
|
|
"Encre Artistique": { |
|
|
"prompt_prefix": "artistic ink illustration, dynamic brushwork, high contrast black and white, expressive strokes", |
|
|
"guidance": 14.0, |
|
|
"steps": 75, |
|
|
"negative_prompt": "color, photographic, digital" |
|
|
}, |
|
|
"Pop Art": { |
|
|
"prompt_prefix": "pop art style, bold colors, halftone dots, comic book aesthetic, Warhol-inspired", |
|
|
"guidance": 11.5, |
|
|
"steps": 60, |
|
|
"negative_prompt": "realistic, painterly, subtle" |
|
|
}, |
|
|
"Dark Fantasy": { |
|
|
"prompt_prefix": "dark fantasy art, atmospheric mood, dramatic lighting, mythological elements", |
|
|
"guidance": 13.0, |
|
|
"steps": 70, |
|
|
"negative_prompt": "bright, cheerful, realistic photo" |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
COLLECTIONS = { |
|
|
"Nature Artistique": { |
|
|
"prompts": ["artistic nature illustration", "stylized botanical elements"], |
|
|
"styles": ["organic composition", "natural textures", "dynamic flow"], |
|
|
"negative": "artificial, synthetic, urban" |
|
|
}, |
|
|
"Portraits Expressifs": { |
|
|
"prompts": ["expressive portrait art", "emotional character illustration"], |
|
|
"styles": ["dramatic lighting", "powerful expression", "artistic interpretation"], |
|
|
"negative": "flat lighting, emotionless, stiff pose" |
|
|
}, |
|
|
"Urbain Stylisé": { |
|
|
"prompts": ["stylized urban art", "city aesthetic interpretation"], |
|
|
"styles": ["architectural elements", "urban textures", "street art influence"], |
|
|
"negative": "rural, natural, realistic photo" |
|
|
}, |
|
|
"Fantaisie Épique": { |
|
|
"prompts": ["epic fantasy artwork", "mythological interpretation"], |
|
|
"styles": ["magical elements", "dramatic atmosphere", "legendary creatures"], |
|
|
"negative": "mundane, realistic, modern" |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
CUSTOM_CSS = """ |
|
|
.container { max-width: 1200px; margin: auto; } |
|
|
.welcome { text-align: center; margin: 20px 0; padding: 20px; background: #1e293b; border-radius: 10px; } |
|
|
.quality-controls { margin: 10px 0; padding: 10px; background: #2d3748; border-radius: 5px; } |
|
|
""" |
|
|
|
|
|
def create_interface(): |
|
|
def generate_image(format_size, orientation, prompt, style, quality, creativity, collection=None): |
|
|
API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0" |
|
|
headers = {"Authorization": f"Bearer {os.getenv('HUGGINGFACE_TOKEN')}"} |
|
|
|
|
|
try: |
|
|
|
|
|
base_width = 1024 |
|
|
base_height = 1024 |
|
|
|
|
|
if orientation == "Portrait": |
|
|
if format_size in ["A4", "A3"]: |
|
|
base_width = 768 |
|
|
base_height = 1024 |
|
|
else: |
|
|
if format_size in ["A4", "A3"]: |
|
|
base_width = 1024 |
|
|
base_height = 768 |
|
|
|
|
|
|
|
|
style_config = ART_STYLES[style] |
|
|
enhanced_prompt = f"{style_config['prompt_prefix']}, {prompt}" |
|
|
|
|
|
if collection and collection in COLLECTIONS: |
|
|
collection_data = COLLECTIONS[collection] |
|
|
collection_elements = collection_data["prompts"] + collection_data["styles"] |
|
|
enhanced_prompt += f", {', '.join(collection_elements)}" |
|
|
negative_prompt = f"{style_config['negative_prompt']}, {collection_data['negative']}" |
|
|
else: |
|
|
negative_prompt = style_config['negative_prompt'] |
|
|
|
|
|
|
|
|
payload = { |
|
|
"inputs": enhanced_prompt, |
|
|
"parameters": { |
|
|
"negative_prompt": negative_prompt, |
|
|
"num_inference_steps": int(style_config['steps'] * (quality/100)), |
|
|
"guidance_scale": style_config['guidance'] * (creativity/10), |
|
|
"width": base_width, |
|
|
"height": base_height |
|
|
} |
|
|
} |
|
|
|
|
|
print(f"Génération avec paramètres: {payload}") |
|
|
|
|
|
response = requests.post(API_URL, headers=headers, json=payload, timeout=60) |
|
|
|
|
|
if response.status_code == 200: |
|
|
image = Image.open(io.BytesIO(response.content)) |
|
|
return image, f"✨ Création {style} réussie!" |
|
|
else: |
|
|
print(f"Erreur API: {response.text}") |
|
|
return None, f"⚠️ Erreur {response.status_code}: Ajustez les paramètres" |
|
|
|
|
|
except Exception as e: |
|
|
print(f"Exception: {str(e)}") |
|
|
return None, f"⚠️ Erreur: {str(e)}" |
|
|
|
|
|
with gr.Blocks(css=CUSTOM_CSS) as app: |
|
|
gr.HTML(""" |
|
|
<div class="welcome"> |
|
|
<h1>🤖 Equity Artisan 3.0</h1> |
|
|
<p>Créez des affiches professionnelles étape par étape avec notre assistant créatif augmenté.</p> |
|
|
</div> |
|
|
""") |
|
|
|
|
|
with gr.Column(elem_classes="container"): |
|
|
|
|
|
with gr.Group(): |
|
|
gr.Markdown("### 📏 Étape 1: Format et Orientation") |
|
|
with gr.Row(): |
|
|
format_choice = gr.Dropdown( |
|
|
choices=["A4", "A3", "A2", "A1", "A0"], |
|
|
value="A4", |
|
|
label="Format" |
|
|
) |
|
|
orientation = gr.Radio( |
|
|
choices=["Portrait", "Paysage"], |
|
|
value="Portrait", |
|
|
label="Orientation" |
|
|
) |
|
|
|
|
|
|
|
|
with gr.Group(): |
|
|
gr.Markdown("### 🎨 Étape 2: Création Visuelle") |
|
|
|
|
|
with gr.Row(): |
|
|
with gr.Column(): |
|
|
style = gr.Dropdown( |
|
|
choices=list(ART_STYLES.keys()), |
|
|
value="Renaissance", |
|
|
label="Style artistique" |
|
|
) |
|
|
|
|
|
collection = gr.Dropdown( |
|
|
choices=list(COLLECTIONS.keys()), |
|
|
label="Collection (optionnel)", |
|
|
info="Thèmes artistiques spécialisés" |
|
|
) |
|
|
|
|
|
prompt = gr.Textbox( |
|
|
lines=3, |
|
|
label="Description", |
|
|
placeholder="Décrivez votre vision artistique..." |
|
|
) |
|
|
|
|
|
with gr.Group(elem_classes="quality-controls"): |
|
|
quality = gr.Slider( |
|
|
minimum=30, |
|
|
maximum=50, |
|
|
value=35, |
|
|
step=5, |
|
|
label="Qualité", |
|
|
info="Influence le niveau de détail" |
|
|
) |
|
|
|
|
|
creativity = gr.Slider( |
|
|
minimum=5, |
|
|
maximum=15, |
|
|
value=7.5, |
|
|
step=0.5, |
|
|
label="Créativité", |
|
|
info="Balance entre fidélité et créativité" |
|
|
) |
|
|
|
|
|
generate_btn = gr.Button("✨ Générer", variant="primary") |
|
|
clear_btn = gr.Button("🗑️ Effacer", variant="secondary") |
|
|
|
|
|
image_output = gr.Image( |
|
|
label="Aperçu", |
|
|
type="pil" |
|
|
) |
|
|
|
|
|
|
|
|
status = gr.Textbox( |
|
|
label="Status", |
|
|
interactive=False, |
|
|
value="Prêt à générer" |
|
|
) |
|
|
|
|
|
|
|
|
generate_btn.click( |
|
|
generate_image, |
|
|
inputs=[ |
|
|
format_choice, |
|
|
orientation, |
|
|
prompt, |
|
|
style, |
|
|
quality, |
|
|
creativity, |
|
|
collection |
|
|
], |
|
|
outputs=[image_output, status] |
|
|
) |
|
|
|
|
|
clear_btn.click( |
|
|
lambda: (None, "Image effacée"), |
|
|
outputs=[image_output, status] |
|
|
) |
|
|
|
|
|
return app |
|
|
|
|
|
if __name__ == "__main__": |
|
|
app = create_interface() |
|
|
app.launch() |
