Update utils.py

utils.py

@@ -352,33 +352,109 @@ def _get_scene_camera_setup(scene_type: str) -> str:
 
 
 def _get_essential_keywords(description: str, camera_setup: str, analysis_metadata: Optional[Dict[str, Any]]) -> List[str]:
-    """Get essential style keywords without redundancy"""
+    """Get essential lighting and composition keywords without redundancy"""
     try:
         keywords = []
         description_lower = description.lower()
         
-        # Only add depth of field if not already mentioned
+        # Detect and add lighting information
+        lighting = _detect_lighting_from_description(description_lower)
+        if lighting:
+            keywords.append(lighting)
+        
+        # Detect and add composition technique
+        composition = _detect_composition_from_description(description_lower, camera_setup)
+        if composition:
+            keywords.append(composition)
+        
+        # Only add depth of field if not already mentioned and relevant
         if "depth" not in description_lower and "bokeh" not in description_lower:
             if any(term in camera_setup for term in ["f/1.4", "f/2.8", "85mm"]):
                 keywords.append("shallow depth of field")
         
-        # Add professional photography only if no specific camera mentioned
-        if "shot on" not in camera_setup:
+        # Add professional photography only if no specific camera mentioned (fallback)
+        if "shot on" not in camera_setup and len(keywords) == 0:
             keywords.append("professional photography")
         
-        # Scene-specific keywords
-        if "portrait" in description_lower and "studio lighting" not in description_lower:
-            keywords.append("professional portrait")
-        
-        # Technical quality (only if needed)
-        if len(keywords) < 2:
-            keywords.append("high quality")
-        
         return keywords[:3]  # Limit to 3 essential keywords
         
     except Exception as e:
         logger.warning(f"Keyword extraction failed: {e}")
-        return ["professional photography"]
+        return ["natural lighting"]
+
+
+def _detect_lighting_from_description(description_lower: str) -> Optional[str]:
+    """Detect lighting type from description"""
+    try:
+        # Check for existing lighting mentions
+        if any(term in description_lower for term in ["lighting", "light", "lit", "illuminated"]):
+            return None  # Already has lighting info
+        
+        # Detect lighting conditions from scene context
+        if any(term in description_lower for term in ["sunset", "sunrise", "golden", "warm"]):
+            return "golden hour lighting"
+        elif any(term in description_lower for term in ["twilight", "dusk", "evening", "blue hour"]):
+            return "blue hour lighting"
+        elif any(term in description_lower for term in ["overcast", "cloudy", "soft", "diffused"]):
+            return "soft natural lighting"
+        elif any(term in description_lower for term in ["bright", "sunny", "daylight", "outdoor"]):
+            return "natural daylight"
+        elif any(term in description_lower for term in ["indoor", "interior", "inside"]):
+            return "ambient lighting"
+        elif any(term in description_lower for term in ["studio", "controlled", "professional"]):
+            return "studio lighting"
+        elif any(term in description_lower for term in ["dramatic", "moody", "shadow"]):
+            return "dramatic lighting"
+        else:
+            # Default based on scene type
+            if any(term in description_lower for term in ["portrait", "person", "face"]):
+                return "natural lighting"
+            elif any(term in description_lower for term in ["landscape", "outdoor", "nature"]):
+                return "natural daylight"
+            else:
+                return "natural lighting"
+                
+    except Exception as e:
+        logger.warning(f"Lighting detection failed: {e}")
+        return "natural lighting"
+
+
+def _detect_composition_from_description(description_lower: str, camera_setup: str) -> Optional[str]:
+    """Detect composition technique from description and camera setup"""
+    try:
+        # Check for existing composition mentions
+        if any(term in description_lower for term in ["composition", "framing", "rule of thirds"]):
+            return None  # Already has composition info
+        
+        # Detect composition from perspective/angle
+        if any(term in description_lower for term in ["elevated", "above", "overhead", "aerial"]):
+            return "elevated perspective"
+        elif any(term in description_lower for term in ["low angle", "looking up", "from below"]):
+            return "low angle composition"
+        elif any(term in description_lower for term in ["close-up", "tight", "detailed"]):
+            return "close-up framing"
+        elif any(term in description_lower for term in ["wide", "expansive", "panoramic"]):
+            return "wide composition"
+        
+        # Detect composition from subject arrangement
+        elif any(term in description_lower for term in ["centered", "center", "middle"]):
+            return "centered composition"
+        elif any(term in description_lower for term in ["symmetr", "balanced", "mirror"]):
+            return "symmetrical composition"
+        elif any(term in description_lower for term in ["leading", "lines", "path", "diagonal"]):
+            return "leading lines"
+        
+        # Default composition based on camera setup
+        elif any(term in camera_setup for term in ["85mm", "135mm", "f/1.4", "f/2.8"]):
+            return "rule of thirds"  # Portrait/shallow DOF typically uses rule of thirds
+        elif any(term in camera_setup for term in ["24mm", "35mm", "wide"]):
+            return "dynamic composition"  # Wide angle allows for dynamic compositions
+        else:
+            return "rule of thirds"  # Universal fallback
+            
+    except Exception as e:
+        logger.warning(f"Composition detection failed: {e}")
+        return "rule of thirds"
 
 
 def _build_optimized_prompt(description: str, camera_setup: str, keywords: List[str]) -> str: