Update app.py

app.py

@@ -93,11 +93,14 @@ class ImageGenerator:
     def _build_prompt(self, params: Dict[str, Any]) -> str:
         style_info = ART_STYLES.get(params["style"], ART_STYLES["Neo Vintage"])
         prompt = f"{style_info['prompt_prefix']}, {params['subject']}"
-
-        for param_type in ['layout', 'ambiance', 'palette']:
-            if params.get(param_type):
-                prompt += f", {COMPOSITION_PARAMS[param_type.capitalize()][params[param_type]]}"
-
+        
+        if params.get("layout"):
+            prompt += f", {COMPOSITION_PARAMS['Layouts'][params['layout']]}"
+        if params.get("ambiance"):
+            prompt += f", {COMPOSITION_PARAMS['Ambiances'][params['ambiance']]}"
+        if params.get("palette"):
+            prompt += f", {COMPOSITION_PARAMS['Palette'][params['palette']]}"
+        
         for param, description in [
             ("detail_level", "highly detailed" if params.get("detail_level", 0) > 7 else "moderately detailed"),
             ("contrast", "high contrast" if params.get("contrast", 0) > 7 else "balanced contrast"),
@@ -105,10 +108,10 @@ class ImageGenerator:
         ]:
             if params.get(param):
                 prompt += f", {description}"
-
+        
         if params.get("title"):
             prompt += f", with text saying '{params['title']}'"
-
+        
         logger.debug(f"Prompt final: {prompt}")
         return prompt