Switch to Gemini Pro to bypass Flash 404 errors

agents/visual_analyst.py

@@ -14,23 +14,22 @@ class VisualAnalyst:
 
         genai.configure(api_key=self.api_key)
         
-        # --- SMART MODEL SELECTION ---
-        print("🔍 Checking available Gemini models for this key...")
+        print("🔍 Checking available Gemini models...")
         try:
             # Filter for models that support content generation
             my_models = [m.name for m in genai.list_models() if 'generateContent' in m.supported_generation_methods]
             print(f"📋 Available Models: {my_models}")
             
-            # UPDATED PRIORITY: Specific versions FIRST
+            # UPDATED PRIORITY: Force 'Pro' model to bypass Flash 404s
             preferred_order = [
-                'models/gemini-1.5-flash-001',  # <--- Specific version (Safest)
-                'models/gemini-1.5-flash',      # Alias (Fallback)
+                'models/gemini-1.5-pro',      # <--- The Heavy Hitter (Smarter, diff backend)
                 'models/gemini-1.5-pro-001',
-                'models/gemini-1.5-pro',
+                'models/gemini-1.5-flash',
+                'models/gemini-1.5-flash-001',
                 'models/gemini-pro-vision'
             ]
             
-            selected_model = "models/gemini-1.5-flash-001" # Default safety
+            selected_model = "models/gemini-1.5-pro" # Default
             
             for model_name in preferred_order:
                 if model_name in my_models:
@@ -41,8 +40,8 @@ class VisualAnalyst:
             self.model = genai.GenerativeModel(selected_model)
             
         except Exception as e:
-            print(f"⚠️ Model list failed ({e}), defaulting to gemini-1.5-flash-001")
-            self.model = genai.GenerativeModel('gemini-1.5-flash-001')
+            print(f"⚠️ Model list failed ({e}), defaulting to gemini-1.5-pro")
+            self.model = genai.GenerativeModel('gemini-1.5-pro')
 
     async def analyze_image(self, image_path: str):
         # Adaptation: Read file path to bytes, as main.py passes a path
@@ -62,6 +61,7 @@ class VisualAnalyst:
             "Return a JSON object with keys: main_color, product_type, design_style, visual_features."
         )
         try:
+            # Note: Pro model is sometimes stricter with image formats, but 'parts' usually works.
             # Adaptation: Run in thread to allow async await
             response = await asyncio.to_thread(
                 self.model.generate_content,