"""
دورة القانون – Hugging Face Space
النسخة العربية الكاملة
Compatible avec le Docker HF (gradio 4.44.1 bundled + Python 3.11).
TTS: Microsoft Edge Neural Voices (edge-tts) — صوت عربي طبيعي
LLM: خادم gemma4cpu عبر HTTP (OpenAI-compatible API)
     https://ramedde-gemma4cpu.hf.space/v1/chat/completions
الأصوات المتاحة:
  ar-DZ-IsmaelNeural   — رجل، لهجة جزائرية  ✅ (موصى به)
  ar-DZ-AminaNeural    — امرأة، لهجة جزائرية
  ar-SA-HamedNeural    — رجل، فصحى سعودية
  ar-SA-ZariyahNeural  — امرأة، فصحى سعودية
  ar-EG-SalmaNeural    — امرأة، مصرية
  ar-EG-ShakirNeural   — رجل، مصري
  ar-MA-JamalNeural    — رجل، مغربي
"""
import os, json, time, tempfile, random, threading, asyncio
import sys
import urllib.request
import urllib.error

try:
    import pyaudioop
    sys.modules["audioop"] = pyaudioop
except ImportError:
    pass

import gradio as gr

try:
    import gradio_client.utils as _gcu
    import gradio.blocks as _gb
    _orig = _gcu.json_schema_to_python_type
    def _safe(schema):
        try: return _orig(schema)
        except Exception: return "Any"
    _gcu.json_schema_to_python_type = _safe
    _gb.client_utils.json_schema_to_python_type = _safe
except Exception:
    pass

# ── إعدادات الخادم البعيد ──────────────────────────────────────────────────────
# عنوان خادم gemma4cpu — يمكن تجاوزه بمتغير بيئة LLM_BASE_URL
LLM_BASE_URL = os.getenv(
    "LLM_BASE_URL",
    "https://ramedde-gemma4cpu.hf.space"
).rstrip("/")

LLM_CHAT_ENDPOINT  = f"{LLM_BASE_URL}/v1/chat/completions"
LLM_MODEL_NAME     = os.getenv("LLM_MODEL_NAME", "gemma-4-e2b-q4")
LLM_TIMEOUT        = int(os.getenv("LLM_TIMEOUT", "120"))   # ثانية

# ── TTS: edge-tts (Microsoft Neural Arabic) ──────────────────────────────────
try:
    import edge_tts
    _EDGE_TTS_AVAILABLE = True
    print("[law-app] edge-tts متاح ✅")
except ImportError:
    _EDGE_TTS_AVAILABLE = False
    print("[law-app] edge-tts غير متاح — سيتم استخدام gTTS كبديل")

DEFAULT_VOICE = "ar-DZ-IsmaelNeural"
DEFAULT_RATE  = "-10%"
DEFAULT_PITCH = "+0Hz"

# ── دروس القانون ─────────────────────────────────────────────────────────────
LESSONS = [
    {
        "title": "الدرس الأول — المبادئ التوجيهية للإثبات والمحاكمة العادلة",
        "text": """
بناءً على الفقه الجنائي الفرنسي، يرتكز الإطار العام للخصومة الجزائية على مبدأين أساسيين: المحاكمة العادلة وحرية الإثبات.
الجزء الأول: الحق في محاكمة عادلة وقرينة البراءة.
يُعتبر الحق في محاكمة عادلة حجر الزاوية في الإجراءات. هذا المبدأ يفرض توازناً بين حقوق الدفاع وسلطات الاتهام. وتُعد قرينة البراءة امتداداً طبيعياً لهذا الحق، حيث يتحمل الادعاء عبء إثبات التهمة، ويُفسر أي شك لصالح المتهم.
الإسقاط على القانون الجزائري: تبنى المشرع الجزائري هذه الفلسفة صراحة في المادة الأولى من قانون الإجراءات الجزائية، حيث نص على أن القانون يقوم على مبادئ الشرعية والمحاكمة العادلة واحترام كرامة وحقوق الإنسان، وأكد أن الشك يفسر في كل الأحوال لصالح المتهم.
الجزء الثاني: حرية الإثبات والاقتناع الشخصي.
في المادة الجنائية، وخلافاً للمادة المدنية، النظام السائد هو الأدلة المعنوية وليس الأدلة القانونية المقيدة. القاضي يبني حكمه على قناعته الصميمة بناءً على ما يُعرض أمامه من أدلة خضعت للنقاش الوجاهي.
الإسقاط على القانون الجزائري: نجد هذا المبدأ مجسداً في المادة ثلاثمائة وتسعة وأربعين، التي تنص على جواز إثبات الجرائم بأي طريق، وأن القاضي يصدر حكمه تبعاً لاقتناعه الخاص.
الجزء الثالث: مبدأ مشروعية الدليل.
يشير الفقه الفرنسي إلى أن حرية الإثبات ليست مطلقة، إذ يُمنع على السلطات العامة استخدام الحيل أو الاستفزاز لارتكاب الجريمة من أجل الحصول على دليل.
""",
        "questions": [
            {"q": "ما هو المبدأ اللاتيني الذي يعني أن الشك يفسر لصالح المتهم، وأين نجد تطبيقه في القانون الجزائري؟",
             "hint": "مبدأ In dubio pro reo، وتم تكريسه بوضوح في المادة (1) من قانون الإجراءات الجزائية الجزائري."},
            {"q": "ما هو الفرق بين نظام الأدلة القانونية ونظام الاقتناع الشخصي؟",
             "hint": "الأدلة القانونية تلزم القاضي بقيمة محددة للدليل (كالمدني)، بينما الاقتناع الشخصي يترك للقاضي حرية تقييم الدليل بناءً على وجدانه وضميره (كما في المادة 349 جزائري)."},
            {"q": "ما المقصود بمبدأ مشروعية الدليل بالنسبة لسلطات الأمن؟",
             "hint": "عدم لجوء السلطات إلى الحيل الخداعية أو الاستفزاز لارتكاب الجريمة من أجل الإيقاع بالمتهم وجمع الأدلة ضده."},
        ],
    },
    {
        "title": "الدرس الثاني — أجهزة الدعوى: النيابة العامة والشرطة القضائية",
        "text": """
تعتمد الإجراءات الجزائية على أجهزة مكلفة باكتشاف الجريمة وتحريك الدعوى، وهي الشرطة القضائية والنيابة العامة.
الجزء الأول: الشرطة القضائية والفرق بينها وبين الشرطة الإدارية.
يفرق الفقه الفرنسي بوضوح بين الشرطة الإدارية، وهدفها وقائي بحت لمنع الإخلال بالنظام العام، والشرطة القضائية، وهدفها قمعي يبدأ عند وقوع الجريمة للبحث عن مرتكبيها. رجال الشرطة القضائية يتصرفون تحت إدارة النيابة العامة.
الإسقاط على القانون الجزائري: نجد هذا التقسيم الهرمي في القانون الجزائري، حيث توضع الشرطة القضائية تحت إشراف النائب العام وإدارة وكيل الجمهورية ورقابة غرفة الاتهام.
الجزء الثاني: النيابة العامة ومبدأ ملاءمة المتابعة.
النيابة العامة هي خصم في الدعوى، ولكنها تدافع عن المجتمع. تتميز النيابة العامة بخاصيتين فقهيتين:
الأولى: التبعية التدرجية، والتي يُعبر عنها بالمثل الفرنسي المأثور: القلم عبد لكن الكلمة حرة، أي أن القاضي ملزم بالتعليمات المكتوبة لرؤسائه، لكنه حر في إبداء رأيه شفهياً في الجلسة.
الثانية: مبدأ ملاءمة المتابعة، حيث أن النيابة ليست ملزمة بمتابعة كل جريمة تصل لعلمها، بل تُقّدر ما إذا كانت المتابعة ملائمة أم أن حفظ الملف أفضل.
الإسقاط على القانون الجزائري: تم تكريس مبدأ الاستقلالية الشفوية في المادة إحدى وأربعين، كما تكرس المادة سبعة وأربعون مبدأ ملاءمة المتابعة.
""",
        "questions": [
            {"q": "اشرح القاعدة الفقهية: القلم عبد والكلمة حرة وكيف طبقها القانون الجزائري؟",
             "hint": "تعني خضوع عضو النيابة للتعليمات المكتوبة لرؤسائه، لكنه يتمتع بحرية إبداء رأيه الشفوي في الجلسة. كرسها القانون الجزائري في المادة (41)."},
            {"q": "ما الفرق الجوهري بين الشرطة الإدارية والشرطة القضائية؟",
             "hint": "الشرطة الإدارية وقائية (تمنع الجريمة)، والشرطة القضائية قمعية وإجرائية (تبدأ بعد وقوع الجريمة لجمع الأدلة)."},
            {"q": "ماذا يُقصد بمبدأ ملاءمة المتابعة الممنوح لوكيل الجمهورية؟",
             "hint": "سلطة وكيل الجمهورية في تقييم ما إذا كان من الأفضل للمجتمع تحريك الدعوى أو حفظ الملف لعدم الملاءمة (المادة 47)."},
        ],
    },
    {
        "title": "الدرس الثالث — الدعاوى الناشئة عن الجريمة وأسباب انقضائها",
        "text": """
تولد الجريمة حقين: حق المجتمع في توقيع العقاب وهي الدعوى العمومية، وحق الضحية في التعويض وهي الدعوى المدنية.
الجزء الأول: استقلالية وتداخل الدعويين.
الدعوى العمومية هي الهدف الأصلي للإجراءات، بينما الدعوى المدنية هي دعوى تابعة. ومع ذلك، للضحية خيار إما اللجوء للقضاء المدني المستقل، أو الانضمام كطرف مدني أمام القضاء الجزائي.
الإسقاط على القانون الجزائري: سمح القانون الجزائري في المادتين الثالثة والرابعة للطرف المتضرر بمباشرة الدعوى المدنية مع الدعوى العمومية. كما كرس القاعدة الفقهية الجزائي يعقل المدني في المادة الخامسة، التي توجب على القاضي المدني إرجاء الفصل حتى يفصل القضاء الجزائي.
الجزء الثاني: انقضاء الدعوى والتقادم.
حسب الفقه، التقادم مبرره فقدان الأدلة مع مرور الزمن، ونسيان المجتمع للاضطراب الذي خلفته الجريمة.
هناك إشكالية فقهية حول الجرائم الخفية، وهي متى يبدأ تقادمها؟ استقر الفقه الفرنسي على تأخير بدء التقادم إلى يوم اكتشاف الجريمة، مع وضع سقف زمني لضمان عدم بقاء التهديد بالمتابعة للأبد.
الإسقاط على القانون الجزائري: تطرق المشرع الجزائري لانقضاء الدعوى في المادة التاسعة. وتبنى نظرية الجرائم الخفية والمخفية في المادة الخامسة عشرة، حيث جعل سريان التقادم يبدأ من يوم ظهورها للعلن.
""",
        "questions": [
            {"q": "ماذا تعني القاعدة الفقهية الجزائي يعقل المدني المكرسة في المادة الخامسة من القانون الجزائري؟",
             "hint": "تعني أن القاضي المدني يجب أن يوقف الفصل في دعوى التعويض المرفوعة أمامه، حتى يصدر حكم نهائي في الدعوى الجزائية المرتبطة بها."},
            {"q": "ما هي الجريمة المخفية وكيف يعالجها القانون من حيث التقادم؟",
             "hint": "هي الجريمة التي يضع الجاني مناورات لإخفائها. لا يبدأ تقادمها إلا من يوم ظهورها للعلن، مع احترام سقف زمني أقصى حدده القانون الجزائري في المادة (15)."},
            {"q": "هل تنازل الضحية عن الدعوى المدنية يوقف الدعوى العمومية؟",
             "hint": "كقاعدة عامة لا يوقفها (لأن الدعوى العمومية ملك للمجتمع)، إلا في الحالات الاستثنائية التي يعلق فيها القانون المتابعة على شكوى (المادة 3)."},
        ],
    },
    {
        "title": "الدرس الرابع — التحقيق التحضيري والحرية الفردية",
        "text": """
يعتبر التحقيق التحضيري مرحلة وسطى بين التحريات والمحاكمة، ويتميز بطبيعة مزدوجة تمزج بين النظام التنقيبي السري والمكتوب من جهة، والنظام الاتهامي الوجاهي الذي يكفل ضمانات الدفاع من جهة أخرى.
الجزء الأول: إخطار قاضي التحقيق العيني.
من المبادئ الفقهية الكبرى أن قاضي التحقيق يُخطر بالوقائع لا بالأشخاص. لا يملك القاضي حق التصدي من تلقاء نفسه، بل يجب أن يأتيه طلب افتتاحي من النيابة. إذا اكتشف وقائع جديدة، لا يجوز له التحقيق فيها إلا بعد الحصول على طلب إضافي.
الإسقاط على القانون الجزائري: كرس المشرع الجزائري هذا المبدأ في المادة مئة وأربعين بمنع قاضي التحقيق من إجراء تحقيق إلا بموجب طلب من وكيل الجمهورية.
الجزء الثاني: المساس بالحرية، الحبس المؤقت والرقابة.
يؤكد الفقه على أن المتهم بريء حتى تثبت إدانته، وبالتالي فالحرية هي الأصل والتقييد هو الاستثناء. الحبس المؤقت هو أشد تدبير، ويُشترط الفقه أن يكون الخيار الأخير.
الإسقاط على القانون الجزائري: تماشياً مع هذا، قررت المادة مئة وسبعة وتسعون أن المتهم يبقى حراً، ويمكن اللجوء للرقابة القضائية. ولا يُلجأ إلى الحبس المؤقت إلا بصفة استثنائية وبموجب أمر مسبب في المادة مئتان وواحد.
""",
        "questions": [
            {"q": "ما معنى قاعدة الإخطار العيني لقاضي التحقيق؟",
             "hint": "تعني أن قاضي التحقيق مقيد بالتحقيق في الوقائع (الأفعال) المحددة في طلب النيابة العامة، ولا يملك حق التحقيق في وقائع جديدة يكتشفها من تلقاء نفسه (المادة 140 جزائري)."},
            {"q": "كيف وازن القانون بين حق المجتمع في التحقيق وحق المتهم في الحرية؟",
             "hint": "بجعل الحرية هي الأصل (المادة 197)، والرقابة القضائية هي البديل الأول، واعتبار الحبس المؤقت تدبيراً استثنائياً له مبررات صارمة ومدد زمنية قصوى (المادة 201)."},
            {"q": "ماذا يسمى الطلب الذي يوجهه وكيل الجمهورية لقاضي التحقيق للتحقيق في وقائع جديدة مكتشفة؟",
             "hint": "يُسمى الطلب الإضافي للتحقيق، وبالفرنسية Réquisitoire supplétif."},
        ],
    },
    {
        "title": "الدرس الخامس — المحاكمة وقوة الشيء المقضي فيه",
        "text": """
تعتبر مرحلة المحاكمة تتويجاً للإجراءات، وتتميز بالعلنية والمشافهة والوجاهية أي النقاش الحضوري.
الجزء الأول: المحاكمة الغيابية.
تاريخياً، كان يُطلق على محاكمة الجنايات غيابياً مصطلح التمرد، وكانت إجراءاتها قاسية جداً تحرم المتهم من الدفاع. تطور الفقه الأوروبي للقضاء على هذه التسمية واستبدالها بإجراءات الغياب التي تضمن حداً أدنى من الحقوق.
الإسقاط على القانون الجزائري: القانون الجزائري نظم المحاكمة في الغياب أمام محكمة الجنايات بشكل يضمن أنه إذا حضر محامي المتهم الغائب بعذر مقبول، فإنه يرافع عنه وتعتبر المحاكمة حضورية. وإذا كان الغياب بدون عذر، يحاكم غيابياً بدون محلفين وفق المادة أربعمئة واثنين وخمسين.
الجزء الثاني: حجية الشيء المقضي فيه.
ينص المبدأ الفقهي الشهير Non bis in idem على أنه لا يجوز محاكمة الشخص عن نفس الفعل مرتين. ويشترط لنجاح هذا الدفع توافر ثلاثة عناصر: وحدة الوقائع، وحدة الأطراف، ووحدة السبب.
الإسقاط على القانون الجزائري: تُعتبر قوة الشيء المقضي فيه سبباً من أسباب انقضاء الدعوى العمومية في المادة التاسعة. وقد نصت المادة الأولى بصيغة صريحة على أنه لا يجوز متابعة أو محاكمة أو معاقبة شخص مرتين من أجل نفس الأفعال ولو تم إعطاؤها وصفاً مغايراً.
""",
        "questions": [
            {"q": "ما هو مبدأ عدم المحاكمة مرتين وأين تم النص عليه في قانون الإجراءات الجزائية الجزائري؟",
             "hint": "هو مبدأ عدم جواز محاكمة الشخص مرتين عن نفس الأفعال، وقد كُرس في المادة (1) كنقطة مبدئية، وفي المادة (9) كسبب لانقضاء الدعوى."},
            {"q": "كيف تعامل المشرع الجزائري مع غياب المتهم أمام محكمة الجنايات؟",
             "hint": "إذا غاب المتهم غياباً غير مبرر، يُحاكم غيابياً بواسطة القضاة فقط (بدون إشراك المحلفين) وفقاً للمادة (451)."},
            {"q": "لماذا شدد الفقه الحديث على إلغاء إجراءات التمرد؟",
             "hint": "لأنها كانت تحرم المتهم الغائب من أي حق في الدفاع حتى لو حضر محاميه، وهو ما يتعارض مع الحقوق الأساسية للمحاكمة العادلة."},
        ],
    },
    {
        "title": "الدرس السادس — بدائل المتابعة والعدالة التفاوضية",
        "text": """
ظهر في الفقه الجنائي الحديث توجه نحو العدالة التفاوضية لتخفيف العبء عن المحاكم، وتسريع الإجراءات في الجرائم غير الخطيرة.
الجزء الأول: الوساطة الجزائية.
هي بديل للمتابعة يعتمد على الصلح بين الجاني والضحية. تهدف إلى جبر الضرر وإعادة السلم الاجتماعي دون الحاجة لصدور حكم إدانة.
الإسقاط على القانون الجزائري: نظمها المشرع في المواد من تسعة وخمسين إلى ثمانية وستين. يمكن لوكيل الجمهورية اقتراحها قبل تحريك الدعوى في جنح معينة كخيانة الأمانة والتهديد والسب العلني. تنفيذ اتفاق الوساطة يؤدي إلى انقضاء الدعوى العمومية.
الجزء الثاني: الأمر الجزائي.
هو إجراء مبسط بدون جلسة مرافعة.
الإسقاط على القانون الجزائري: حددته المواد من خمسمئة وواحد وثلاثين إلى خمسمئة وثمانية وثلاثين في المخالفات والجنح البسيطة التي عقوبتها أقل من سنتين. يفصل القاضي من مكتبه بغرامة أو براءة بناء على طلب النيابة، وللمتهم حق الاعتراض.
الجزء الثالث: المثول بناءً على الاعتراف المسبق بالذنب.
مستوحى من النظم الأنجلوسكسونية، وهو إجراء يعرض فيه المدعي العام عقوبة مخففة على المتهم مقابل اعترافه.
الإسقاط على القانون الجزائري: كرسه القانون الجزائري في المواد من خمسمئة وتسعة وثلاثين إلى خمسمئة وثمانية وأربعين. إذا اعترف المتهم، يقترح عليه وكيل الجمهورية عقوبة لا تتجاوز نصف الحد الأقصى للحبس. يُصادق عليها قاضي الحكم في جلسة، ويكون للمحضر قوة الإدانة.
""",
        "questions": [
            {"q": "ما هي الميزة الأساسية للوساطة الجزائية في القانون الجزائري؟",
             "hint": "أن تنفيذ اتفاق الوساطة بين الطرفين يؤدي إلى انقضاء الدعوى العمومية دون تسجيل إدانة قضائية (المادتان 9 و 59)."},
            {"q": "في إجراء المثول بناء على الاعتراف المسبق بالذنب، ما هو الحد الأقصى للعقوبة التي يمكن للنيابة اقتراحها؟",
             "hint": "يجب ألا تتجاوز مدة الحبس المقترحة نصف الحد الأقصى للعقوبة المقررة قانوناً لتلك الجنحة (المادة 541)."},
            {"q": "بماذا يختلف إجراء الأمر الجزائي عن المحاكمة العادية؟",
             "hint": "الأمر الجزائي يتم بدون مرافعة وجاهية في الجلسة، حيث يفصل القاضي بناءً على الملف فقط، وبعقوبة الغرامة أو البراءة (المادة 533)."},
        ],
    }
]

LISTEN_SECONDS = 20   # 15 دقيقة
PASS_SCORE     = 6

# ── ذاكرة تخزين الصوت ────────────────────────────────────────────────────────
_audio_cache = {}
_audio_lock  = threading.Lock()

# ── TTS helpers ───────────────────────────────────────────────────────────────
def _sync_run(text, path, voice, rate, pitch):
    """توليد الصوت في thread مستقل لتجنب تعارض event loops."""
    new_loop = asyncio.new_event_loop()
    asyncio.set_event_loop(new_loop)
    try:
        async def _synthesize():
            communicate = edge_tts.Communicate(text, voice=voice, rate=rate, pitch=pitch)
            await communicate.save(path)
        new_loop.run_until_complete(_synthesize())
    finally:
        new_loop.close()


def get_lecture_audio(lesson_idx: int, voice: str = DEFAULT_VOICE):
    """توليد الصوت للدرس باستخدام edge-tts مع gTTS كاحتياطي."""
    cache_key = f"{lesson_idx}_{voice}"
    if cache_key in _audio_cache and os.path.exists(_audio_cache[cache_key]):
        return _audio_cache[cache_key]

    with _audio_lock:
        if cache_key in _audio_cache and os.path.exists(_audio_cache[cache_key]):
            return _audio_cache[cache_key]

        text = LESSONS[lesson_idx]["text"]

        if _EDGE_TTS_AVAILABLE:
            try:
                import concurrent.futures
                path = tempfile.mktemp(suffix=".mp3")
                with concurrent.futures.ThreadPoolExecutor(max_workers=1) as executor:
                    future = executor.submit(_sync_run, text, path, voice, DEFAULT_RATE, DEFAULT_PITCH)
                    future.result(timeout=120)
                if os.path.exists(path) and os.path.getsize(path) > 1000:
                    _audio_cache[cache_key] = path
                    print(f"[law-app] ✅ edge-tts: درس {lesson_idx + 1} ({voice})")
                    return path
                print(f"[law-app] ⚠️  edge-tts أنتج ملفاً فارغاً، سيتم الرجوع إلى gTTS")
            except Exception as e:
                print(f"[law-app] ⚠️  edge-tts فشل ({e})، سيتم الرجوع إلى gTTS")

        try:
            from gtts import gTTS
            tts  = gTTS(text=text, lang="ar", slow=False)
            path = tempfile.mktemp(suffix=".mp3")
            tts.save(path)
            _audio_cache[cache_key] = path
            print(f"[law-app] ✅ gTTS (احتياطي): درس {lesson_idx + 1}")
            return path
        except Exception as e:
            print(f"[law-app] ❌ خطأ gTTS: {e}")
            return None


# ── التقييم عبر الخادم البعيد (OpenAI-compatible) ──────────────────────────
def _call_remote_llm(system_prompt: str, user_prompt: str) -> str:
    """
    استدعاء خادم gemma4cpu عبر /v1/chat/completions.
    يعيد النص الخام من الرد، أو يرفع استثناء عند الفشل.
    """
    payload = json.dumps({
        "model": LLM_MODEL_NAME,
        "messages": [
            {"role": "system", "content": system_prompt},
            {"role": "user",   "content": user_prompt},
        ],
        "max_tokens":  300,
        "temperature": 0.1,
        "stream":      False,
    }).encode("utf-8")

    req = urllib.request.Request(
        LLM_CHAT_ENDPOINT,
        data=payload,
        headers={"Content-Type": "application/json"},
        method="POST",
    )
    with urllib.request.urlopen(req, timeout=LLM_TIMEOUT) as resp:
        data = json.loads(resp.read().decode("utf-8"))

    # استخراج النص من رد OpenAI-compatible
    return data["choices"][0]["message"]["content"].strip()


def _keyword_score(answer: str, hint: str):
    """تسجيل احتياطي بالكلمات المفتاحية عند غياب الخادم."""
    kw    = [w.strip().lower() for w in hint.split("،")]
    hits  = sum(1 for k in kw if k in answer.lower())
    score = min(10, round((hits / max(len(kw), 1)) * 10))
    return score, f"(وضع الكلمات المفتاحية – {hits}/{len(kw)} مطابقة)"


def evaluate_answer(question: str, answer: str, hint: str):
    """
    تقييم إجابة الطالب.
    يحاول أولاً استخدام الخادم البعيد gemma4cpu،
    ويرجع إلى تقييم الكلمات المفتاحية عند الفشل.
    """
    system_prompt = (
        "أنت أستاذ قانون جزائري صارم ومتفهم. "
        "تقيّم إجابات الطلاب وتردّ فقط بصيغة JSON صحيحة بدون أي نص إضافي."
    )
    user_prompt = (
        f"السؤال: {question}\n"
        f"إجابة الطالب: {answer}\n"
        f"العناصر المتوقعة: {hint}\n\n"
        "قيّم هذه الإجابة من 10 وأعطِ تعليقاً موجزاً باللغة العربية.\n"
        'أجب فقط بهذا JSON في سطر واحد بدون أي مقدمة:\n'
        '{"score": <عدد صحيح 0-10>, "feedback": "<تعليق موجز>"}'
    )

    try:
        print(f"[law-app] 🌐 إرسال طلب التقييم إلى {LLM_CHAT_ENDPOINT}")
        raw = _call_remote_llm(system_prompt, user_prompt)
        print(f"[law-app] رد الخادم: {raw}")

        # استخراج أول JSON صالح من الرد
        start = raw.find("{")
        end   = raw.rfind("}") + 1
        if start == -1 or end == 0:
            raise ValueError("لا يوجد JSON في الرد")
        data = json.loads(raw[start:end])
        score    = max(0, min(10, int(data["score"])))
        feedback = data.get("feedback", "")
        return score, feedback

    except urllib.error.URLError as e:
        print(f"[law-app] ⚠️  خطأ شبكة للخادم البعيد: {e} — وضع الكلمات المفتاحية")
        return _keyword_score(answer, hint)
    except Exception as e:
        print(f"[law-app] ⚠️  خطأ في التقييم البعيد: {e} — وضع الكلمات المفتاحية")
        return _keyword_score(answer, hint)


# ── فحص صحة الخادم عند الإطلاق ──────────────────────────────────────────────
def _check_remote_server():
    """فحص توفر خادم LLM البعيد عند بدء التشغيل."""
    try:
        req = urllib.request.Request(
            f"{LLM_BASE_URL}/",
            method="GET",
            headers={"Accept": "application/json"},
        )
        with urllib.request.urlopen(req, timeout=15) as resp:
            print(f"[law-app] ✅ خادم LLM البعيد متاح: {LLM_BASE_URL} (HTTP {resp.status})")
    except Exception as e:
        print(f"[law-app] ⚠️  خادم LLM البعيد غير متاح: {e}")
        print(f"[law-app]    سيُستخدم وضع الكلمات المفتاحية عند فشل كل استدعاء.")


# ── الحالة ────────────────────────────────────────────────────────────────────
def fresh():
    return json.dumps({
        "phase":        "idle",
        "listen_start": None,
        "question_idx": None,
        "lesson_idx":   0,
        "voice":        DEFAULT_VOICE,
    })

def S(s): return json.loads(s)
def D(d): return json.dumps(d)


# ── المعالجات ──────────────────────────────────────────────────────────────────
def start_listening(ss):
    state      = S(ss)
    lesson_idx = state.get("lesson_idx", 0)
    voice      = state.get("voice", DEFAULT_VOICE)
    lesson     = LESSONS[lesson_idx]
    state.update({"phase": "listen", "listen_start": time.time(), "question_idx": None})
    audio = get_lecture_audio(lesson_idx, voice)
    total = len(LESSONS)
    msg = (
        f"🎧 **{lesson['title']}** ({lesson_idx + 1}/{total})\n\n"
        "استمع لمدة **15 دقيقة** كاملة، ثم انقر على **🔄 التحقق من الوقت**."
    )
    return (
        audio, msg,
        gr.update(visible=True),
        gr.update(visible=False),
        gr.update(visible=False),
        gr.update(visible=False),
        gr.update(interactive=False, value="⏳ جارٍ الاستماع…"),
        D(state),
    )


def change_voice_and_reload(voice_choice, ss):
    state = S(ss)
    state["voice"] = voice_choice
    return D(state)


def check_timer(ss):
    state = S(ss)
    if state["phase"] != "listen" or not state["listen_start"]:
        return gr.update(), "⚠️ ابدأ الاستماع أولاً.", ss
    rem  = max(0, LISTEN_SECONDS - (time.time() - state["listen_start"]))
    m, s = divmod(int(rem), 60)
    if rem <= 0:
        return (
            gr.update(interactive=True, value="✅ الانتقال إلى الاختبار"),
            "✅ انتهى الوقت! انقر على **الانتقال إلى الاختبار**.",
            ss,
        )
    return (
        gr.update(interactive=False, value=f"⏳ تبقّى {m:02d}:{s:02d}"),
        f"🎧 انتظر **{m} دقيقة و{s:02d} ثانية** إضافية.",
        ss,
    )


def open_quiz(ss):
    state   = S(ss)
    elapsed = time.time() - (state["listen_start"] or 0)
    if elapsed < LISTEN_SECONDS:
        m, s = divmod(int(LISTEN_SECONDS - elapsed), 60)
        return (
            gr.update(visible=True), gr.update(visible=False),
            gr.update(visible=False), gr.update(visible=False),
            f"⚠️ انتظر **{m} دقيقة و{s:02d} ثانية** إضافية.", "", "", ss,
        )
    lesson_idx = state.get("lesson_idx", 0)
    idx = random.randrange(len(LESSONS[lesson_idx]["questions"]))
    state.update({"phase": "quiz", "question_idx": idx})
    return (
        gr.update(visible=False), gr.update(visible=True),
        gr.update(visible=False), gr.update(visible=False),
        "", LESSONS[lesson_idx]["questions"][idx]["q"], "", D(state),
    )


def submit_answer(answer, ss):
    state      = S(ss)
    lesson_idx = state.get("lesson_idx", 0)

    if state["phase"] != "quiz":
        yield (
            gr.update(visible=True), gr.update(visible=False),
            gr.update(visible=False), gr.update(visible=False),
            "",
            gr.update(value="📤 إرسال الإجابة", interactive=True),
            gr.update(interactive=True),
            ss,
        )
        return

    # الخطوة 1: تعطيل الزر فوراً وإبلاغ المستخدم
    yield (
        gr.update(visible=False), gr.update(visible=True),
        gr.update(visible=False), gr.update(visible=False),
        "",
        gr.update(value="⏳ جارٍ التصحيح بواسطة Gemma…", interactive=False),
        gr.update(interactive=False),
        ss,
    )

    # الخطوة 2: التقييم عبر الخادم البعيد
    q = LESSONS[lesson_idx]["questions"][state["question_idx"]]
    score, fb = evaluate_answer(q["q"], answer, q["hint"])
    total   = len(LESSONS)
    is_last = (lesson_idx >= total - 1)

    if score >= PASS_SCORE:
        state["phase"] = "pass"
        result = (
            f"## 🎉 أحسنت! اجتزت الدرس {lesson_idx + 1} من {total}.\n\n"
            f"**النتيجة: {score}/10**\n\n**التعليق:** {fb}"
        )
    else:
        state["phase"] = "fail"
        result = (
            f"## ❌ النتيجة غير كافية: {score}/10\n\n"
            f"**التعليق:** {fb}\n\n"
            f"الحد الأدنى المطلوب: **{PASS_SCORE}/10**. "
            "يُرجى **إعادة الاستماع إلى الدرس** قبل المحاولة مجدداً."
        )

    yield (
        gr.update(visible=False), gr.update(visible=False),
        gr.update(visible=True),  gr.update(visible=False),
        result,
        gr.update(value="📤 إرسال الإجابة", interactive=True),
        gr.update(value="", interactive=True),
        D(state),
    )


def handle_next(ss):
    state      = S(ss)
    lesson_idx = state.get("lesson_idx", 0)
    total      = len(LESSONS)

    if state["phase"] == "fail":
        return start_listening(D(state))

    if lesson_idx >= total - 1:
        state["phase"] = "done"
        cert_text = (
            "# 🏆 تهانينا!\n\n"
            "لقد أتممت دورة **القانون الجنائي الإجرائي** بنجاح.\n\n"
            f"اجتزت جميع الدروس الستة وحصلت على شهادة الإتمام.\n\n"
            "---\n*استمر في التعلم واستكشاف مجالات القانون الأخرى.*"
        )
        return (
            None, "",
            gr.update(visible=False),
            gr.update(visible=False),
            gr.update(visible=False),
            gr.update(visible=True, value=cert_text),
            gr.update(interactive=False, value="✅ انتهى"),
            D(state),
        )
    else:
        state["lesson_idx"]   = lesson_idx + 1
        state["listen_start"] = None
        return start_listening(D(state))


def restart_from_cert(ss):
    return start_listening(fresh())


# ── واجهة المستخدم ─────────────────────────────────────────────────────────────
CSS = """
footer { display: none !important; }
#title { text-align: center; }
.gradio-container {
    direction: rtl;
    font-family: 'Amiri', 'Noto Sans Arabic', 'Segoe UI', sans-serif;
}
.gr-button { font-family: 'Amiri', 'Noto Sans Arabic', sans-serif; }
#voice-selector { direction: rtl; }
.voice-info  { font-size: 0.85em; color: #666; margin-top: 4px; }
.server-info { font-size: 0.80em; color: #4a7c59; margin-top: 4px; }
"""

VOICE_CHOICES = [
    ("🇩🇿 جزائري — إسماعيل (رجل)",  "ar-DZ-IsmaelNeural"),
    ("🇩🇿 جزائرية — أمينة (امرأة)",  "ar-DZ-AminaNeural"),
    ("🇸🇦 سعودي — حامد (رجل)",       "ar-SA-HamedNeural"),
    ("🇸🇦 سعودية — زارية (امرأة)",    "ar-SA-ZariyahNeural"),
    ("🇪🇬 مصرية — سلمى (امرأة)",      "ar-EG-SalmaNeural"),
    ("🇪🇬 مصري — شاكر (رجل)",        "ar-EG-ShakirNeural"),
    ("🇲🇦 مغربي — جمال (رجل)",        "ar-MA-JamalNeural"),
]

with gr.Blocks(title="دورة القانون", css=CSS) as demo:
    session = gr.State(fresh())

    gr.Markdown(
        "# ⚖️ دورة القانون — التكوين القانوني\n"
        "استمع إلى كل درس · أجب عن السؤال · تقدّم درساً بعد درس",
        elem_id="title",
    )

    # ── معلومات الخادم ────────────────────────────────────────────────────────
    gr.Markdown(
        f"<div class='server-info'>🤖 التقييم: <strong>Gemma 4 E2B</strong> عبر "
        f"<code>{LLM_BASE_URL}</code></div>"
    )

    # ── اختيار الصوت ──────────────────────────────────────────────────────────
    with gr.Row():
        with gr.Column(scale=2):
            voice_dropdown = gr.Dropdown(
                choices=[(label, code) for label, code in VOICE_CHOICES],
                value=DEFAULT_VOICE,
                label="🎙️ اختر صوت القراءة",
                interactive=True,
                elem_id="voice-selector",
            )
        with gr.Column(scale=1):
            gr.Markdown(
                "<div class='voice-info'>الصوت الجزائري موصى به للمحتوى القانوني الجزائري.<br>"
                "سيُطبَّق الصوت عند بدء الدرس القادم.</div>"
            )

    # ── مرحلة الاستماع ────────────────────────────────────────────────────────
    with gr.Column(visible=True) as listen_col:
        gr.Markdown("## 🎧 الخطوة الأولى — الاستماع إلى الدرس")
        listen_msg = gr.Markdown("انقر على **ابدأ** للبدء.")
        audio_out  = gr.Audio(
            label="الدرس الصوتي (صوت عربي طبيعي)",
            autoplay=False,
            interactive=False,
        )
        timer_info = gr.Markdown("")
        with gr.Row():
            start_btn = gr.Button("▶️ بدء الاستماع",          variant="primary")
            check_btn = gr.Button("🔄 التحقق من الوقت",       variant="secondary")
            quiz_btn  = gr.Button("✅ الانتقال إلى الاختبار", variant="secondary",
                                  interactive=False)

    # ── مرحلة الاختبار ────────────────────────────────────────────────────────
    with gr.Column(visible=False) as quiz_col:
        gr.Markdown("## 📝 الخطوة الثانية — سؤال التقييم")
        question_md = gr.Markdown("")
        answer_box  = gr.Textbox(
            label="إجابتك",
            placeholder="اكتب إجابتك هنا…",
            lines=5,
        )
        submit_btn = gr.Button("📤 إرسال الإجابة", variant="primary")

    # ── النتيجة ───────────────────────────────────────────────────────────────
    with gr.Column(visible=False) as result_col:
        gr.Markdown("## 📊 النتيجة")
        result_md = gr.Markdown("")
        next_btn  = gr.Button("➡️ الدرس التالي", variant="primary")

    # ── الشهادة ───────────────────────────────────────────────────────────────
    with gr.Column(visible=False) as cert_col:
        cert_md     = gr.Markdown("")
        restart_btn = gr.Button("🔄 إعادة البدء من الأول", variant="secondary")

    # ── ربط الأحداث ──────────────────────────────────────────────────────────
    LISTEN_OUT = [
        audio_out, listen_msg,
        listen_col, quiz_col, result_col, cert_col,
        quiz_btn, session,
    ]

    voice_dropdown.change(
        fn=change_voice_and_reload,
        inputs=[voice_dropdown, session],
        outputs=[session],
    )

    start_btn.click(start_listening, [session], LISTEN_OUT)
    check_btn.click(check_timer,     [session], [quiz_btn, timer_info, session])

    QUIZ_OUT = [
        listen_col, quiz_col, result_col, cert_col,
        timer_info, question_md, answer_box, session,
    ]
    quiz_btn.click(open_quiz, [session], QUIZ_OUT)

    RES_OUT = [
        listen_col, quiz_col, result_col, cert_col,
        result_md, submit_btn, answer_box, session,
    ]
    submit_btn.click(submit_answer, [answer_box, session], RES_OUT)
    answer_box.submit(submit_answer, [answer_box, session], RES_OUT)

    next_btn.click(handle_next,          [session], LISTEN_OUT)
    restart_btn.click(restart_from_cert, [session], LISTEN_OUT)

# ── فحص الخادم عند الإطلاق ────────────────────────────────────────────────────
threading.Thread(target=_check_remote_server, daemon=True).start()

demo.queue()

if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True)