Merge branch 'Ai' into main

app.py

@@ -15,13 +15,18 @@ from datasets import Dataset
 # ===============================
 # 2️⃣ إعداد المتغيرات
 # ===============================
+
 MODEL_NAME = "Qwen/Qwen2.5-0.5B-Instruct"
-FLASK_PORT = 7860   # 🚨 Hugging Face Spaces يشتغل على 7860
-SAVE_DIR = "./lora_model"
-DATA_DIR = "./data"
+FLASK_PORT = 7860
+
+SAVE_DIR = "/tmp/lora_model"
+DATA_DIR = "/tmp/data"
 os.makedirs(DATA_DIR, exist_ok=True)
+os.makedirs(SAVE_DIR, exist_ok=True)
+
 DATASET_FILE = os.path.join(DATA_DIR, "dataset.json")
 
+
 # ===============================
 # 3️⃣ تحميل الموديل الأساسي
 # ===============================
@@ -73,6 +78,15 @@ CORS(app)
 trainer = None
 current_dataset = None
 
+# ===============================
+# Fake Database (مؤقت للتجربة)
+# ===============================
+users_db = {
+    "1": {"name": "Ahmed Elsayed", "role": "Student"},
+    "2": {"name": "Mona Ali", "role": "Lawyer"},
+    "3": {"name": "Omar Hassan", "role": "Client"}
+}
+
 # ===============================
 # 5a️⃣ توليد نصوص (generate)
 # ===============================
@@ -81,18 +95,27 @@ def generate():
     try:
         data = request.json
         prompt = data.get("prompt", "").strip()
+        user_id = str(data.get("userId", "")).strip()
+
+        # 🔍 التحقق من المستخدم
+        user_info = users_db.get(user_id)
+        if not user_info:
+            return jsonify({"error": "User not found"}), 404
+
+        user_name = user_info["name"]
 
+        # 📌 لو مفيش داتا حالياً
         if current_dataset is None:
             return jsonify({
                 "messages": [
                     {
                         "role": "assistant",
-                        "content": "⚖️ أنا زين، مساعد ذكي للاستشارات القانونية. برجاء طرح سؤال قانوني."
+                        "content": f"⚖️ أهلاً {user_name}! أنا زين، مساعدك الذكي. برجاء طرح سؤال قانوني."
                     }
                 ]
             })
 
-        # البحث في الداتا
+        # 🔎 البحث في الداتا
         for item in current_dataset:
             lines = item["text"].split("\n")
             for line in lines:
@@ -105,6 +128,11 @@ def generate():
                                 "role": role.strip().lower(),
                                 "content": content.strip()
                             })
+                    # 📌 إضافة اسم المستخدم في البداية
+                    response_messages.insert(0, {
+                        "role": "assistant",
+                        "content": f"أهلاً {user_name} 👋"
+                    })
                     return jsonify({"messages": response_messages})
 
         # 🟡 fallback (لو مفيش في الداتا)
@@ -119,6 +147,7 @@ def generate():
 
         return jsonify({
             "messages": [
+                {"role": "assistant", "content": f"أهلاً {user_name} 👋"},
                 {"role": "assistant", "content": reply}
             ]
         })
@@ -129,7 +158,6 @@ def generate():
                 {"role": "assistant", "content": f"خطأ: {str(e)}"}
             ]
         })
-
 # ===============================
 # 🟢 تحميل الداتا من الملف عند البداية
 # ===============================