Update app/utils.py

app/utils.py

@@ -208,79 +208,79 @@ Respond only with the category name if found or 'None' if not found.
         
         return None
     
-def analyze_image(self, image, claude_api_key=None, user_allergies=None):
-    """تحليل الصورة للكشف عن الحساسيات مع مراعاة حساسيات المستخدم"""
-    try:
-        if not self.allergy_data:
-            raise ValueError("Allergy data not loaded")
-        
-        if not user_allergies:
-            raise ValueError("User allergies not provided")
-        
-        # استخراج النص من الصورة
-        extracted_text = self.ocr_model.process_image(image)
-        if extracted_text.startswith("Error processing image"):
-            raise ValueError(extracted_text)
-        
-        logger.info(f"Extracted text: {extracted_text[:200]}...")
-        
-        # تحويل النص إلى tokens
-        tokens = self.tokenize_text(extracted_text)
-        if not tokens:
-            raise ValueError("No tokens extracted from text")
-        
-        database_matches = {}
-        claude_matches = {}
-        
-        for token in tokens:
-            # البحث أولاً في قاعدة البيانات للحساسيات المحددة فقط
-            allergy = self.check_allergen_in_excel(token, user_allergies)
-            if allergy:
-                if allergy not in database_matches:
-                    database_matches[allergy] = set()  # استخدام set لمنع التكرار
-                database_matches[allergy].add(token)
-            elif claude_api_key:
-                # إذا لم يُوجد في ملف الإكسل، استدعِ Claude API للحساسيات المحددة فقط
-                allergy = self.check_allergy_risk(token, claude_api_key, user_allergies)
+    def analyze_image(self, image, claude_api_key=None, user_allergies=None):
+        """تحليل الصورة للكشف عن الحساسيات مع مراعاة حساسيات المستخدم"""
+        try:
+            if not self.allergy_data:
+                raise ValueError("Allergy data not loaded")
+            
+            if not user_allergies:
+                raise ValueError("User allergies not provided")
+            
+            # استخراج النص من الصورة
+            extracted_text = self.ocr_model.process_image(image)
+            if extracted_text.startswith("Error processing image"):
+                raise ValueError(extracted_text)
+            
+            logger.info(f"Extracted text: {extracted_text[:200]}...")
+            
+            # تحويل النص إلى tokens
+            tokens = self.tokenize_text(extracted_text)
+            if not tokens:
+                raise ValueError("No tokens extracted from text")
+            
+            database_matches = {}
+            claude_matches = {}
+            
+            for token in tokens:
+                # البحث أولاً في قاعدة البيانات للحساسيات المحددة فقط
+                allergy = self.check_allergen_in_excel(token, user_allergies)
                 if allergy:
-                    if allergy not in claude_matches:
-                        claude_matches[allergy] = set()  # استخدام set لمنع التكرار
-                    claude_matches[allergy].add(token)
-        
-        # إنشاء قائمة الحساسيات المكتشفة مع كل الكلمات المرتبطة بها
-        detected_allergens = []
-        seen_allergens = set()
-        
-        # إضافة الحساسيات من قاعدة البيانات أولاً
-        for allergy, words in database_matches.items():
-            if allergy not in seen_allergens:
-                detected_allergens.append({
-                    "allergen": allergy,
-                    "related_words": list(words)  # تحويل set إلى list
-                })
-                seen_allergens.add(allergy)
-        
-        # إضافة الحساسيات من Claude API
-        for allergy, words in claude_matches.items():
-            if allergy not in seen_allergens:
-                detected_allergens.append({
-                    "allergen": allergy,
-                    "related_words": list(words)  # تحويل set إلى list
-                })
-                seen_allergens.add(allergy)
+                    if allergy not in database_matches:
+                        database_matches[allergy] = set()  # استخدام set لمنع التكرار
+                    database_matches[allergy].add(token)
+                elif claude_api_key:
+                    # إذا لم يُوجد في ملف الإكسل، استدعِ Claude API للحساسيات المحددة فقط
+                    allergy = self.check_allergy_risk(token, claude_api_key, user_allergies)
+                    if allergy:
+                        if allergy not in claude_matches:
+                            claude_matches[allergy] = set()  # استخدام set لمنع التكرار
+                        claude_matches[allergy].add(token)
+            
+            # إنشاء قائمة الحساسيات المكتشفة مع كل الكلمات المرتبطة بها
+            detected_allergens = []
+            seen_allergens = set()
+            
+            # إضافة الحساسيات من قاعدة البيانات أولاً
+            for allergy, words in database_matches.items():
+                if allergy not in seen_allergens:
+                    detected_allergens.append({
+                        "allergen": allergy,
+                        "related_words": list(words)  # تحويل set إلى list
+                    })
+                    seen_allergens.add(allergy)
+            
+            # إضافة الحساسيات من Claude API
+            for allergy, words in claude_matches.items():
+                if allergy not in seen_allergens:
+                    detected_allergens.append({
+                        "allergen": allergy,
+                        "related_words": list(words)  # تحويل set إلى list
+                    })
+                    seen_allergens.add(allergy)
+            
+            return {
+                "extracted_text": extracted_text,
+                "detected_allergens": detected_allergens,
+                "database_matches": {k: list(v) for k, v in database_matches.items()},  # تحويل sets إلى lists
+                "claude_matches": {k: list(v) for k, v in claude_matches.items()},  # تحويل sets إلى lists
+                "analyzed_tokens": tokens,
+                "success": True
+            }
         
-        return {
-            "extracted_text": extracted_text,
-            "detected_allergens": detected_allergens,
-            "database_matches": {k: list(v) for k, v in database_matches.items()},  # تحويل sets إلى lists
-            "claude_matches": {k: list(v) for k, v in claude_matches.items()},  # تحويل sets إلى lists
-            "analyzed_tokens": tokens,
-            "success": True
-        }
-    
-    except Exception as e:
-        logger.error(f"Error analyzing image: {str(e)}", exc_info=True)
-        return {
-            "error": str(e),
-            "success": False
-        }
+        except Exception as e:
+            logger.error(f"Error analyzing image: {str(e)}", exc_info=True)
+            return {
+                "error": str(e),
+                "success": False
+            }