import gradio as gr import os from PIL import Image, ImageDraw, ImageFont, ImageEnhance import requests import io import json # Définition des styles ART_STYLES = { "Renaissance": { "prompt_prefix": "renaissance style masterpiece, classical composition, chiaroscuro lighting", "text_style": "elegant classical typography, golden ratio composition", "guidance": 15.0, "steps": 80, "negative_prompt": "modern, digital, low quality, blurry" }, "Impressionnisme": { "prompt_prefix": "impressionist painting style, vibrant brushstrokes, atmospheric lighting", "text_style": "artistic text integration, painterly typography", "guidance": 12.0, "steps": 65, "negative_prompt": "sharp details, digital art, photorealistic" }, "Art Moderne": { "prompt_prefix": "modern art style, geometric shapes, bold composition", "text_style": "modern typography, geometric letter design", "guidance": 11.0, "steps": 60, "negative_prompt": "traditional, realistic, detailed texture" }, "Neo Vintage": { "prompt_prefix": "vintage style poster, retro advertisement design, classic illustration", "text_style": "vintage typography, retro lettering style", "guidance": 12.5, "steps": 65, "negative_prompt": "modern, digital, contemporary" }, "Pop Art": { "prompt_prefix": "pop art style, comic book aesthetic, bold colors, halftone dots", "text_style": "bold comic book typography, retro lettering", "guidance": 11.5, "steps": 60, "negative_prompt": "subtle, realistic, painterly" } } # Effets de texte TEXT_EFFECTS = { "Standard": { "prompt_suffix": "with clear readable text", "text_params": {"weight": 1.0} }, "Néon": { "prompt_suffix": "with glowing neon text effect", "text_params": {"weight": 1.2} }, "3D": { "prompt_suffix": "with 3D text effect, depth and shadows", "text_params": {"weight": 1.3} }, "Métallique": { "prompt_suffix": "with metallic text effect, reflective surface", "text_params": {"weight": 1.1} }, "Graffiti": { "prompt_suffix": "with graffiti style text, urban art typography", "text_params": {"weight": 1.2} } } # Collections thématiques THEME_COLLECTIONS = { "Nature": { "prompts": ["natural elements", "organic composition"], "styles": ["flowing lines", "natural textures"], "negative": "artificial, synthetic" }, "Urbain": { "prompts": ["urban landscape", "city elements"], "styles": ["street art", "architectural elements"], "negative": "rural, natural" }, "Tech": { "prompts": ["technological elements", "digital aesthetic"], "styles": ["circuit patterns", "tech elements"], "negative": "organic, traditional" } } # CSS personnalisé 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; } .style-group { background: #2d3748; padding: 15px; border-radius: 5px; margin: 10px 0; } .text-effects { background: #374151; padding: 12px; border-radius: 5px; margin: 8px 0; } .preview-panel { position: relative; } .parameters { display: flex; gap: 10px; } """ def enhance_prompt(subject, style, text_effect, collection=None, additional_details=""): """Génération de prompt optimisée""" style_config = ART_STYLES[style] text_config = TEXT_EFFECTS[text_effect] base_prompt = f"{style_config['prompt_prefix']}, {subject}" if text_effect != "Standard": base_prompt += f", {text_config['prompt_suffix']}" if collection and collection in THEME_COLLECTIONS: collection_data = THEME_COLLECTIONS[collection] collection_elements = collection_data["prompts"] + collection_data["styles"] base_prompt += f", {', '.join(collection_elements)}" if additional_details: base_prompt += f", {additional_details}" negative_prompt = style_config['negative_prompt'] if collection: negative_prompt += f", {THEME_COLLECTIONS[collection]['negative']}" return base_prompt, negative_prompt def generate_image(format_size, orientation, subject, style, text_effect, collection, title, subtitle, quality, creativity, additional_details=""): """Fonction de génération d'image""" 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 enhanced_prompt, negative_prompt = enhance_prompt( subject, style, text_effect, collection, additional_details ) if title: enhanced_prompt += f", with text: '{title}'" if subtitle: enhanced_prompt += f", subtitle: '{subtitle}'" print(f"Prompt: {enhanced_prompt}") # Debug payload = { "inputs": enhanced_prompt, "parameters": { "negative_prompt": negative_prompt, "num_inference_steps": int(ART_STYLES[style]['steps'] * (quality/100)), "guidance_scale": ART_STYLES[style]['guidance'] * (creativity/10), "width": base_width, "height": base_height } } 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} avec effet {text_effect} 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)}" def create_interface(): with gr.Blocks(css=CUSTOM_CSS) as app: gr.HTML("""

🤖 Equity Artisan 3.0

Créez des affiches artistiques professionnelles avec notre assistant créatif augmenté.

""") with gr.Column(elem_classes="container"): # Format with gr.Group(elem_classes="style-group"): gr.Markdown("### 📏 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" ) # Style et Contenu with gr.Group(elem_classes="style-group"): gr.Markdown("### 🎨 Style et Création") with gr.Row(): with gr.Column(scale=1): style = gr.Dropdown( choices=list(ART_STYLES.keys()), value="Neo Vintage", label="Style artistique" ) text_effect = gr.Dropdown( choices=list(TEXT_EFFECTS.keys()), value="Standard", label="Effet de texte" ) collection = gr.Dropdown( choices=list(THEME_COLLECTIONS.keys()), label="Collection (optionnel)" ) with gr.Column(scale=2): subject = gr.Textbox( label="Sujet principal", placeholder="Ex: loup, paysage urbain..." ) additional_details = gr.Textbox( lines=2, label="Détails additionnels", placeholder="Ajoutez des détails..." ) # Texte with gr.Group(elem_classes="text-effects"): gr.Markdown("### ✍️ Texte") with gr.Row(): title = gr.Textbox( label="Titre principal", placeholder="Texte principal..." ) subtitle = gr.Textbox( label="Sous-titre", placeholder="Texte secondaire..." ) # Paramètres with gr.Group(elem_classes="quality-controls"): with gr.Row(): quality = gr.Slider( minimum=30, maximum=50, value=35, step=5, label="Qualité" ) creativity = gr.Slider( minimum=5, maximum=15, value=7.5, step=0.5, label="Créativité" ) with gr.Row(): 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="Statut", interactive=False) # Events generate_btn.click( generate_image, inputs=[ format_choice, orientation, subject, style, text_effect, collection, title, subtitle, quality, creativity, additional_details ], 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()