Update models.py

models.py

@@ -62,6 +62,41 @@ class BagelAPIAnalyzer(BaseImageAnalyzer):
             logger.error(f"BAGEL API client initialization failed: {e}")
             return False
 
+    def _extract_camera_setup(self, description: str) -> Optional[str]:
+        """Extract camera setup recommendation from BAGEL response"""
+        try:
+            # Look for CAMERA_SETUP section
+            if "CAMERA_SETUP:" in description:
+                parts = description.split("CAMERA_SETUP:")
+                if len(parts) > 1:
+                    camera_part = parts[1].strip()
+                    # Clean up any additional formatting
+                    camera_part = camera_part.replace("\n", " ").strip()
+                    return camera_part
+            
+            # Alternative patterns for camera recommendations
+            camera_patterns = [
+                "Shot on ",
+                "Camera: ",
+                "Equipment: ",
+                "Recommended camera:",
+                "Camera setup:"
+            ]
+            
+            for pattern in camera_patterns:
+                if pattern in description:
+                    # Extract text after the pattern
+                    idx = description.find(pattern)
+                    camera_text = description[idx:].split('.')[0]  # Take first sentence
+                    if len(camera_text) > len(pattern) + 10:  # Ensure meaningful content
+                        return camera_text.strip()
+            
+            return None
+            
+        except Exception as e:
+            logger.warning(f"Failed to extract camera setup: {e}")
+            return None
+
     def _save_temp_image(self, image: Image.Image) -> str:
         """Save image to temporary file for API call"""
         try:
@@ -101,7 +136,13 @@ class BagelAPIAnalyzer(BaseImageAnalyzer):
         try:
             # Default prompt for detailed image analysis
             if prompt is None:
-                prompt = "Describe this image in rich detail as a single flowing paragraph. Include the visual elements, composition, lighting, colors, artistic style, mood, and atmosphere. Write it as a continuous narrative description without using numbered lists or bullet points. Focus on creating a vivid, cohesive description that captures the essence and details of the image."
+                prompt = """Analyze this image and provide a detailed description in two parts:
+
+1. DESCRIPTION: Write a flowing paragraph describing the visual elements, composition, lighting, colors, artistic style, mood, and atmosphere. Start directly with the subject (e.g., "A black and white illustration..." not "The image shows..."). Write as a continuous narrative without numbered lists.
+
+2. CAMERA_SETUP: Based on the photographic characteristics you observe, recommend the specific camera system, lens, and settings that would best capture this type of image. Consider focal length, aperture, lighting setup, and shooting style that matches what you see.
+
+Format your response clearly with these two sections."""
             
             # Save image to temporary file
             temp_path = self._save_temp_image(image)
@@ -127,11 +168,20 @@ class BagelAPIAnalyzer(BaseImageAnalyzer):
             else:
                 description = str(result)
             
-            # Clean up the description
+            # Clean up the description and extract camera setup if present
             if isinstance(description, str) and description.strip():
                 description = description.strip()
+                
+                # Store camera setup separately if found
+                camera_setup = self._extract_camera_setup(description)
+                if camera_setup:
+                    metadata["camera_setup"] = camera_setup
+                    metadata["has_camera_suggestion"] = True
+                else:
+                    metadata["has_camera_suggestion"] = False
             else:
                 description = "Detailed image analysis completed successfully"
+                metadata["has_camera_suggestion"] = False
             
             # Prepare metadata
             metadata = {
@@ -158,7 +208,13 @@ class BagelAPIAnalyzer(BaseImageAnalyzer):
 
     def analyze_for_flux_prompt(self, image: Image.Image) -> Tuple[str, Dict[str, Any]]:
         """Analyze image specifically for FLUX prompt generation"""
-        flux_prompt = """Create a detailed, flowing description of this image suitable for FLUX generation. Write as a single coherent paragraph describing the photographic style, composition, lighting, colors, mood, and technical details. Focus on artistic and photographic elements that would help recreate this image. Avoid numbered lists or bullet points - write it as a natural, descriptive narrative."""
+        flux_prompt = """Analyze this image for FLUX generation and provide two sections:
+
+1. DESCRIPTION: Create a detailed, flowing description suitable for FLUX generation. Write as a single coherent paragraph starting directly with the subject (avoid "The image shows..."). Describe photographic style, composition, lighting, colors, mood, and artistic elements.
+
+2. CAMERA_SETUP: Recommend the specific professional camera system, lens, aperture, and technical settings that would recreate this exact image. Be specific about equipment brands, focal lengths, and shooting parameters based on the visual characteristics you observe.
+
+Provide both sections clearly formatted."""
         
         return self.analyze_image(image, flux_prompt)