Malaji71 commited on
Commit
785cf73
·
verified ·
1 Parent(s): 4ab3467

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +2 -9
models.py CHANGED
@@ -101,7 +101,7 @@ class BagelAPIAnalyzer(BaseImageAnalyzer):
101
  try:
102
  # Default prompt for detailed image analysis
103
  if prompt is None:
104
- prompt = "Provide a detailed description of this image, including objects, people, setting, composition, lighting, colors, mood, and artistic style. Focus on elements that would be useful for generating a similar image."
105
 
106
  # Save image to temporary file
107
  temp_path = self._save_temp_image(image)
@@ -158,14 +158,7 @@ class BagelAPIAnalyzer(BaseImageAnalyzer):
158
 
159
  def analyze_for_flux_prompt(self, image: Image.Image) -> Tuple[str, Dict[str, Any]]:
160
  """Analyze image specifically for FLUX prompt generation"""
161
- flux_prompt = """Analyze this image and generate a detailed FLUX prompt description. Focus on:
162
- - Photographic and artistic style
163
- - Composition and framing
164
- - Lighting conditions and mood
165
- - Colors and visual elements
166
- - Camera settings that would recreate this image
167
- - Technical photography details
168
- Provide a comprehensive description suitable for FLUX image generation."""
169
 
170
  return self.analyze_image(image, flux_prompt)
171
 
 
101
  try:
102
  # Default prompt for detailed image analysis
103
  if prompt is None:
104
+ 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."
105
 
106
  # Save image to temporary file
107
  temp_path = self._save_temp_image(image)
 
158
 
159
  def analyze_for_flux_prompt(self, image: Image.Image) -> Tuple[str, Dict[str, Any]]:
160
  """Analyze image specifically for FLUX prompt generation"""
161
+ 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."""
 
 
 
 
 
 
 
162
 
163
  return self.analyze_image(image, flux_prompt)
164