Update app.py

app.py

@@ -108,7 +108,9 @@ 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')
@@ -132,7 +134,9 @@ 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']}'"
@@ -169,7 +173,9 @@ 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,
@@ -204,6 +210,7 @@ def create_interface():
     """
     
     generator = ImageGenerator()
+    
     with gr.Blocks(css=css) as app:
         gr.HTML("""
             <div class="welcome">
@@ -251,7 +258,9 @@ 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",
@@ -308,7 +317,8 @@ 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],
@@ -360,5 +370,4 @@ def create_interface():
 if __name__ == "__main__":
     app = create_interface()
     logger.info("Démarrage de l'application")
-    app.launch()
-    
+    app.launch()