Update main.py

main.py

@@ -1,12 +1,13 @@
 """
-Saudi Legal AI API — v4.0
-FastAPI + RAG + Groq + Gemini
+Saudi Legal AI API — v4.1
+FastAPI + RAG + Groq + Gemini Multi-Model
 """
 import os, gc, re, time, logging
 from collections import deque, defaultdict
 from datetime import datetime
 from contextlib import asynccontextmanager
 from dotenv import load_dotenv
+from typing import Dict
 
 from fastapi import FastAPI, HTTPException, Request
 from fastapi.middleware.cors import CORSMiddleware
@@ -31,67 +32,68 @@ logger = logging.getLogger(__name__)
 # ══════════════════════════════════════════════════════════
 # Config
 # ══════════════════════════════════════════════════════════
-GROQ_API_KEY     = os.getenv('GROQ_API_KEY', '')
-HF_TOKEN         = os.getenv('HF_TOKEN', '')
-GEMINI_KEY       = os.getenv('GEMINI_API_KEY', '')
-HF_REPO_ID       = os.getenv('HF_REPO_ID', 'WafaaFraih/saudi-legal-moj')
-CHROMA_PATH      = os.getenv('CHROMA_PATH', './chroma_db')
+GROQ_API_KEY  = os.getenv('GROQ_API_KEY', '')
+HF_TOKEN      = os.getenv('HF_TOKEN', '')
+GEMINI_KEY    = os.getenv('GEMINI_API_KEY', '')
+HF_REPO_ID    = os.getenv('HF_REPO_ID', 'WafaaFraih/saudi-legal-moj')
+CHROMA_PATH   = os.getenv('CHROMA_PATH', './chroma_db')
 
 # ══════════════════════════════════════════════════════════
 # Global State
 # ══════════════════════════════════════════════════════════
-vectorstore       = None
-bm25_index        = None
-bm25_texts        = []
-bm25_metadatas    = []
-embeddings        = None
-ACTIVE_MODELS     = []
-GEMINI_AVAILABLE  = False
-gemini_model      = None
-groq_client       = None
-request_log       = deque(maxlen=500)
-stats             = {'total': 0, 'success': 0, 'blocked': 0, 'errors': 0}
-active_ips        = {}
-_expansion_cache  = {}
-_rewrite_cache    = {}
+vectorstore        = None
+bm25_index         = None
+bm25_texts         = []
+bm25_metadatas     = []
+embeddings         = None
+ACTIVE_MODELS      = []
+GEMINI_AVAILABLE   = False
+gemini_clients: Dict = {}
+groq_client        = None
+request_log        = deque(maxlen=500)
+stats              = {'total': 0, 'success': 0, 'blocked': 0, 'errors': 0}
+active_ips         = {}
+_expansion_cache   = {}
+_rewrite_cache     = {}
 _translation_cache = {}
 
+GEMINI_MODELS = [
+    'models/gemini-2.5-flash',
+    'models/gemini-2.0-flash',
+    'models/gemma-3-27b-it',
+]
+
 # ══════════════════════════════════════════════════════════
 # Extra Laws
 # ══════════════════════════════════════════════════════════
 EXTRA_LAWS = [
-    # نظام العمل
-    {'text':'المادة الثالثة والثمانون: عند انتهاء عقد العمل يستحق العامل مكافأة عن مدة خدمته تحسب على أساس أجر نصف شهر عن كل سنة من السنوات الخمس الأولى، وأجر شهر عن كل سنة بعد ذلك. وتحسب المكافأة على أساس آخر أجر تقاضاه العامل. ويستحق العامل مكافأة عن كسور السنة بنسبة ما قضاه منها في العمل.',
+    {'text':'المادة الثالثة والثمانون: عند انتهاء عقد العمل يستحق العامل مكافأة عن مدة خدمته تحسب على أساس أجر نصف شهر عن كل سنة من السنوات الخمس الأولى، وأجر شهر عن كل سنة بعد ذلك. وتحسب المكافأة على أساس آخر أجر تقاضاه العامل.',
      'article_number':'المادة الثالثة والثمانون','law_name':'نظام العمل','law_type':'نظام','source':'hrsd.gov.sa'},
-    {'text':'المادة الثامنة والثمانون: إذا أنهى صاحب العمل عقد العمل غير المحدد المدة دون سبب مشروع وجب عليه أن يدفع للعامل تعويضاً عن الفصل التعسفي يعادل أجر خمسة عشر يوماً عن كل سنة من سنوات خدمة العامل، ولا يقل التعويض في جميع الأحوال عن أجر شهرين.',
+    {'text':'المادة الثامنة والثمانون: إذا أنهى صاحب العمل عقد العمل غير المحدد المدة دون سبب مشروع وجب عليه أن يدفع للعامل تعويضاً عن الفصل التعسفي يعادل أجر خمسة عشر يوماً عن كل سنة من سنوات خدمة العامل، ولا ي��ل التعويض عن أجر شهرين.',
      'article_number':'المادة الثامنة والثمانون','law_name':'نظام العمل','law_type':'نظام','source':'hrsd.gov.sa'},
-    {'text':'المادة التاسعة والثمانون: لا يجوز لصاحب العمل فصل العامل بسبب تقدمه بشكوى أو مطالبة بحقوقه. وإذا أثبت العامل أن الفصل كان تعسفياً وجب على صاحب العمل دفع تعويض عادل إضافة إلى مكافأة نهاية الخدمة وسائر مستحقاته.',
+    {'text':'المادة التاسعة والثمانون: لا يجوز لصاحب العمل فصل العامل بسبب تقدمه بشكوى. وإذا أثبت العامل أن الفصل كان تعسفياً وجب على صاحب العمل دفع تعويض عادل إضافة إلى مكافأة نهاية الخدمة وسائر مستحقاته.',
      'article_number':'المادة التاسعة والثمانون','law_name':'نظام العمل','law_type':'نظام','source':'hrsd.gov.sa'},
-    {'text':'المادة الخامسة والستون: لا يجوز تشغيل العامل تشغيلاً فعلياً أكثر من ثماني ساعات في اليوم أو ثماني وأربعين ساعة في الأسبوع. وتنخفض ساعات العمل في شهر رمضان إلى ست ساعات في اليوم للعمال المسلمين.',
+    {'text':'المادة الخامسة والستون: لا يجوز تشغيل العامل تشغيلاً فعلياً أكثر من ثماني ساعات في اليوم أو ثماني وأربعين ساعة في الأسبوع.',
      'article_number':'المادة الخامسة والستون','law_name':'نظام العمل','law_type':'نظام','source':'hrsd.gov.sa'},
     {'text':'المادة الثامنة والستون: العمل الإضافي هو كل عمل يؤديه العامل زيادة على ساعات العمل النظامية. ويستحق العامل عن ساعات العمل الإضافية أجراً إضافياً لا يقل عن أجره الأصلي مضافاً إليه خمسون بالمئة.',
      'article_number':'المادة الثامنة والستون','law_name':'نظام العمل','law_type':'نظام','source':'hrsd.gov.sa'},
-    {'text':'المادة الثالثة والستون: مدة إجازة العامل السنوية واحد وعشرون يوماً تزاد إلى ثلاثين يوماً إذا أمضى العامل خمس سنوات متواصلة في خدمة صاحب عمل واحد.',
+    {'text':'المادة الثالثة والستون: مدة إجازة العامل السنوية واحد وعشرون يوماً تزاد إلى ثلاثين يوماً إذا أمضى العامل خمس سنوات متواصلة.',
      'article_number':'المادة الثالثة والستون','law_name':'نظام العمل','law_type':'نظام','source':'hrsd.gov.sa'},
-    {'text':'المادة الثالثة والثلاثون: تستحق العاملة إجازة وضع بأجر كامل مدتها عشرة أسابيع. ويُحظر تشغيل المرأة في الأسابيع الستة التالية للوضع مباشرة.',
+    {'text':'المادة الثالثة والثلاثون: تستحق العاملة إجازة وضع بأجر كامل مدتها عشرة أسابيع. ويُحظر تشغيل المرأة في الأسابيع الستة التالية للوضع.',
      'article_number':'المادة الثالثة والثلاثون','law_name':'نظام العمل','law_type':'نظام','source':'hrsd.gov.sa'},
-    {'text':'المادة الخامسة والسبعون: إذا أراد أحد طرفي عقد العمل غير المحدد المدة إنهاءه وجب عليه إخطار الطرف الآخر كتابةً قبل الإنهاء بمدة لا تقل عن ستين يوماً إذا كان أجر العامل يدفع شهرياً.',
+    {'text':'المادة الخامسة والسبعون: إذا أراد أحد طرفي عقد العمل غير المحدد المدة إنهاءه وجب عليه إخطار الطرف الآخر كتابةً قبل الإنهاء بستين يوماً إذا كان أجر العامل يدفع شهرياً.',
      'article_number':'المادة الخامسة والسبعون','law_name':'نظام العمل','law_type':'نظام','source':'hrsd.gov.sa'},
-    # نظام القضاء
-    {'text':'المادة الثانية والثلاثون: يشترط فيمن يُعيَّن في وظائف القضاء أن يكون سعودي الجنسية بالأصل، وأن يكون مسلماً، وأن يكون حاصلاً على شهادة الأهلية في الشريعة الإسلامية من المعهد العالي للقضاء أو ما يعادلها، وأن ينجح في الامتحان المقرر، وأن يكون حسن السيرة والسلوك، وألا يكون محكوماً عليه في جريمة مخلة بالشرف.',
+    {'text':'المادة الثانية والثلاثون: يشترط فيمن يُعيَّن في وظائف القضاء أن يكون سعودي الجنسية بالأصل، وأن يكون مسلماً، وأن يكون حاصلاً على شهادة الأهلية في الشريعة الإسلامية أو ما يعادلها، وأن ينجح في الامتحان المقرر.',
      'article_number':'المادة الثانية والثلاثون','law_name':'نظام القضاء','law_type':'نظام','source':'boe.gov.sa'},
-    {'text':'المادة الحادية والثلاثون: يُعيَّن القضاة بمرسوم ملكي بناءً على توصية المجلس الأعلى للقضاء. ويشترط في القاضي: أن يكون سعودي الجنسية، متمتعاً بالأهلية الكاملة، حاصلاً على مؤهل شرعي، ناجحاً في الاختبار، غير محكوم عليه.',
+    {'text':'المادة الحادية والثلاثون: يُعيَّن القضاة بمرسوم ملكي. ويشترط في القاضي: أن يكون سعودي الجنسية، متمتعاً بالأهلية الكاملة، حاصلاً على مؤهل شرعي، ناجحاً في الاختبار.',
      'article_number':'المادة الحادية والثلاثون','law_name':'نظام القضاء','law_type':'نظام','source':'boe.gov.sa'},
-    # نظام المحاماة
-    {'text':'المادة الثالثة: يشترط فيمن يزاول مهنة المحاماة أن يكون: أولاً سعودي الجنسية. ثانياً حاصلاً على شهادة البكالوريوس في تخصص الشريعة الإسلامية أو الأنظمة أو ما يعادلها. ثالثاً اسمه مقيداً في جدول المحامين الممارسين. رابعاً ألا يكون محكوماً عليه في جريمة مخلة بالشرف.',
+    {'text':'المادة الثالثة: يشترط فيمن يزاول مهنة المحاماة أن يكون: سعودي الجنسية، حاصلاً على شهادة البكالوريوس في الشريعة الإسلامية أو الأنظمة، اسمه مقيداً في جدول المحامين الممارسين.',
      'article_number':'المادة الثالثة','law_name':'نظام المحاماة','law_type':'نظام','source':'boe.gov.sa'},
-    # نظام الجرائم المعلوماتية
-    {'text':'المادة الثالثة: يعاقب بالسجن مدة لا تزيد على سنة وبغرامة لا تزيد على خمسمائة ألف ريال كل شخص يرتكب جريمة الدخول غير المشروع إلى موقع إلكتروني أو التشهير بالآخرين عبر وسائل التقنية المعلوماتية.',
+    {'text':'المادة الثالثة: يعاقب بالسجن مدة لا تزيد على سنة وبغرامة لا تزيد على خمسمائة ألف ريال كل شخص يرتكب جريمة الدخول غير المشروع إلى موقع إلكتروني أو التشهير عبر التقنية.',
      'article_number':'المادة الثالثة','law_name':'نظام مكافحة الجرائم المعلوماتية','law_type':'نظام','source':'boe.gov.sa'},
     {'text':'المادة السادسة: يعاقب بالسجن مدة لا تزيد على خمس سنوات وبغرامة لا تزيد على ثلاثة ملايين ريال كل شخص يرتكب جريمة اختراق الأنظمة الحاسوبية الحكومية.',
      'article_number':'المادة السادسة','law_name':'نظام مكافحة الجرائم المعلوماتية','law_type':'نظام','source':'boe.gov.sa'},
-    # نظام حماية البيانات
-    {'text':'المادة الرابعة: لا يجوز معالجة البيانات الشخصية إلا لتحقيق الغرض المشروع والمحدد الذي جُمعت من أجله. ويجب الحصول على موافقة صريحة من صاحب البيانات قبل معالجتها.',
+    {'text':'المادة الرابعة: لا يجوز معالجة البيانات الشخصية إلا لتحقيق الغرض المشروع الذي جُمعت من أجله مع الحصول على موافقة صريحة.',
      'article_number':'المادة الرابعة','law_name':'نظام حماية البيانات الشخصية','law_type':'نظام','source':'boe.gov.sa'},
     {'text':'المادة التاسعة والعشرون: يعاقب على الإفصاح عن البيانات الشخصية دون وجه حق بالسجن مدة لا تزيد على سنتين وبغرامة لا تزيد على ثلاثة ملايين ريال.',
      'article_number':'المادة التاسعة والعشرون','law_name':'نظام حماية البيانات الشخصية','law_type':'نظام','source':'boe.gov.sa'},
@@ -119,16 +121,12 @@ NUMBERS_AR = {
 }
 
 ENGLISH_TO_ARABIC = {
-    # ── جمل كاملة (ال��طول أولاً) ──
     'conditions for lawyer license':  'شروط مزاولة مهنة المحاماة',
     'conditons for laywer license':   'شروط مزاولة مهنة المحاماة',
     'conditions for lawyer':          'شروط مزاولة مهنة المحاماة',
-    'conditons for laywer':           'شروط مزاولة مهنة المحاماة',
     'what are the conditions':        'ما هي شروط',
-    'what are conditons':             'ما هي شروط',
     'what is the penalty':            'ما هي عقوبة',
     'what is the penaly':             'ما هي عقوبة',
-    'what is the punishment':         'ما هي عقوبة',
     'what are my rights':             'ما هي حقوقي',
     'how to calculate':               'كيف تحسب',
     'how is calculated':              'كيف تحسب',
@@ -137,71 +135,56 @@ ENGLISH_TO_ARABIC = {
     'i was fired':                    'تم فصلي',
     'i got fired':                    'تم فصلي',
     'am i entitled':                  'هل يحق لي',
-    'am i eligible':                  'هل يحق لي',
     'end of service':                 'مكافأة نهاية الخدمة',
     'end of servise':                 'مكافأة نهاية الخدمة',
-    'end of serivce':                 'مكافأة نهاية الخدمة',
     'money laundering':               'غسل الأموال',
     'money laudering':                'غسل الأموال',
     'mony laundering':                'غسل الأموال',
-    'money laundring':                'غسل الأموال',
     'moeny laundering':               'غسل الأموال',
     'wrongful termination':           'فصل تعسفي',
-    'wrongful terminaton':            'فصل تعسفي',
     'unfair dismissal':               'فصل تعسفي',
-    'wrongful dismissal':             'فصل تعسفي',
     'working hours':                  'ساعات العمل',
-    'work hours':                     'ساعات العمل',
     'annual leave':                   'الإجازة السنوية',
     'maternity leave':                'إجازة الأمومة',
     'notice period':                  'مهلة الإشعار',
     'data protection':                'حماية البيانات الشخصية',
-    'personal data':                  'البيانات الشخصية',
     'labor law':                      'نظام العمل',
     'labour law':                     'نظام العمل',
     'my employer':                    'صاحب العمل',
-    'my company':                     'صاحب العمل',
     'my rights':                      'حقوقي',
     'if fired':                       'عند الفصل',
-    # ── كلمات مفردة ──
-    'gratuity':      'مكافأة نهاية الخدمة',
-    'indemnity':     'تعويض نهاية الخدمة',
-    'wrongful':      'تعسفي',
-    'cybercrime':    'الجرائم المعلوماتية',
-    'hacking':       'الاختراق الإلكتروني',
-    'bribery':       'الرشوة',
-    'fraud':         'الاحتيال',
-    'arbitration':   'تحكيم',
-    'bankruptcy':    'إفلاس',
-    'divorce':       'طلاق',
-    'custody':       'حضانة',
-    'notary':        'كاتب عدل',
-    'lawyer':        'محامي',
-    'attorney':      'محامي',
-    'judge':         'قاضي',
-    'penalty':       'عقوبة',
-    'penaly':        'عقوبة',
-    'penelty':       'عقوبة',
-    'fine':          'غرامة',
-    'imprisonment':  'سجن',
-    'rights':        'حقوق',
-    'rigths':        'حقوق',
-    'contract':      'عقد',
-    'salary':        'الأجر',
-    'wage':          'الأجر',
-    'employee':      'عامل',
-    'employer':      'صاحب عمل',
-    'terminated':    'فُصلت',
-    'dismissed':     'فُصلت',
-    'dissmised':     'فُصلت',
-    'overtime':      'العمل الإضافي',
-    'probation':     'فترة التجربة',
-    'evidence':      'إثبات',
-    'lawsuit':       'دعوى قضائية',
-    'conditions':    'شروط',
-    'conditons':     'شروط',
-    'requirements':  'شروط',
-    'i worked':      'عملت',
+    'gratuity':    'مكافأة نهاية الخدمة',
+    'wrongful':    'تعسفي',
+    'cybercrime':  'الجرائم المعلوماتية',
+    'arbitration': 'تحكيم',
+    'bankruptcy':  'إفلاس',
+    'divorce':     'طلاق',
+    'custody':     'حضانة',
+    'notary':      'كاتب عدل',
+    'lawyer':      'محامي',
+    'attorney':    'محامي',
+    'judge':       'قاضي',
+    'penalty':     'عقوبة',
+    'penaly':      'عقوبة',
+    'fine':        'غرامة',
+    'imprisonment':'سجن',
+    'rights':      'حقوق',
+    'rigths':      'حقوق',
+    'contract':    'عقد',
+    'salary':      'الأجر',
+    'employee':    'عامل',
+    'employer':    'صاحب عمل',
+    'terminated':  'فُصلت',
+    'dismissed':   'فُصلت',
+    'dissmised':   'فُصلت',
+    'overtime':    'العمل الإضافي',
+    'evidence':    'إثبات',
+    'conditions':  'شروط',
+    'requirements':'شروط',
+    'i worked':    'عملت',
+    'fraud':       'الاحتيال',
+    'bribery':     'الرشوة',
+    'hacking':     'الاختراق الإلكتروني',
 }
 
 LAW_KEYWORDS = {
@@ -235,7 +218,6 @@ LAW_KEYWORDS = {
     'راتب':'نظام العمل','أجر':'نظام العمل',
     'استقالة':'نظام العمل','إشعار':'نظام العمل',
     'حقوقي':'نظام العمل','مستحقاتي':'نظام العمل',
-    'كيفية حساب بدل':'نظام العمل',
     'لقد تم فصلي':'نظام العمل','ما هي حقوقي':'نظام العمل',
     'جرائم معلوماتية':'نظام مكافحة الجرائم المعلوماتية',
     'اختراق':'نظام مكافحة الجرائم المعلوماتية',
@@ -256,7 +238,7 @@ LEGAL_KEYWORDS = [
 
 PRACTICAL_PATTERNS = [
     'أنا موظف','أنا عامل','اشتغلت','فصلوني','هل لي',
-    'هل يحق','هل أستحق','صاحب العمل','في شركة',
+    'هل يحق','هل أستحق','صاحب العمل',
     'طردوني','حقي','حقوقي','مستحقاتي',
     'تم فصلي','فُصلت','عملت',
 ]
@@ -273,9 +255,8 @@ QUERY_EXPANSION = {
     'شروط اتفاق التحكيم':        'اتفاق التحكيم مكتوب باطل',
     'إجراءات تسجيل العقار':      'طلب القيد إدارة التسجيل العقاري',
     'شروط تعيين القاضي':         'يشترط فيمن يُعيَّن في وظائف القضاء أن يكون سعودي الجنسية بالأصل',
-    'متطلبات تعيين القضاة':      'يشترط فيمن يُعيَّن في وظائف القضاء سعودي شريعة إسلامية',
+    'متطلبات تعيين القضاة':      'يشترط فيمن يُعيَّن في وظائف القضاء سعودي شريعة إسلامية امتحان',
     'شروط ترخيص المحاماة':       'يشترط فيمن يزاول مهنة المحاماة أن يكون سعودي الجنسية مقيداً في جدول المحامين',
-    'ترخيص المحاماة':            'يشترط فيمن يزاول مهنة المحاماة سعودي الجنسية شهادة شريعة',
     'مكافأة نهاية الخدمة':      'يستحق العامل مكافأة عن مدة خدمته نصف شهر سنة',
     'هل لي مكافأة':             'يستحق العامل مكافأة نهاية الخدمة',
     'اشتغلت':                   'يستحق العامل مكافأة عن مدة خدمته',
@@ -289,7 +270,6 @@ QUERY_EXPANSION = {
     'فُصلت':                    'إنهاء عقد العمل تعسفي تعويض فصل',
     'حقوقي':                   'حقوق العامل يستحق مكافأة تعويض نظام العمل',
     'مستحقاتي':                'يستحق العامل مكافأة نهاية الخدمة تعويض',
-    'كيفية حساب بدل نهاية':    'مكافأة نهاية الخدمة يستحق العامل نصف شهر سنة',
     'عقوبة غسيل الأموال':      'يعاقب على جريمة غسل الأموال سجن غرامة',
 }
 
@@ -378,12 +358,12 @@ async def lifespan(app: FastAPI):
 
 async def startup():
     global vectorstore, bm25_index, bm25_texts, bm25_metadatas
-    global embeddings, ACTIVE_MODELS, GEMINI_AVAILABLE, gemini_model, groq_client
+    global embeddings, ACTIVE_MODELS, GEMINI_AVAILABLE, gemini_clients, groq_client
 
-    logger.info("Starting Saudi Legal AI v4...")
+    logger.info("Starting Saudi Legal AI v4.1...")
 
+    # ── Groq ─────────────────────────────────────────────
     groq_client = Groq(api_key=GROQ_API_KEY)
-
     for m in [
         {'model': 'llama-3.3-70b-versatile', 'name': 'Groq llama-3.3'},
         {'model': 'llama-3.1-8b-instant',    'name': 'Groq llama-3.1-8b'},
@@ -397,22 +377,24 @@ async def startup():
         except Exception as e:
             logger.warning(f"❌ {m['name']}: {str(e)[:30]}")
 
+    # ── Gemini Multi-Model ────────────────────────────────
     if GEMINI_KEY:
-        try:
-            genai.configure(api_key=GEMINI_KEY)
-            gemini_model     = genai.GenerativeModel('models/gemma-3-1b-it')
-            gemini_model.generate_content('hi')
-            GEMINI_AVAILABLE = True
-            logger.info("✅ Gemini")
-        except Exception as e:
-            logger.warning(f"❌ Gemini: {str(e)[:40]}")
-
+        genai.configure(api_key=GEMINI_KEY)
+        for model_name in GEMINI_MODELS:
+            try:
+                gemini_clients[model_name] = genai.GenerativeModel(model_name)
+                GEMINI_AVAILABLE = True
+                logger.info(f"✅ {model_name}")
+            except Exception as e:
+                logger.warning(f"❌ {model_name}: {str(e)[:40]}")
+
+    # ── Embeddings ────────────────────────────────────────
     logger.info("Loading embeddings...")
     embeddings = SentenceTransformerEmbeddings(
         model_name='sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2')
 
+    # ── ChromaDB ──────────────────────────────────────────
     chroma_client_persist = chromadb.PersistentClient(path=CHROMA_PATH)
-
     try:
         collection = chroma_client_persist.get_collection('saudi_legal_v4')
         if collection.count() > 100:
@@ -440,32 +422,28 @@ async def startup():
                 })
             for item in dataset if len(item.get('text','')) > 30
         ]
-
         for a in EXTRA_LAWS:
             docs.append(Document(page_content=a['text'], metadata={
-                'article_number': a['article_number'],
-                'law_name': a['law_name'],
-                'law_type': a['law_type'],
-                'source': a['source'],
+                'article_number': a['article_number'], 'law_name': a['law_name'],
+                'law_type': a['law_type'], 'source': a['source'],
             }))
 
         splitter = RecursiveCharacterTextSplitter(chunk_size=1500, chunk_overlap=200)
         chunks   = splitter.split_documents(docs)
-
         vectorstore = Chroma.from_documents(
             documents=chunks, embedding=embeddings,
             client=chroma_client_persist, collection_name='saudi_legal_v4')
         logger.info(f"✅ ChromaDB created: {vectorstore._collection.count()} chunks")
 
+    # ── BM25 ──────────────────────────────────────────────
     stop_words = {'من','في','على','إلى','عن','مع','هي','هو','ما','لا','أن','إن'}
     def tokenize(text): return [w for w in text.split() if len(w) > 2 and w not in stop_words]
-
     all_chunks     = vectorstore.get()
     bm25_texts     = all_chunks['documents']
     bm25_metadatas = all_chunks['metadatas']
     bm25_index     = BM25Okapi([tokenize(t) for t in bm25_texts])
     logger.info(f"✅ BM25: {len(bm25_texts)} docs")
-    logger.info("✅ Saudi Legal AI v4 Ready!")
+    logger.info("✅ Saudi Legal AI v4.1 Ready!")
 
 # ══════════════════════════════════════════════════════════
 # Pipeline Functions
@@ -526,14 +504,12 @@ def fuzzy_correct_word(word, threshold=88):
 
 def fuzzy_normalize(question):
     clean = re.sub(r'[؟?،,.]', '', question)
-    words, corrections, result = clean.split(), [], []
+    words, result = clean.split(), []
     for word in words:
         if len(word) < 5 or word.isdigit() or not any('\u0600' <= c <= '\u06ff' for c in word):
             result.append(word); continue
-        corrected = fuzzy_correct_word(word)
-        if corrected != word: corrections.append(f'{word}→{corrected}')
-        result.append(corrected)
-    return ' '.join(result), corrections
+        result.append(fuzzy_correct_word(word))
+    return ' '.join(result)
 
 def normalize_question(question):
     log      = []
@@ -545,10 +521,8 @@ def normalize_question(question):
     if translated != question:
         log.append(f'🌐 {translated}')
         question = translated
-    corrected, corrections = fuzzy_normalize(question)
-    if corrections:
-        log.append(f'✏️ {" | ".join(corrections)}')
-        question = corrected
+    corrected = fuzzy_normalize(question)
+    if corrected != question: question = corrected
     for col, formal in COLLOQUIAL.items():
         question = re.sub(rf'\b{col}\b', formal, question, flags=re.IGNORECASE)
     question = re.sub(r'[؟?]+', '؟', question).strip()
@@ -599,7 +573,7 @@ def expand_query(question):
         elif any(w in question for w in ['حقوق','حقوقي','مستحقات']):
             result.append('حقوق العامل يستحق مكافأة تعويض نظام العمل')
     cleaned = question
-    for prefix in ['ما هي ','ما هو ','هل ','متى ','كيف ']: cleaned = cleaned.replace(prefix, '')
+    for prefix in ['ما هي ','ما هو ','هل ','متى ','كيف ']: cleaned = cleaned.replace(prefix,'')
     cleaned = cleaned.replace('؟','').strip()
     if cleaned != question and len(cleaned) > 5: result.append(cleaned)
     words = [w for w in question.split() if len(w) > 3 and w not in
@@ -679,6 +653,9 @@ def post_process(answer, docs):
              or any(s in l for s in ['📋','✅','📝','⚠️','•','-','─'])]
     return '\n'.join(clean).strip()
 
+# ══════════════════════════════════════════════════════════
+# Generation
+# ══════════════════════════════════════════════════════════
 def _call_groq(messages):
     for m in sorted(ACTIVE_MODELS, key=lambda x: 0 if '70b' in x['model'] else 1):
         try:
@@ -693,20 +670,24 @@ def _call_groq(messages):
             raise e
     return None, None
 
-def _call_gemini(messages):
-    if not GEMINI_AVAILABLE: return None, None
-    try:
-        system = next((m['content'] for m in messages if m['role']=='system'),'')
-        user   = next((m['content'] for m in messages if m['role']=='user'),'')
-        r      = gemini_model.generate_content(f'{system}\n\n{user}')
-        answer = r.text
-        if not answer: return None, None
-        arabic = sum(1 for c in answer if '\u0600' <= c <= '\u06ff')
-        if arabic / max(len([c for c in answer if c.strip()]),1) < 0.3: return None, None
-        return answer, 'Gemini'
-    except Exception as e:
-        logger.warning(f"Gemini: {str(e)[:40]}")
-        return None, None
+def _call_gemini_chain(messages):
+    system = next((m['content'] for m in messages if m['role']=='system'), '')
+    user   = next((m['content'] for m in messages if m['role']=='user'), '')
+    prompt = f'{system}\n\n{user}'
+
+    for model_name, client in gemini_clients.items():
+        try:
+            r      = client.generate_content(prompt)
+            answer = r.text
+            if not answer: continue
+            arabic = sum(1 for c in answer if '\u0600' <= c <= '\u06ff')
+            if arabic / max(len([c for c in answer if c.strip()]),1) < 0.3: continue
+            logger.info(f"Gemini: {model_name.split('/')[-1]}")
+            return answer, model_name.split('/')[-1]
+        except Exception as e:
+            if '429' in str(e): continue
+            continue
+    return None, None
 
 def generate_with_fallback(messages):
     if rate_limiter.can_use('groq'):
@@ -718,14 +699,21 @@ def generate_with_fallback(messages):
         except Exception as e:
             if '429' in str(e) or 'rate' in str(e).lower():
                 for _ in range(28): rate_limiter.requests['groq'].append(time.time())
-    result, model = _call_gemini(messages)
+    else:
+        pass
+
+    result, model = _call_gemini_chain(messages)
     if result: return result, model
+
     wait = rate_limiter.wait_time('groq')
     if wait > 0:
         time.sleep(min(wait + 1, 15))
         return generate_with_fallback(messages)
     return None, None
 
+# ══════════════════════════════════════════════════════════
+# ask_legal
+# ══════════════════════════════════════════════════════════
 def ask_legal_core(question: str) -> dict:
     original = question
     question, norm_log = normalize_question(question)
@@ -758,8 +746,7 @@ def ask_legal_core(question: str) -> dict:
             docs = vectorstore.similarity_search(q, k=k_per_query)
         all_docs.extend(docs)
 
-    bm25_docs    = []
-    keyword_docs = []
+    bm25_docs, keyword_docs = [], []
     for q in queries: bm25_docs.extend(bm25_search(q, k=5, target_law=target_law))
 
     if target_law:
@@ -817,7 +804,7 @@ def ask_legal_core(question: str) -> dict:
 app = FastAPI(
     title='Saudi Legal AI',
     description='نظام الذكاء الاصطناعي للقانون السعودي — وزارة العدل',
-    version='4.0.0',
+    version='4.1.0',
     lifespan=lifespan
 )
 
@@ -846,15 +833,21 @@ class QuestionResponse(BaseModel):
 
 @app.get('/')
 def root():
-    return {'name': 'Saudi Legal AI', 'version': '4.0.0', 'status': 'running', 'docs': '/docs'}
+    return {
+        'name':    'Saudi Legal AI',
+        'version': '4.1.0',
+        'status':  'running',
+        'docs':    '/docs'
+    }
 
 @app.get('/health')
 def health():
     return {
-        'status':  'healthy',
-        'chunks':  vectorstore._collection.count() if vectorstore else 0,
-        'models':  [m['name'] for m in ACTIVE_MODELS] + (['Gemini'] if GEMINI_AVAILABLE else []),
-        'version': '4.0.0'
+        'status':        'healthy',
+        'chunks':        vectorstore._collection.count() if vectorstore else 0,
+        'groq_models':   [m['name'] for m in ACTIVE_MODELS],
+        'gemini_models': list(gemini_clients.keys()),
+        'version':       '4.1.0'
     }
 
 @app.post('/ask', response_model=QuestionResponse)
@@ -865,7 +858,6 @@ async def ask(req: QuestionRequest, request: Request):
         raise HTTPException(status_code=400, detail='السؤال طويل جداً')
 
     ip = request.headers.get('X-Forwarded-For','unknown').split(',')[0].strip()
-
     if not check_rate_limit(ip):
         raise HTTPException(status_code=429, detail='حاول مرة أخرى بعد دقيقة')
 
@@ -880,9 +872,11 @@ async def ask(req: QuestionRequest, request: Request):
     ms     = int((time.time() - t0) * 1000)
     status = 'blocked' if result['model'] in ['out_of_scope','quality_check'] else 'success'
 
-    request_log.appendleft({'time': datetime.now().strftime('%H:%M:%S'),
-                             'ip': ip, 'question': req.question[:60],
-                             'model': result['model'], 'status': status, 'ms': ms})
+    request_log.appendleft({
+        'time': datetime.now().strftime('%H:%M:%S'),
+        'ip': ip, 'question': req.question[:60],
+        'model': result['model'], 'status': status, 'ms': ms
+    })
     stats['total']  += 1
     stats[status if status in stats else 'errors'] += 1
     active_ips[ip]  = active_ips.get(ip, 0) + 1
@@ -898,9 +892,11 @@ async def ask(req: QuestionRequest, request: Request):
 @app.get('/stats')
 def get_stats():
     return {
-        'total': stats['total'], 'success': stats['success'],
-        'blocked': stats['blocked'], 'unique_ips': len(active_ips),
-        'recent': list(request_log)[:10]
+        'total':      stats['total'],
+        'success':    stats['success'],
+        'blocked':    stats['blocked'],
+        'unique_ips': len(active_ips),
+        'recent':     list(request_log)[:10]
     }
 
 @app.get('/laws')