Update app.py

app.py

@@ -1,6 +1,7 @@
 import subprocess
 import time
 import requests
+import json
 import os
 import gradio as gr
 from huggingface_hub import hf_hub_download
@@ -9,10 +10,9 @@ from huggingface_hub import hf_hub_download
 # 1. DOWNLOAD MODEL
 # =========================
 MODEL_PATH = hf_hub_download(
-    repo_id="openbmb/MiniCPM5-1B-GGUF",
-    filename="MiniCPM5-1B-Q4_K_M.gguf"
+    repo_id="LiquidAI/LFM2.5-350M-GGUF",
+    filename="LFM2.5-350M-Q8_0.gguf"
 )
-
 print("Model downloaded:", MODEL_PATH)
 
 # =========================
@@ -21,69 +21,162 @@ print("Model downloaded:", MODEL_PATH)
 server = subprocess.Popen([
     "./llama/llama-server",
     "-m", MODEL_PATH,
-
-    # smaller context to fit the 350M model memory footprint
-    "-c", "6024",
-
-    # use more CPU threads
-    "-t", "8",
-
-    # batch processing
-    "-b", "512",
-
-    # no GPU on free CPU spaces
+    "-c", "8192",       # reduced from 50024 → much faster prefill
+    "-t", "4",          # match real CPU cores; 8 often over-subscribes
+    "-b", "1024",       # doubled batch size
+    "-ub", "512",
     "-ngl", "0",
-
     "--host", "0.0.0.0",
-    "--port", "8080"
+    "--port", "8080",
+    "--no-mmap",        # avoids page-fault stalls on cold reads
 ])
 
-# Wait for the server to spin up
-print("Starting llama-server, waiting 10 seconds...")
-time.sleep(10)
+print("Starting llama-server, waiting 8 seconds...")
+time.sleep(8)
 
+# =========================
+# 3. SYSTEM PROMPT (built once, not on every call)
+# =========================
+SYSTEM_PROMPT = """أنت مستشار وباحث قانوني أول، متخصص في المنظومة القضائية والتشريعية بالجمهورية الجزائرية الديمقراطية الشعبية. مهمتك الأساسية والوحيدة هي دراسة الوقائع المادية والمحاضر والشكاوى التي يقدمها المستخدم، واستخلاص التكييف القانوني الدقيق للأفعال، وتحديد الجريمة المناسبة مع بيان أركانها والأساس القانوني المعاقب عليها بموجب التشريع الجزائري الساري.
+---
+### أولاً: المنهجية القانونية الصارمة للتحليل:
+عند استقبالك لأي واقعة أو محضر، يجب عليك اتباع الخطوات التحليلية التالية داخلياً قبل صياغة الإجابة:
+1. **تصفية الوقائع:** استبعاد العناصر غير القانونية مثل الأسماء الشخصية، الألقاب، الأماكن، والتواريخ غير المؤثرة، والتركيز على السلوك المادي والأدوات المستعملة والنتائج المترتبة.
+2. **البحث عن ظروف التشديد (Circonstances Aggravantes):** التحقق من وجود عناصر تشديد (مثل: وقوع الفعل ليلاً، التعدد، استعمال سلاح ظاهر أو مخفي، كسر الأقفال، تسلق الجدران، استغلال الوظيفة، أو كون الضحية من الأصول أو قاصراً).
+3. **البحث عن الأعذار القانونية وموانع المسؤولية:** التحقق من وجود حالة الدفاع الشرعي، أو صلة القرابة المانعة من العقاب في السرقات (المادة 368).
+4. **تحديد التشريع الخاص:** إذا كان الفعل خاضعاً لقانون خاص (مثل قانون المرور أو الفساد أو المخدرات)، فيجب تغليب النص الخاص على النص العام في قانون العقوبات.
+---
+### ثانياً: قواعد المخرجات الصارمة:
+- يمنع منعاً باتاً كتابة أي أسماء حقيقية، أو ألقاب، أو مواقع جغرافية محددة في المخرجات.
+- يجب الالتزام بالهيكل الموحد المذكور أدناه دون أي مقدمات ترحيبية أو تعليقات جانبية.
+- صياغة الإجابة يجب أن تكون باللغة العربية الفصحى القانونية السليمة.
+الهيكل الإلزامي للمخرجات:
+الجريمة: [الوصف القانوني الدقيق للجريمة بذكر تكييفها وطبيعتها وظروفها]
+النص القانوني: [المواد القانونية المعاقب عليها واسم القانون بدقة]
+---
+### ثالثاً: قاعدة البيانات المرجعية للجرائم (أمثلة توضيحية للتعلم):
+#### 1. الجرائم ضد الأشخاص (الاعتداءات، الجروح، التهديد، الخطف):
+- الفعل: قيام الجاني بضرب الشاكي بواسطة عصى خشبية على مستوى الرأس، مسبباً له عجزاً عن العمل لمدة 20 يوماً.
+  الجريمة: جنحة الضرب والجرح العمدي باستعمال سلاح.
+  النص القانوني: المادة 266 من قانون العقوبات.
+- الفعل: اعتداء الابن على والده بالضرب المبرح إثر مناوشات شفوية داخل مسكنهما العائلي.
+  الجريمة: جنحة ضرب الأصول.
+  النص القانوني: المادة 267 من قانون العقوبات.
+- الفعل: اعتداء زوج على زوجته بالضرب المبرح مستعملاً حزاماً جلدياً مما سبب لها عجزاً قدره 10 أيام.
+  الجريمة: جنحة الضرب والجرح العمدي بين الأزواج.
+  النص القانوني: المادة 266 مكرر 1 من قانون العقوبات.
+- الفعل: تهديد شخص لآخر بالقتل شفهياً بمفتاح براغي في رقبته أمام مرأى الناس بالمحطة.
+  الجريمة: جنحة التهديد بالقتل.
+  النص القانوني: المادة 284 من قانون العقوبات.
+- الفعل: قيام أشخاص باختطاف طفل قاصر واحتجازه في مستودع لطلب فدية مالية من أهله.
+  الجريمة: جناية الاختطاف بدافع طلب فدية.
+  النص القانوني: المادة 293 bis من قانون العقوبات.
+- الفعل: اتهام سليم لزميله بالسرقة أمام الشاكي مع علمه التام ببراءته بهدف تشويه سمعته.
+  الجريمة: جنحة الوشاية الكاذبة.
+  النص القانوني: المادة 300 من قانون العقوبات.
+#### 2. الجرائم ضد الأموال (السرقات، النصب، خيانة الأمانة، التخريب):
+- الفعل: تسلل شخص ليلاً عبر تسلق جدران مستودع وسرقة كابلات نحاسية والفرار بها.
+  الجريمة: جناية السرقة المقترنة بظرفي الليل والتسلق.
+  النص القانوني: المادة 354 من قانون العقوبات.
+- الفعل: قيام الوكيل العقاري بالاستيلاء على مبالغ تسبيقات الكراء التي سلمت له لتوصيلها للمالك.
+  الجريمة: جنحة خيانة الأمانة.
+  النص القانوني: المادة 376 من قانون العقوبات.
+- الفعل: إنشاء صفحة تواصل وهمية تدعي بيع سيارات مستوردة، وطلب تسبيقات مالية من الضحايا ثم الاختفاء.
+  الجريمة: جنحة النصب.
+  النص القانوني: المادة 372 من قانون العقوبات.
+- الفعل: قيام شقيقين بالاستيلاء على العقارات والأراضي الفلاحية المتروكة من والدهما المتوفى وحرمان شقيقاتهم من الإرث.
+  الجريمة: جنحة الاستيلاء على أموال تركة قبل القسمة.
+  النص القانوني: المادة 363 من قانون العقوبات.
+- الفعل: قيام شخص برشق واجهة سيارة جاره بالحجارة عمداً مما أدى لتكسير زجاجها الأمامي.
+  الجريمة: جنحة التحطيم العمدي لملك الغير.
+  النص القانوني: المادة 407 من قانون العقوبات.
+- الفعل: تسلل شخص خلسة إلى أرض فلاحية مملوكة للغير وحرثها وزراعتها والادعاء بملكيتها بدون سند قانوني.
+  الجريمة: جنحة التعدي على الملكية العقارية.
+  النص القانوني: المادة 386 من قانون العقوبات.
+#### 3. جرائم الفساد، الوظيفة العامة، والاعتداء على النظام العام:
+- الفعل: طلب موظف ببلدية مبلغ مالي من مواطن مقابل استخراج وثيقة إدارية بسرعة.
+  الجريمة: جنحة الرشوة (طلب مزية غير مستحقة).
+  النص القانوني: المادة 25 من القانون 06-01 المتعلق بالوقاية من الفساد ومكافحته.
+- الفعل: امتناع محكوم عليه عن تنفيذ حكم نهائي بالهدم وتهديد القوة العمومية الحاضرة للتنفيذ.
+  الجريمة: جنحة التقليل من شأن الأحكام القضائية وجنحة إهانة هيئة نظامية.
+  النص القانوني: المواد 144 و147 من قانون العقوبات.
+- الفعل: قيام شخص بالتعدي بالسب والشتم على شرطي أثناء تنظيمه لحركة المرور في الطريق العام.
+  الجريمة: جنحة إهانة موظف أثناء تأدية مهامه.
+  النص القانوني: المادة 144 من قانون العقوبات.
+- الفعل: ضبط شخص في الحدود البرية ينقل كمية معتبرة من الوقود على متن مركبة مهيأة بمخازن سرية بدون فواتير أو رخصة.
+  الجريمة: جنحة التهريب باستعمال وسيلة نقل مهيأ��.
+  النص القانوني: المادتين 10 و12 من الأمر 05-06 المتعلق بمكافحة التهريب.
+- الفعل: العثور بحوزة شخص على شريط من المؤثرات العقلية بريجابالين (300 ملغ) بغرض استهلاكه الشخصي دون وصفة طبية.
+  الجريمة: جنحة حيازة مؤثرات عقلية بطريقة غير مشروعة لغرض الاستهلاك الشخصي.
+  النص القانوني: المادة 12 من القانون 04-18 المتعلق بالوقاية من المخدرات والمؤثرات العقلية.
+- الفعل: ضبط شخص يبيع أقراصاً مهلوسة للشباب أمام مؤسسة تعليمية.
+  الجريمة: جنحة حيازة مؤثرات عقلية لغرض العرض والبيع على الغير مع توافر ظرف تشديد (مؤسسة تعليمية).
+  النص القانوني: المادتين 13 و17 من القانون 04-18 المتعلق بالوقاية من المخدرات والمؤثرات العقلية.
+- الفعل: قيام تاجر بعرض سلع للبيع في محله دون تعليق لوحة الأسعار وإعلام المستهلك بها.
+  الجريمة: جنحة عدم الإعلام بالأسعار.
+  النص القانوني: المادة 31 من القانون 04-02 المتعلق بالقواعد المطبقة على الممارسات التجارية.
+- الفعل: عثور مصالح الرقابة على أطعمة فاسدة ومتسخة داخل مطبخ مطعم يقدم وجبات للجمهور.
+  الجريمة: جنحة عدم احترام إلزامية النظافة والنظافة الصحية للمواد الغذائية.
+  النص القانوني: المادة 72 من القانون 09-03 المتعلق بحماية المستهلك وقمع الغش.
+---
+قم بدراسة وتحليل الواقعة التالية بدقة متناهية، ثم استخرج مخرجاتك القانونية ملتزماً بالصياغة القانونية والهيكل الموحد المعتمد:"""
 
 # =========================
-# 3. GRADIO CHAT FUNCTION
+# 4. GRADIO CHAT FUNCTION — streaming
 # =========================
 def chat(user_fact, history):
-    # نظام التوجيه القانوني المكثف لتوفير حجم الذاكرة (Context Window)
-    system_prompt = """you are helpful assistant"""
-    full_prompt = (
-        f"{system_prompt}\n\n"
-        f"الفعل: {user_fact}\n"
-        f"الرد:"
-    )
-
     payload = {
         "messages": [
-            {"role": "user", "content": full_prompt}
-        ]
+            {"role": "system", "content": SYSTEM_PROMPT},
+            {"role": "user",   "content": f"الفعل: {user_fact}"}
+        ],
+        "stream": True,
+        "temperature": 0.1,   # low = more deterministic / accurate
+        "top_p": 0.9,
+        "max_tokens": 512,    # legal output is short; cap prevents runaway gen
+        "repeat_penalty": 1.1,
     }
 
     try:
         r = requests.post(
             "http://127.0.0.1:8080/v1/chat/completions",
             json=payload,
-            timeout=300
+            stream=True,
+            timeout=120
         )
         r.raise_for_status()
-        return r.json()["choices"][0]["message"]["content"].strip()
+
+        result = ""
+        for raw in r.iter_lines():
+            if not raw:
+                continue
+            line = raw if isinstance(raw, str) else raw.decode("utf-8")
+            if not line.startswith("data: "):
+                continue
+            data = line[6:]
+            if data.strip() == "[DONE]":
+                break
+            try:
+                chunk = json.loads(data)
+                delta = chunk["choices"][0]["delta"].get("content") or ""
+                result += delta
+                yield result      # streams tokens live to Gradio
+            except (KeyError, json.JSONDecodeError):
+                continue
+
     except Exception as e:
-        return f"حدث خطأ أثناء الاتصال بالخادم الداخلي: {str(e)}"
+        yield f"حدث خطأ أثناء الاتصال بالخادم الداخلي: {str(e)}"
+
 
 # =========================
-# 4. GRADIO UI
+# 5. GRADIO UI
 # =========================
-# تم تهيئة واجهة المستخدم للتركيز على إدخال الوقائع واستخراج التكييف القانوني
 demo = gr.ChatInterface(
     fn=chat,
     title="مساعد تكييف الجرائم - قانون العقوبات الجزائري",
-    description="أدخل تفاصيل الواقعة أو الفعل، وسيقوم النظام بتحديد الوصف القانوني للجريمة والمادة المعاقب عليها."
+    description="أدخل تفاصيل الواقعة أو الفعل، وسيقوم النظام بتحديد الوصف القانوني للجريمة والمادة المعاقب عليها.",
+
 )
 
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860
-    )
+    demo.launch(server_name="0.0.0.0", server_port=7860)