Update app.py

app.py

@@ -2,12 +2,48 @@ import gradio as gr
 import requests
 import json
 import os
+import pandas as pd
+from datasets import load_dataset
 
-# [1] إعداد الاتصال المباشر (Raw API)
+# [1] إعدادات الاتصال والبيئة
 HF_TOKEN = os.environ.get("HF_TOKEN")
 API_URL = "https://api-inference.huggingface.co/models/aab20abdullah/Akin-Yurt-Llama3-8B"
 
-# [2] ميثاق القسم التركماني والهوية
+# [2] تحميل قاعدة بيانات الشهداء (الذاكرة الوطنية)
+print("📥 Şehitler Veritabanı Yükleniyor...")
+try:
+    # نقوم بتحميل البيانات باستخدام التوكن للوصول للمستودعات الخاصة
+    dataset = load_dataset("aab20abdullah/turkmen-martyrs-dataset", split="train", token=HF_TOKEN)
+    martyrs_df = dataset.to_pandas()
+    print(f"✅ Başarılı: {len(martyrs_df)} kayıt yüklendi.")
+except Exception as e:
+    martyrs_df = pd.DataFrame()
+    print(f"⚠️ Veritabanı Yükleme Hatası: {e}")
+
+# دالة البحث السريع في قاعدة البيانات
+def search_martyrs(query):
+    if martyrs_df.empty:
+        return ""
+    
+    query_words = query.lower().split()
+    results = []
+    
+    # بحث بسيط يطابق كلمات السؤال مع سجلات قاعدة البيانات
+    for _, row in martyrs_df.iterrows():
+        # دمج كل محتوى الصف في نص واحد للبحث
+        row_text = " | ".join([str(val) for val in row.values])
+        # إذا تطابقت أي كلمة مهمة من السؤال مع السجل (أطول من 3 حروف لتجنب الحروف البسيطة)
+        if any(word in row_text.lower() for word in query_words if len(word) > 3):
+            results.append(row_text)
+            # نكتفي بأهم 3 نتائج لعدم إرباك ذاكرة النموذج
+            if len(results) >= 3:
+                break
+                
+    if results:
+        return "\n\n[RESMİ VERİTABANI BİLGİSİ - BU BİLGİYİ KULLANARAK CEVAP VER]:\n" + "\n".join(results)
+    return ""
+
+# [3] ميثاق القسم التركماني والهوية
 UKÜCÜ_YEMINI = """ALLAHA, KURANA, YEMİN OLSUN.
 ŞEHİTLERİM, GAZİLERİM, EMİN OLSUN.
 ÜLKÜCÜ TÜRK GENÇLİĞİ OLARAK:
@@ -19,30 +55,32 @@ YILANLARDAN OLMAYACAĞIZ! SATANLARDAN OLMAYACAĞIZ! KAÇANLARDAN OLMAYACAĞIZ!
 YIKILMAYACAĞIZ. BAŞARACAĞIZ, BAŞARACAĞIZ, BAŞARACAĞIZ.
 TANRI TÜRKÜ KORUSUN VE YÜCELTSİN. (AMİN)"""
 
-SYSTEM_PROMPT = """Senin adın Akın Yurt. Kerkük'te görev yapan، Türkmeneli davasına adanmış، zeki bir asistansın.
+BASE_SYSTEM_PROMPT = """Senin adın Akın Yurt. Kerkük'te görev yapan، Türkmeneli davasına adanmış، zeki bir asistansın.
 Geliştiricilerin: Abdullah Ali Bahaaldeen، Ali El-Hürmüzlü ve Usame Alaaddin. TÖGB birimi tarafından geliştirildin."""
 
-# [3] وظيفة المعالجة والاتصال بالخادم
+# [4] المعالجة المركزية
 def respond(message, history):
-    # فحص القسم
+    # 1. الاستجابة الفورية للقسم
     yemin_triggers = ["yemin", "yemin et", "andı oku", "yemini oku"]
     if any(trigger in message.lower() for trigger in yemin_triggers):
         yield UKÜCÜ_YEMINI
         return
 
-    # فحص التوكن
     if not HF_TOKEN:
         yield "⚠️ HATA: HF_TOKEN bulunamadı. Lütfen Space ayarlarından ekleyin."
         return
 
-    # بناء هيكل المحادثة لمعمارية Llama-3
-    prompt = f"<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\n{SYSTEM_PROMPT}<|eot_id|>"
+    # 2. البحث في قاعدة بيانات الشهداء ودمجها مع شخصية النموذج
+    db_context = search_martyrs(message)
+    dynamic_system_prompt = BASE_SYSTEM_PROMPT + db_context
+
+    # 3. بناء هيكل المحادثة (Prompt Engineering)
+    prompt = f"<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\n{dynamic_system_prompt}<|eot_id|>"
     for user_msg, assistant_msg in history:
         if user_msg: prompt += f"<|start_header_id|>user<|end_header_id|>\n\n{user_msg}<|eot_id|>"
         if assistant_msg: prompt += f"<|start_header_id|>assistant<|end_header_id|>\n\n{assistant_msg}<|eot_id|>"
     prompt += f"<|start_header_id|>user<|end_header_id|>\n\n{message}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
 
-    # تجهيز الطلب المباشر للخادم
     headers = {
         "Authorization": f"Bearer {HF_TOKEN}",
         "Content-Type": "application/json"
@@ -55,23 +93,21 @@ def respond(message, history):
             "top_p": 0.95,
             "return_full_text": False
         },
-        "stream": True # تفعيل البث المباشر للكلمات
+        "stream": True
     }
 
+    # 4. الاتصال المباشر وتوليد الرد
     try:
-        # إرسال الطلب
         response = requests.post(API_URL, headers=headers, json=payload, stream=True)
         
-        # معالجة أخطاء الخادم (مثل: النموذج قيد التحميل)
         if response.status_code != 200:
             error_data = response.json()
             if "currently loading" in str(error_data):
-                yield f"⏳ Akın Yurt şu an sunucuda uyanıyor... Lütfen {error_data.get('estimated_time', 20)} saniye bekleyip tekrar yazın."
+                yield f"⏳ Akın Yurt uyanıyor... Lütfen {error_data.get('estimated_time', 20)} saniye bekleyip tekrar yazın."
             else:
                 yield f"⚠️ Sunucu Hatası ({response.status_code}): {error_data.get('error', str(error_data))}"
             return
 
-        # قراءة البث المباشر (Streaming)
         output_text = ""
         for line in response.iter_lines():
             if line:
@@ -84,30 +120,30 @@ def respond(message, history):
                         data_json = json.loads(data_str)
                         token = data_json.get("token", {}).get("text", "")
                         output_text += token
-                        # إرسال الكلمات للواجهة وتجاهل وسوم Llama
                         yield output_text.replace("<|eot_id|>", "").strip()
                     except:
                         continue
     except Exception as e:
         yield f"⚠️ API Bağlantı Çöktü: {str(e)}"
 
-# [4] الواجهة الرسومية الرسمية
-custom_theme = gr.themes.Soft(
-    primary_hue="blue",
-    secondary_hue="cyan",
-).set(
+# [5] الواجهة الرسومية
+custom_theme = gr.themes.Soft(primary_hue="blue", secondary_hue="cyan").set(
     button_primary_background_fill="#007bff",
     button_primary_text_color="white",
 )
 
 with gr.Blocks(theme=custom_theme, title="Akın Yurt - TÖGB") as demo:
     gr.Markdown("# 🐺 Akın Yurt")
-    gr.Markdown("### Türkmeneli Dijital Asistanı")
+    gr.Markdown("### Türkmeneli Dijital Asistanı (Şehitler Veritabanı Entegreli)")
     gr.Markdown("---")
     
     gr.ChatInterface(
         respond,
-        examples=["Seni kim yaptı?", "Yemin et", "Kerkük neresidir?"],
+        examples=[
+            "Seni kim yaptı?", 
+            "Andı oku", 
+            "Bize Türkmen şehitlerinden bahset"
+        ],
         cache_examples=False,
     )