Update app.py

app.py

@@ -1273,6 +1273,9 @@ def yolo11_fast_classify(img, declared_text):
     # ═══════════════════════════════════════════════════════
     # خريطة الكلمات المفتاحية → اسم الصنف في قاعدة البيانات
     DECLARED_KEYWORD_MAP = {
+        # موز
+        'banana': 'banana', 'bananas': 'banana', 'موز': 'banana',
+        'indian banana': 'banana', 'cavendish': 'banana',
         # أرز
         'rice': 'rice', 'أرز': 'rice', 'رز': 'rice',
         'sella': 'rice', 'basmati': 'rice', 'بسمتي': 'rice',
@@ -1285,16 +1288,32 @@ def yolo11_fast_classify(img, declared_text):
         'oil': 'oil', 'زيت': 'oil',
         # شاي
         'tea': 'tea', 'شاي': 'tea',
+        # فواكه وخضروات
+        'apple': 'fruits', 'apples': 'fruits', 'تفاح': 'fruits',
+        'orange': 'fruits', 'oranges': 'fruits', 'برتقال': 'fruits',
+        'grape': 'fruits', 'grapes': 'fruits', 'عنب': 'fruits',
+        'date': 'fruits', 'dates': 'fruits', 'تمر': 'fruits',
+        'vegetable': 'vegetables', 'vegetables': 'vegetables', 'خضروات': 'vegetables',
         # بلاستيك
         'plastic': 'plastic', 'بلاستيك': 'plastic',
         # أقمشة
         'fabric': 'fabric', 'textile': 'fabric', 'قماش': 'fabric',
+        'garment': 'clothes', 'garments': 'clothes', 'ملابس': 'clothes',
+        'clothes': 'clothes', 'clothing': 'clothes',
         # أحذية
         'shoes': 'shoes', 'footwear': 'shoes', 'أحذية': 'shoes',
         # إطارات
         'tires': 'tires', 'tyres': 'tires', 'إطارات': 'tires',
         # وقود
         'fuel': 'fuel', 'diesel': 'fuel', 'وقود': 'fuel',
+        # أدوية
+        'medicine': 'medicine', 'medicines': 'medicine', 'pharmaceutical': 'medicine',
+        'أدوية': 'medicine', 'دواء': 'medicine',
+        # إلكترونيات
+        'electronics': 'electronics', 'electronic': 'electronics', 'إلكترونيات': 'electronics',
+        # مواد بناء
+        'ceramic': 'ceramics', 'tiles': 'ceramics', 'سيراميك': 'ceramics',
+        'steel': 'steel', 'iron': 'steel', 'حديد': 'steel', 'فولاذ': 'steel',
     }
 
     override_applied = False
@@ -1322,21 +1341,23 @@ def yolo11_fast_classify(img, declared_text):
 
             # شروط تطبيق الـ Override:
             # 1. YOLO لا يوافق البيان
-            # 2. ثقة YOLO أقل من 75%
-            if not yolo_agrees and top_yolo_conf < 0.75:
+            # 2. ✅ ثقة YOLO أقل من 85% (صور السونار صعبة، YOLO قد يُخطئ بثقة 52-80%)
+            if not yolo_agrees and top_yolo_conf < 0.85:
                 override_applied = True
                 override_reason = (
                     f"البيان: '{declared_text}' | "
-                    f"YOLO أعطى: {top_yolo_name} ({top_yolo_conf:.1%}) < 75% | "
+                    f"YOLO أعطى: {top_yolo_name} ({top_yolo_conf:.1%}) < 85% | "
                     f"Override → {matched_category}"
                 )
                 print(f"\n🔄 OVERRIDE: {override_reason}")
-                # استبدل النتيجة بالبيان
-                items_list = [(matched_category, 0.60)]  # ثقة 60% للـ override
+                # ✅ البضاعة المصرّحة أولاً + باقي نتائج YOLO كمشبوهة
+                declared_item = (matched_category, 0.70)
+                other_items = [(n, c) for n, c in items_list[1:] if n != matched_category]
+                items_list = [declared_item] + other_items[:3]
 
-            elif not yolo_agrees and top_yolo_conf >= 0.75:
-                print(f"\n⚠️ تعارض: البيان='{matched_keyword}' لكن YOLO واثق {top_yolo_conf:.1%} من {top_yolo_name}")
-                print(f"   → نحتفظ بنتيجة YOLO لأن ثقته عالية")
+            elif not yolo_agrees and top_yolo_conf >= 0.85:
+                print(f"\n⚠️ تعارض قوي: البيان='{matched_keyword}' لكن YOLO واثق {top_yolo_conf:.1%} من {top_yolo_name}")
+                print(f"   → نحتفظ بنتيجة YOLO (مشبوه!)")
 
             else:
                 print(f"\n✅ YOLO يوافق البيان: {matched_category} ({top_yolo_conf:.1%})")
@@ -2842,25 +2863,19 @@ def api_analyze(img, declared_text=""):
     return json.dumps(result, ensure_ascii=False)
 
 if __name__ == "__main__":
-    # ═══════════════════════════════════════════════════════════════
-    # 🔄 Keep-Alive Thread — يمنع Hugging Face Space من النوم
-    # يرسل ping كل 4 دقائق لإبقاء الـ Space نشطاً 24/7
-    # ═════════════════════════════════════════════════���═════════════
+    # 🔄 Keep-Alive — يمنع Hugging Face Space من النوم
     def keep_alive():
         import requests as _req
-        url = "http://localhost:7860/"
-        time_module.sleep(30)  # انتظر حتى يبدأ الـ server
+        time_module.sleep(30)
         while True:
             try:
-                _req.get(url, timeout=5)
-                print("💓 Keep-Alive ping OK")
+                _req.get("http://localhost:7860/", timeout=5)
+                print("💓 Keep-Alive OK")
             except Exception as e:
-                print(f"💓 Keep-Alive ping: {e}")
-            time_module.sleep(240)  # كل 4 دقائق
-
-    ka_thread = threading.Thread(target=keep_alive, daemon=True, name="KeepAlive")
-    ka_thread.start()
-    print("💓 Keep-Alive thread started (ping every 4 min)")
+                print(f"💓 Keep-Alive: {e}")
+            time_module.sleep(240)
+    threading.Thread(target=keep_alive, daemon=True, name="KeepAlive").start()
+    print("💓 Keep-Alive started (every 4 min)")
     print(f"🔱 SONAR-AI v{VERSION}")
     print(f"📦 Cargo: {len(CARGO_DATABASE)} | TSC: {len(TSC_DATABASE)}")
     print(f"🤖 Models: {len(CLASSIFICATION_MODELS)} cls + {'det' if DETECTION_MODEL else 'no det'}")