Update app/utils.py

app/utils.py

@@ -10,6 +10,7 @@ import nltk
 import requests
 import io
 from flask import current_app
+import base64
 
 logger = logging.getLogger(__name__)
 
@@ -144,7 +145,7 @@ class AllergyAnalyzer:
                 results.append(allergy)
         return results
     
-    def analyze_image(self, image, user_allergens):
+    def analyze_image(self, image, user_allergens, claude_api_key=None):
         """تحليل الصورة مباشرة للكشف عن الحساسيات"""
         try:
             # استخراج النص من الصورة
@@ -167,10 +168,9 @@ class AllergyAnalyzer:
                         database_matches[allergy] = database_matches.get(allergy, []) + [token]
                 else:
                     # إذا لم توجد في قاعدة البيانات، نستخدم Claude API
-                    claude_api_key = current_app.config.get('CLAUDE_API_KEY')
                     if claude_api_key:
                         for allergy in user_allergens:
-                            if self.check_claude_allergens(token, allergy, claude_api_key):
+                            if self.check_claude_allergens(token, allergy, claude_api_key, image):
                                 detected_allergens.add(allergy)
                                 claude_matches[allergy] = claude_matches.get(allergy, []) + [token]
             
@@ -191,7 +191,7 @@ class AllergyAnalyzer:
                 "error": str(e)
             }
     
-    def check_claude_allergens(self, token, allergy, api_key):
+    def check_claude_allergens(self, token, allergy, api_key, image):
         """الاستعلام من Claude API عن الحساسيات"""
         try:
             # تحضير الصورة للطلب
@@ -200,10 +200,10 @@ class AllergyAnalyzer:
             img_byte_arr = img_byte_arr.getvalue()
             
             prompt = f"""
-Analyze if this product contains or is derived from {allergy}. 
-Focus on the ingredient: {token}.
-Respond ONLY with 'Yes' or 'No'. No explanations.
-"""
+    Analyze if this product contains or is derived from {allergy}. 
+    Focus on the ingredient: {token}.
+    Respond ONLY with 'Yes' or 'No'. No explanations.
+    """
             
             url = "https://api.anthropic.com/v1/messages"
             headers = {
@@ -245,4 +245,4 @@ Respond ONLY with 'Yes' or 'No'. No explanations.
             
         except Exception as e:
             logger.error(f"Error querying Claude API: {str(e)}")
-            return
+            return False