Update app.py

app.py

@@ -1,6 +1,7 @@
 import subprocess
 import time
 import requests
+import os
 import gradio as gr
 from huggingface_hub import hf_hub_download
 
@@ -21,7 +22,7 @@ server = subprocess.Popen([
     "./llama/llama-server",
     "-m", MODEL_PATH,
 
-    # smaller context
+    # smaller context to fit the 350M model memory footprint
     "-c", "1024",
 
     # use more CPU threads
@@ -33,39 +34,77 @@ server = subprocess.Popen([
     # no GPU on free CPU spaces
     "-ngl", "0",
 
-    
-
     "--host", "0.0.0.0",
     "--port", "8080"
 ])
 
+# Wait for the server to spin up
+print("Starting llama-server, waiting 10 seconds...")
 time.sleep(10)
 
 
 # =========================
 # 3. GRADIO CHAT FUNCTION
 # =========================
-def chat(message, history):
+def chat(user_fact, history):
+    # نظام التوجيه القانوني المكثف لتوفير حجم الذاكرة (Context Window)
+    system_prompt = (
+        "أنت مساعد قانوني متخصص في قانون العقوبات الجزائري والقوانين المكملة له. "
+        "مهمتك هي استخراج الجريمة المناسبة وإرفاق المادة القانونية المعاقب عليها بناءً على الفعل المدخل. "
+        "قوانين هامة:\n"
+        "- لا تكتب أي أسماء أو ألقاب أو أماكن في الإجابة.\n"
+        "- اكتب فقط اسم الجريمة والمادة القانونية مباشرة دون مقدمات.\n\n"
+        "أمثلة توضيحية:\n"
+        "الفعل: ضرب شخص فسبب له عجز 40 يوم\n"
+        "الرد: الضرب والجرح العمدي المادة 266 من قانون العقوبات\n"
+        "الفعل: سرق هاتف نقال من محل تجاري\n"
+        "الرد: السرقة المادة 350 من قانون العقوبات\n"
+        "الفعل: قيادة مركبة وهو في حالة سكر\n"
+        "الرد: السياقة في حالة سكر المادة 74 من قانون المرور\n"
+        "الفعل: بناء طابق بدون حيازة رخصة بناء\n"
+        "الرد: جنحة البناء بدون رخصة المادة 77 من قانون التهيأة والتعمير\n"
+        "الفعل: صرف شيك لا يقابله رصيد قائم وقابل للصرف\n"
+        "الرد: إصدار شيك بدون رصيد المادة 374 من قانون العقوبات\n"
+        "الفعل: تصرف في عتاد بعدما امتنع عن تسديد مبلغ القرض\n"
+        "الرد: جنحة تبديد أموال مرهونة المادة 364 من قانون العقوبات"
+    )
+
+    # دمج التوجيهات مع الواقعة التي أدخلها المستخدم حالياً
+    full_prompt = (
+        f"{system_prompt}\n\n"
+        f"الفعل: {user_fact}\n"
+        f"الرد:"
+    )
+
     payload = {
         "messages": [
-            {"role": "user", "content": message}
+            {"role": "user", "content": full_prompt}
         ]
     }
 
-    r = requests.post(
-        "http://127.0.0.1:8080/v1/chat/completions",
-        json=payload,
-        timeout=300
-    )
-
-    return r.json()["choices"][0]["message"]["content"]
+    try:
+        r = requests.post(
+            "http://127.0.0.1:8080/v1/chat/completions",
+            json=payload,
+            timeout=300
+        )
+        r.raise_for_status()
+        return r.json()["choices"][0]["message"]["content"].strip()
+    except Exception as e:
+        return f"حدث خطأ أثناء الاتصال بالخادم الداخلي: {str(e)}"
 
 # =========================
 # 4. GRADIO UI
 # =========================
-demo = gr.ChatInterface(chat)
+# تم تهيئة واجهة المستخدم للتركيز على إدخال الوقائع واستخراج التكييف القانوني
+demo = gr.ChatInterface(
+    fn=chat,
+    title="مساعد تكييف الجرائم - قانون العقوبات الجزائري",
+    description="أدخل تفاصيل الواقعة أو الفعل، وسيقوم النظام بتحديد الوصف القانوني للجريمة والمادة المعاقب عليها."
+)
 
-demo.launch(
-    server_name="0.0.0.0",
-    server_port=7860
-)
+if __name__ == "__main__":
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860
+    )