Update models.py

models.py

@@ -97,53 +97,36 @@ class BagelAPIAnalyzer(BaseImageAnalyzer):
             from professional_photography import EXPERT_PHOTOGRAPHY_KNOWLEDGE
             
             # Create the teaching prompt with the complete structure
-            prompt = f"""Analiza esta imagen con conocimiento cinematográfico profesional completo. Estructura tu respuesta así:
+            prompt = f"""Analyze this image using complete professional cinematography knowledge.
 
-GRAMÁTICA: [PLANO] of [SUJETO] [ACCIÓN] [CONTEXTO], [ILUMINACIÓN], [COMPOSICIÓN], shot on [CÁMARA], [LENTE], [CONFIGURACIÓN]
+STRUCTURE: [PLANE] of [SUBJECT] [ACTION] [CONTEXT], [LIGHTING], [COMPOSITION], shot on [CAMERA], [LENS], [SETTINGS]
 
-BUSCA ESTOS CAMPOS:
+PLANE: From {EXPERT_PHOTOGRAPHY_KNOWLEDGE.get('photographic_planes', {})}
+SUBJECT + ACTION: Define accurately what you see
+CONTEXT: Define the environment accurately
+LIGHTING: From {EXPERT_PHOTOGRAPHY_KNOWLEDGE.get('lighting_principles', {})}  
+COMPOSITION: From {EXPERT_PHOTOGRAPHY_KNOWLEDGE.get('composition_rules', {})}
+CAMERA ANGLES: From {EXPERT_PHOTOGRAPHY_KNOWLEDGE.get('camera_angles', {})}
+TECHNICAL SETUP: From {EXPERT_PHOTOGRAPHY_KNOWLEDGE.get('scene_types', {})}
 
-PLANO: Consulta photographic_planes completo:
-{EXPERT_PHOTOGRAPHY_KNOWLEDGE.get('photographic_planes', {})}
-- Encuentra el plano que coincida exactamente
-
-SUJETO + ACCIÓN: Describe exactamente lo que ves - persona, objeto, qué está haciendo
-
-CONTEXTO: Describe el ambiente/entorno completamente - ubicación, elementos del fondo
-
-ILUMINACIÓN: Consulta lighting_principles completo:
-{EXPERT_PHOTOGRAPHY_KNOWLEDGE.get('lighting_principles', {})}
-- Encuentra el tipo de iluminación que coincida
-
-COMPOSICIÓN: Consulta composition_rules completo:
-{EXPERT_PHOTOGRAPHY_KNOWLEDGE.get('composition_rules', {})}
-- Encuentra la técnica de composición que se aplica
-
-CONFIGURACIÓN TÉCNICA: Consulta scene_types completo:
-{EXPERT_PHOTOGRAPHY_KNOWLEDGE.get('scene_types', {})}
-- Encuentra el setup de cámara apropiado para el tipo de escena
-
-USA TODA LA BASE DE DATOS EXPERT_PHOTOGRAPHY_KNOWLEDGE COMPLETA.
-
-Completa cada campo con la información específica que encuentres. Responde en el formato de gramática especificado."""
+Complete the structure using the appropriate elements from each section."""
             
             return prompt
             
         except Exception as e:
             logger.warning(f"Professional knowledge base access failed: {e}")
-            return """Analiza esta imagen con conocimiento cinematográfico profesional. Estructura tu respuesta así:
+            return """Analyze this image using complete professional cinematography knowledge.
 
-GRAMÁTICA: [PLANO] of [SUJETO] [ACCIÓN] [CONTEXTO], [ILUMINACIÓN], [COMPOSICIÓN], shot on [CÁMARA], [LENTE], [CONFIGURACIÓN]
+STRUCTURE: [PLANE] of [SUBJECT] [ACTION] [CONTEXT], [LIGHTING], [COMPOSITION], shot on [CAMERA], [LENS], [SETTINGS]
 
-BUSCA Y COMPLETA CADA CAMPO:
-- PLANO: wide shot, medium shot, close-up, etc.
-- SUJETO + ACCIÓN: exactamente lo que ves
-- CONTEXTO: ambiente y entorno
-- ILUMINACIÓN: tipo de luz
-- COMPOSICIÓN: técnica utilizada  
-- CONFIGURACIÓN TÉCNICA: cámara y lente apropiados
+PLANE: wide_shot, medium_shot, close_up, extreme_wide_shot, extreme_close_up, detail_shot
+SUBJECT + ACTION: Define accurately what you see
+CONTEXT: Define the environment accurately
+LIGHTING: golden_hour, natural_daylight, dramatic_lighting, soft_natural, blue_hour, studio_lighting
+COMPOSITION: rule_of_thirds, leading_lines, symmetrical, centered_composition, dynamic_composition
+TECHNICAL SETUP: Professional camera and lens specifications
 
-Usa tu conocimiento profesional completo para completar cada campo."""
+Complete the structure using the appropriate elements."""
 
     def _save_temp_image(self, image: Image.Image) -> str:
         """Save image to temporary file for API call"""
@@ -202,13 +185,13 @@ Usa tu conocimiento profesional completo para completar cada campo."""
             
             logger.info("Calling BAGEL API with professional_photography.py prompt...")
             
-            # Call BAGEL API with professional prompt - let it do its work
+            # Call BAGEL API with professional prompt - FORCE NEW READING
             result = self.client.predict(
                 image=handle_file(temp_path),
                 prompt=prompt,
                 show_thinking=False,
-                do_sample=True,  # Allow creativity
-                text_temperature=0.7,  # Higher temperature for richer descriptions
+                do_sample=True,  # Allow creativity and variation
+                text_temperature=0.8,  # Higher temperature for different responses each time
                 max_new_tokens=1024,  # More tokens for detailed analysis
                 api_name=self.api_endpoint
             )