modificari mari

backend/ai_vision.py

@@ -73,8 +73,18 @@ def extract_label_info(image_bytes: bytes, mode: str = "item"):
                     # 3. Final fallback to hardcoded default
                     prompt = "Extract technical specs. Return JSON with name, category, description, connector, size, color, part_number, ocr_text, quantity."
 
-        # 1. Try Gemini
-        result = gemini.extract(image_bytes, prompt)
+        # 0. Get Active Provider from DB
+        provider_setting = db.query(models.SystemSetting).filter(models.SystemSetting.key == "ai_provider").first()
+        active_provider = provider_setting.value if provider_setting else "gemini"
+        
+        # 1. Execute extraction based on selection
+        result = None
+        if active_provider == "claude":
+            print(f"📡 Using Anthropic Claude for extraction...")
+            result = claude.extract(image_bytes, prompt)
+        else:
+            print(f"📡 Using Google Gemini for extraction...")
+            result = gemini.extract(image_bytes, prompt)
         
         if result:
             # Check if AI returned a list of items or a singular object