Update app/utils.py

app/utils.py

@@ -154,27 +154,29 @@ class AllergyAnalyzer:
             logger.error(f"Error tokenizing text: {str(e)}")
             return []
     
-    def check_allergen_in_excel(self, token):
-        """التحقق من وجود التوكن في ملف الإكسل"""
+    def check_allergen_in_excel(self, token, user_allergies):
+        """التحقق من وجود التوكن في ملف الإكسل مع مراعاة حساسيات المستخدم"""
         try:
             if not self.allergy_data:
                 return None
                 
             for allergy_name, ingredients in self.allergy_data.items():
-                if token in ingredients:
+                # نتحقق فقط من الحساسيات التي يهتم بها المستخدم
+                if allergy_name.lower() in user_allergies and token in ingredients:
                     return allergy_name
             return None
         except Exception as e:
             logger.error(f"Error checking allergen in Excel: {str(e)}")
             return None
     
-    def check_allergy_risk(self, ingredient, api_key):
-        """الاستعلام من Claude API عن الحساسيات"""
+    def check_allergy_risk(self, ingredient, api_key, user_allergies):
+        """الاستعلام من Claude API عن الحساسيات مع مراعاة حساسيات المستخدم"""
         try:
+            # نطلب من Claude التحقق فقط للحساسيات المحددة من المستخدم
             prompt = f"""
-You are a professional food safety expert. Analyze the ingredient '{ingredient}' and determine which of these allergen categories it belongs to: 
-dairy, eggs, peanuts, soy, tree nuts, wheat, fish, shellfish, sesame. 
-Respond only with the category name or 'None' if not found.
+You are a professional food safety expert. Analyze the ingredient '{ingredient}' and determine if it belongs to any of these allergen categories: 
+{', '.join(user_allergies)}. 
+Respond only with the category name if found or 'None' if not found.
 """
             url = "https://api.anthropic.com/v1/messages"
             headers = {
@@ -196,8 +198,8 @@ Respond only with the category name or 'None' if not found.
             
             if "content" in response_json and isinstance(response_json["content"], list):
                 result = response_json["content"][0]["text"].strip().lower()
-                # التحقق من أن النتيجة هي واحدة من الحساسيات المعروفة
-                if result in self.allergy_data:
+                # نتحقق فقط من الحساسيات التي يهتم بها المستخدم
+                if result in user_allergies:
                     return result
                 return None
                 
@@ -206,12 +208,15 @@ Respond only with the category name or 'None' if not found.
         
         return None
     
-    def analyze_image(self, image, claude_api_key=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"):
@@ -228,15 +233,15 @@ Respond only with the category name or 'None' if not found.
             claude_matches = {}
             
             for token in tokens:
-                # البحث أولاً في قاعدة البيانات
-                allergy = self.check_allergen_in_excel(token)
+                # البحث أولاً في قاعدة البيانات للحساسيات المحددة فقط
+                allergy = self.check_allergen_in_excel(token, user_allergies)
                 if allergy:
                     if allergy not in database_matches:
                         database_matches[allergy] = []
                     database_matches[allergy].append(token)
                 elif claude_api_key:
-                    # إذا لم يُوجد في ملف الإكسل، استدعِ Claude API
-                    allergy = self.check_allergy_risk(token, 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] = []