Update models.py

models.py

@@ -148,13 +148,19 @@ class BagelAPIAnalyzer(BaseImageAnalyzer):
         try:
             # Default prompt for detailed image analysis
             if prompt is None:
-                prompt = """Analyze this image and provide a detailed description in two parts:
+                prompt = """You are analyzing a photograph for FLUX image generation. Provide a detailed analysis in two sections:
 
-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.
+1. DESCRIPTION: Start directly with the subject (e.g., "A color photograph showing..." or "A black and white photograph depicting..."). First, determine if this is a photograph, illustration, or artwork. Then describe the visual elements, composition, lighting, colors (be specific about the color palette - warm tones, cool tones, monochrome, etc.), artistic style, mood, and atmosphere. Write as a flowing paragraph 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.
+2. CAMERA_SETUP: Based on the photographic characteristics and scene type you observe, recommend the specific camera system and lens that would realistically capture this type of scene:
+- For street/documentary photography: suggest cameras like Canon EOS R6, Sony A7 IV, Leica Q2 with 35mm or 24-70mm lenses
+- For portraits: suggest cameras like Canon EOS R5, Sony A7R V with 85mm or 135mm lenses  
+- For landscapes: suggest cameras like Phase One XT, Fujifilm GFX with wide-angle lenses
+- For sports/action: suggest cameras like Canon EOS-1D X, Sony A9 III with telephoto lenses
+- For macro: suggest specialized macro lenses
+Be specific about focal length, aperture, and shooting style based on what you actually see in the image.
 
-Format your response clearly with these two sections."""
+Analyze carefully and be accurate about colors and image type."""
             
             # Save image to temporary file
             temp_path = self._save_temp_image(image)
@@ -214,13 +220,22 @@ Format your response clearly with these two sections."""
 
     def analyze_for_flux_prompt(self, image: Image.Image) -> Tuple[str, Dict[str, Any]]:
         """Analyze image specifically for FLUX prompt generation"""
-        flux_prompt = """Analyze this image for FLUX generation and provide two sections:
+        flux_prompt = """You are analyzing a photograph for professional FLUX generation. 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.
+1. DESCRIPTION: Determine first if this is a real photograph, digital artwork, or illustration. Then create a detailed, flowing description starting directly with the subject. Be precise about:
+- Image type (photograph, illustration, artwork)
+- Color palette (specify if color or black/white, warm/cool tones, specific colors)
+- Photographic style (street, portrait, landscape, documentary, artistic, etc.)
+- Composition, lighting, mood, and atmosphere
+Write as a single coherent paragraph.
 
-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."""
+2. CAMERA_SETUP: Recommend specific professional equipment that would realistically capture this exact scene:
+- Street/urban scenes: Canon EOS R6, Sony A7 IV, Leica Q2 with 24-70mm f/2.8 or 35mm f/1.4
+- Portraits: Canon EOS R5, Sony A7R V, Hasselblad X2D with 85mm f/1.4 or 135mm f/2
+- Landscapes: Phase One XT, Fujifilm GFX 100S with 16-35mm f/2.8 or 40mm f/4
+- Documentary: Canon EOS-1D X, Sony A9 III with 24-105mm f/4 or 70-200mm f/2.8
+- Action/Sports: Canon EOS R3, Sony A1 with 300mm f/2.8 or 400mm f/2.8
+Match the equipment to what you actually observe in the scene type and shooting conditions."""
         
         return self.analyze_image(image, flux_prompt)