Update app.py

app.py

@@ -43,8 +43,6 @@ ART_STYLES = {
         "prompt_prefix": "surrealist art, dreamlike imagery, symbolic elements, dali inspired, metaphysical art",
         "negative_prompt": "realistic, conventional, ordinary, literal"
     },
-
-    # Styles Modernes
     "Art Moderne": {
         "prompt_prefix": "modern art style poster, professional design, contemporary aesthetic",
         "negative_prompt": "traditional, cluttered, busy design, vintage"
@@ -65,8 +63,6 @@ ART_STYLES = {
         "prompt_prefix": "futuristic art style, dynamic movement, technological elements, speed and motion",
         "negative_prompt": "static, traditional, classical, historical"
     },
-
-    # Styles Spéciaux
     "Neo Vintage": {
         "prompt_prefix": "vintage style advertising poster, retro design, classic aesthetic",
         "negative_prompt": "modern, digital, contemporary style"
@@ -112,7 +108,7 @@ COMPOSITION_PARAMS = {
         "Terre": "earthy color palette, natural tones",
         "Néon": "neon color palette, vibrant glowing colors"
     }
-}class ImageGenerator:
+    }class ImageGenerator:
     def __init__(self):
         self.API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
         token = os.getenv('HUGGINGFACE_TOKEN')
@@ -136,9 +132,7 @@ COMPOSITION_PARAMS = {
                 "layout": COMPOSITION_PARAMS['Layouts'][params['layout']],
                 "ambiance": COMPOSITION_PARAMS['Ambiances'][params['ambiance']],
                 "palette": COMPOSITION_PARAMS['Palette'][params['palette']]
-            }
-            
-            # Préparation du prompt initial
+            }# Préparation du prompt initial
             base_prompt = f"{params['subject']}"
             if params.get('title'):
                 base_prompt += f", with text '{params['title']}'"
@@ -175,9 +169,7 @@ COMPOSITION_PARAMS = {
                     "width": 768,
                     "height": 768 if params["orientation"] == "Portrait" else 512
                 }
-            }
-            
-            logger.debug(f"Payload: {json.dumps(payload, indent=2)}")
+            }logger.debug(f"Payload: {json.dumps(payload, indent=2)}")
 
             response = requests.post(
                 self.API_URL,
@@ -212,7 +204,6 @@ def create_interface():
     """
     
     generator = ImageGenerator()
-    
     with gr.Blocks(css=css) as app:
         gr.HTML("""
             <div class="welcome">
@@ -260,9 +251,7 @@ def create_interface():
                         choices=list(COMPOSITION_PARAMS["Palette"].keys()),
                         value="Contrasté",
                         label="Palette"
-                    )
-            
-            with gr.Group(elem_classes="controls-group"):
+                    )with gr.Group(elem_classes="controls-group"):
                 gr.Markdown("### 📝 Contenu")
                 subject = gr.Textbox(
                     label="Description",
@@ -319,8 +308,7 @@ def create_interface():
             
             image_output = gr.Image(label="Aperçu")
             status = gr.Textbox(label="Statut", interactive=False)
-        
-        def generate(*args):
+            def generate(*args):
             logger.info("Démarrage d'une nouvelle génération")
             params = {
                 "format_size": args[0],
@@ -372,4 +360,5 @@ def create_interface():
 if __name__ == "__main__":
     app = create_interface()
     logger.info("Démarrage de l'application")
-    app.launch()
+    app.launch()
+