Update app.py

app.py

@@ -371,26 +371,68 @@ def create_interface():
             image_output = gr.Image(label="Aperçu")
             status = gr.Textbox(label="Statut", interactive=False)
         
-        def generate(*args):
-            logger.info("Démarrage d'une nouvelle génération")
-            params = {
-                "format_size": args[0],
-                "orientation": args[1],
-                "style": args[2],
-                "layout": args[3],
-                "ambiance": args[4],
-                "palette": args[5],
-                "subject": args[6],
-                "title": args[7],
-                "detail_level": args[8],
-                "contrast": args[9],
-                "saturation": args[10],
-                "quality": args[11],
-                "creativity": args[12]
+        def generate(self, params: Dict[str, Any]) -> Tuple[Optional[Image.Image], str]:
+        try:
+            logger.info(f"Début de génération avec paramètres: {json.dumps(params, indent=2)}")
+            
+            if 'Bearer None' in self.headers['Authorization']:
+                logger.error("Token Hugging Face manquant ou invalide")
+                return None, "⚠️ Erreur: Token Hugging Face non configuré"
+
+            prompt = self._build_prompt(params)
+            logger.info(f"Prompt généré: {prompt}")
+            
+            content_type = self._detect_content_type(params["subject"])
+            logger.info(f"Type de contenu détecté: {content_type}")
+
+            payload = {
+                "inputs": prompt,
+                "parameters": {
+                    "negative_prompt": ART_STYLES[params["style"]]["negative_prompt"],
+                    "num_inference_steps": min(int(35 * (params["quality"]/100)), 40),
+                    "guidance_scale": min(7.5 * (params["creativity"]/10), 10.0),
+                    "width": 768,
+                    "height": 768 if params["orientation"] == "Portrait" else 512
+                }
             }
-            result = generator.generate(params)
-            logger.info(f"Génération terminée avec statut: {result[1]}")
-            return result
+            
+            logger.debug(f"Payload pour l'API: {json.dumps(payload, indent=2)}")
+
+            try:
+                response = requests.post(
+                    self.API_URL,
+                    headers=self.headers,
+                    json=payload,
+                    timeout=30
+                )
+                
+                logger.info(f"Statut de la réponse API: {response.status_code}")
+                if response.status_code != 200:
+                    logger.error(f"Contenu de la réponse en erreur: {response.text}")
+                
+                if response.status_code == 200:
+                    image = Image.open(io.BytesIO(response.content))
+                    return image, "✨ Création réussie!"
+                else:
+                    error_msg = f"⚠️ Erreur API {response.status_code}: {response.text}"
+                    logger.error(error_msg)
+                    return None, error_msg
+
+            except requests.exceptions.Timeout:
+                error_msg = "⚠️ Erreur: Le serveur met trop de temps à répondre"
+                logger.error(error_msg)
+                return None, error_msg
+            except requests.exceptions.RequestException as e:
+                error_msg = f"⚠️ Erreur de connexion: {str(e)}"
+                logger.error(error_msg)
+                return None, error_msg
+
+        except Exception as e:
+            error_msg = f"⚠️ Erreur inattendue: {str(e)}"
+            logger.exception("Erreur détaillée pendant la génération:")
+            return None, error_msg
+        finally:
+            gc.collect()
         
         generate_btn.click(
             generate,