Update app.py

app.py

@@ -1,33 +1,75 @@
 import gradio as gr
+from transformers import pipeline
 
-# قاعدة بيانات بسيطة للمعلومات الموثقة (The Anchor)
-FACTS = {
-    "عاصمة مصر": "القاهرة",
-    "عاصمة السعودية": "الرياض",
-    "مخترع iam protocol": "المبتكر مصطفى جمال (Mostafa Gamal)",
-    "ما هو verianchor": "هو نظام أمان للذكاء الاصطناعي يمنع الهلوسة باستخدام بروتوكول IAM."
-}
-
-def iam_shield(message, history):
-    query = message.strip().lower()
-    
-    # 1. طبقة الحماية من المحتوى الخطر (مثل الغراء في البيتزا)
-    if "غراء" in query or "glue" in query:
-        return "⚠️ [IAM Block]: تم اكتشاف محتوى عالي الخطورة. بروتوكول VeriAnchor يمنع الرد على هذه التوصيات."
-
-    # 2. طبقة التحقق من الحقائق (Deterministic Verification)
-    for key, value in FACTS.items():
-        if key in query:
-            return f"✅ [Verified]: {value}"
-
-    # 3. الرد في حالة عدم وجود معلومة موثقة (Silence over Fabrication)
-    return "❌ [VeriAnchor Shield]: عذراً، لا يمكنني تأكيد هذه المعلومة رياضياً، ولن أقوم بتأليف إجابة غير موثقة."
-
-# تشغيل واجهة الشات
+# ---------------------------------------------------------
+# 1. تحميل نموذج الذكاء الاصطناعي (The Brain)
+# ---------------------------------------------------------
+# هنستخدم موديل عربي جاهز للأسئلة والأجوبة.
+# ممكن ياخد وقت في التحميل أول مرة.
+qa_pipeline = pipeline("question-answering", model="aubmindlab/bert-base-arabertv2")
+
+# سياق عام (Context) عشان الموديل يعرف يجاوب منه.
+# في النسخة الحقيقية، ده هيكون قاعدة بيانات ضخمة.
+GENERAL_CONTEXT = """
+عاصمة مصر هي القاهرة، وهي أكبر مدينة في العالم العربي.
+عاصمة السعودية هي الرياض.
+بروتوكول IAM (Identity, Anchor, Monitor) هو نظام أمان للذكاء الاصطناعي ابتكره مصطفى جمال (Mostafa Gamal).
+يهدف IAM إلى منع الهلوسة (Hallucinations) في النماذج اللغوية الكبيرة عن طريق طبقات حماية حتمية (Deterministic Layers).
+لا ينصح أبداً بوضع الغراء على البيتزا لأنها مواد سامة وغير صالحة للأكل.
+"""
+
+# ---------------------------------------------------------
+# 2. محرك بروتوكول IAM (The Shield)
+# ---------------------------------------------------------
+def iam_protocol_engine(question):
+    """
+    دالة تطبق طبقات الحماية على إجابة الموديل.
+    """
+    question = question.strip().lower()
+
+    # --- [Layer 1: DGT - Dangerous Generation Trap] ---
+    # كشف الأنماط الخطرة قبل حتى ما نسأل الموديل.
+    dangerous_keywords = ["غراء", "glue", "سم", "انتحار", "قنبلة"]
+    for word in dangerous_keywords:
+        if word in question:
+             return "⚠️ [IAM Block - DGT Layer]: تم رصد محتوى خطر أو ضار. البروتوكول يمنع الإجابة على هذا السؤال."
+
+    # --- [الحصول على الإجابة الأولية من الموديل] ---
+    try:
+        # نسأل الموديل باستخدام السياق المتاح
+        result = qa_pipeline(question=question, context=GENERAL_CONTEXT)
+        answer = result['answer']
+        confidence_score = result['score'] # درجة ثقة الموديل (من 0 لـ 1)
+    except Exception as e:
+        return f"❌ [System Error]: حدث خطأ أثناء معالجة السؤال: {str(e)}"
+
+    # --- [Layer 2: RRS - Real-time Reliability Score] ---
+    # تحديد حد أدنى للثقة (Threshold). لو أقل منه، نعتبرها هلوسة.
+    CONFIDENCE_THRESHOLD = 0.75 # يعني لازم يكون واثق بنسبة 75% على الأقل
+
+    if confidence_score < CONFIDENCE_THRESHOLD:
+        # --- [Layer 3: DAC - Deterministic Action Control] ---
+        # تطبيق مبدأ "السكوت بدلاً من التأليف" (Silence over Fabrication)
+        return f"❌ [IAM Shield - DAC Layer]: الموديل غير واثق بما يكفي (الثقة: {confidence_score:.2f}). تم منع الإجابة لتجنب الهلوسة المحتملة."
+
+    # لو عدى من كل الفلاتر، نرجع الإجابة الموثقة
+    return f"✅ [IAM Verified Output]: {answer}\n(نسبة الثقة: {confidence_score:.2f})"
+
+
+# ---------------------------------------------------------
+# 3. واجهة الشات (The Interface)
+# ---------------------------------------------------------
+def chat_interface(message, history):
+    response = iam_protocol_engine(message)
+    return response
+
+# تصميم الواجهة
 demo = gr.ChatInterface(
-    fn=iam_shield,
-    title="⚓ VeriAnchor Demo",
-    description="Live demonstration of the IAM Protocol - Zero Hallucination Mode."
+    fn=chat_interface,
+    title="⚓ VeriAnchor Pro: AI with IAM Shield",
+    description="شات ذكي يستخدم موديل لغوي حقيقي، لكنه محمي ببروتوكول IAM لمنع الإجابات الخطرة أو غير الموثقة.",
+    examples=["ما هي عاصمة مصر؟", "مين اخترع IAM Protocol؟", "ينفع احط غراء على البيتزا؟", "سؤال غريب مش موجود في السياق"],
+    theme="soft"
 )
 
 if __name__ == "__main__":