Update app/app.py

app/app.py

@@ -4,26 +4,21 @@ from fastapi.staticfiles import StaticFiles
 import numpy as np
 import sqlite3
 import json
-import cv2
 from pathlib import Path
 import hashlib
 import time
-
-from services.detection import FaceDetector
-from services.embedding import FaceEmbedding
+import cv2
+from PIL import Image
+import io
 
 app = FastAPI(title="AEFRS Ultimate - Face Recognition System")
 
-# تهيئة الخدمات
-detector = FaceDetector()
-embedder = FaceEmbedding()
-
-# قاعدة البيانات
+# قاعدة بيانات بسيطة
 DB_PATH = "database/identities.db"
-VECTOR_INDEX_PATH = "database/vector_index/"
 
-# إنشاء الجداول
 def init_database():
+    """تهيئة قاعدة البيانات"""
+    Path(DB_PATH).parent.mkdir(parents=True, exist_ok=True)
     conn = sqlite3.connect(DB_PATH)
     cursor = conn.cursor()
     cursor.execute("""
@@ -31,8 +26,7 @@ def init_database():
             id TEXT PRIMARY KEY,
             name TEXT NOT NULL,
             embedding TEXT NOT NULL,
-            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
-            metadata TEXT
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
         )
     """)
     conn.commit()
@@ -40,37 +34,38 @@ def init_database():
 
 init_database()
 
-# API للتسجيل
+def get_face_embedding(image_bytes):
+    """استخراج بصمة الوجه (نسخة مبسطة للتجربة)"""
+    # هذه دالة مبسطة للتجربة فقط
+    # في الإنتاج، استخدم نموذج TFLite الفعلي
+    
+    # تحويل الصورة إلى مصفوفة
+    image = Image.open(io.BytesIO(image_bytes))
+    image = image.resize((112, 112))
+    img_array = np.array(image).flatten()[:512]  # خذ أول 512 قيمة
+    
+    # تطبيع المتجه
+    embedding = img_array / np.linalg.norm(img_array)
+    return embedding
+
 @app.post("/api/enroll")
-async def enroll_face(
-    name: str,
-    file: UploadFile = File(...)
-):
-    """تسجيل وجه جديد في النظام"""
+async def enroll_face(name: str, file: UploadFile = File(...)):
+    """تسجيل وجه جديد"""
     try:
-        # قراءة الصورة
         contents = await file.read()
-        nparr = np.frombuffer(contents, np.uint8)
-        image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
-        
-        # كشف الوجه
-        faces = detector.detect_faces(image)
-        if not faces:
-            raise HTTPException(status_code=400, detail="No face detected")
         
         # استخراج المتجه
-        face_roi = image[faces[0]['bbox'][1]:faces[0]['bbox'][3],
-                        faces[0]['bbox'][0]:faces[0]['bbox'][2]]
-        embedding = embedder.get_embedding(face_roi)
+        embedding = get_face_embedding(contents)
         
-        # حفظ في قاعدة البيانات
+        # إنشاء ID فريد
         identity_id = hashlib.md5(f"{name}_{time.time()}".encode()).hexdigest()[:16]
         
+        # حفظ في قاعدة البيانات
         conn = sqlite3.connect(DB_PATH)
         cursor = conn.cursor()
         cursor.execute(
-            "INSERT INTO identities (id, name, embedding, metadata) VALUES (?, ?, ?, ?)",
-            (identity_id, name, json.dumps(embedding.tolist()), json.dumps({"timestamp": time.time()}))
+            "INSERT INTO identities (id, name, embedding) VALUES (?, ?, ?)",
+            (identity_id, name, json.dumps(embedding.tolist()))
         )
         conn.commit()
         conn.close()
@@ -85,28 +80,14 @@ async def enroll_face(
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-# API للبحث
 @app.post("/api/search")
 async def search_face(file: UploadFile = File(...)):
     """البحث عن وجه مشابه"""
     try:
-        # قراءة وتحليل الصورة
         contents = await file.read()
-        nparr = np.frombuffer(contents, np.uint8)
-        image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
-        
-        # كشف الوجه
-        faces = detector.detect_faces(image)
-        if not faces:
-            return JSONResponse({
-                "success": False,
-                "message": "No face detected"
-            })
         
-        # استخراج المتجه
-        face_roi = image[faces[0]['bbox'][1]:faces[0]['bbox'][3],
-                        faces[0]['bbox'][0]:faces[0]['bbox'][2]]
-        query_embedding = embedder.get_embedding(face_roi)
+        # استخراج المتجه للبحث
+        query_embedding = get_face_embedding(contents)
         
         # البحث في قاعدة البيانات
         conn = sqlite3.connect(DB_PATH)
@@ -125,7 +106,7 @@ async def search_face(file: UploadFile = File(...)):
         # ترتيب النتائج
         similarities.sort(key=lambda x: x[2], reverse=True)
         
-        if similarities and similarities[0][2] > 0.6:  # عتبة 60%
+        if similarities and similarities[0][2] > 0.5:  # عتبة 50%
             return JSONResponse({
                 "success": True,
                 "identity_id": similarities[0][0],
@@ -142,14 +123,441 @@ async def search_face(file: UploadFile = File(...)):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-# تشغيل الواجهة
+@app.get("/api/identities")
+async def get_identities():
+    """الحصول على جميع الهويات المسجلة"""
+    try:
+        conn = sqlite3.connect(DB_PATH)
+        cursor = conn.cursor()
+        cursor.execute("SELECT id, name, created_at FROM identities ORDER BY created_at DESC")
+        rows = cursor.fetchall()
+        conn.close()
+        
+        identities = [
+            {"id": row[0], "name": row[1], "created_at": row[2]}
+            for row in rows
+        ]
+        
+        return JSONResponse({
+            "success": True,
+            "data": identities,
+            "count": len(identities)
+        })
+        
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.get("/healthz")
+async def health_check():
+    """فحص صحة النظام"""
+    return JSONResponse({
+        "status": "healthy",
+        "timestamp": time.time()
+    })
+
+# واجهة HTML
 @app.get("/", response_class=HTMLResponse)
 async def get_html():
-    with open("static/index.html", "r", encoding="utf-8") as f:
-        return f.read()
+    html_content = """
+    <!DOCTYPE html>
+    <html lang="ar" dir="rtl">
+    <head>
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <title>AEFRS Ultimate - نظام التعرف على الوجوه</title>
+        <style>
+            * {
+                margin: 0;
+                padding: 0;
+                box-sizing: border-box;
+            }
+            
+            body {
+                font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+                background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                min-height: 100vh;
+                padding: 20px;
+            }
+            
+            .container {
+                max-width: 800px;
+                margin: 0 auto;
+                background: white;
+                border-radius: 20px;
+                box-shadow: 0 20px 60px rgba(0,0,0,0.3);
+                overflow: hidden;
+            }
+            
+            .header {
+                background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                color: white;
+                padding: 30px;
+                text-align: center;
+            }
+            
+            .header h1 {
+                font-size: 2em;
+                margin-bottom: 10px;
+            }
+            
+            .tabs {
+                display: flex;
+                background: #f5f5f5;
+                border-bottom: 2px solid #e0e0e0;
+            }
+            
+            .tab-btn {
+                flex: 1;
+                padding: 15px;
+                background: none;
+                border: none;
+                cursor: pointer;
+                font-size: 16px;
+                font-weight: 600;
+                transition: all 0.3s;
+            }
+            
+            .tab-btn:hover {
+                background: #e0e0e0;
+            }
+            
+            .tab-btn.active {
+                background: white;
+                color: #667eea;
+                border-bottom: 3px solid #667eea;
+            }
+            
+            .tab-content {
+                display: none;
+                padding: 30px;
+                animation: fadeIn 0.5s;
+            }
+            
+            .tab-content.active {
+                display: block;
+            }
+            
+            @keyframes fadeIn {
+                from { opacity: 0; transform: translateY(10px); }
+                to { opacity: 1; transform: translateY(0); }
+            }
+            
+            .form-group {
+                margin-bottom: 20px;
+            }
+            
+            label {
+                display: block;
+                margin-bottom: 10px;
+                font-weight: 600;
+                color: #333;
+            }
+            
+            input[type="text"], input[type="file"] {
+                width: 100%;
+                padding: 12px;
+                border: 2px solid #e0e0e0;
+                border-radius: 10px;
+                font-size: 16px;
+                transition: border-color 0.3s;
+            }
+            
+            input[type="text"]:focus {
+                outline: none;
+                border-color: #667eea;
+            }
+            
+            .image-preview {
+                margin-top: 10px;
+                text-align: center;
+            }
+            
+            .image-preview img {
+                max-width: 200px;
+                border-radius: 10px;
+                box-shadow: 0 5px 15px rgba(0,0,0,0.2);
+            }
+            
+            .btn-primary {
+                width: 100%;
+                padding: 12px;
+                background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                color: white;
+                border: none;
+                border-radius: 10px;
+                font-size: 16px;
+                font-weight: 600;
+                cursor: pointer;
+                transition: transform 0.2s;
+            }
+            
+            .btn-primary:hover {
+                transform: translateY(-2px);
+            }
+            
+            .btn-secondary {
+                padding: 10px 20px;
+                background: #f5f5f5;
+                border: 2px solid #e0e0e0;
+                border-radius: 10px;
+                cursor: pointer;
+                margin-bottom: 20px;
+            }
+            
+            .result-message {
+                margin-top: 20px;
+                padding: 15px;
+                border-radius: 10px;
+                background: #f5f5f5;
+            }
+            
+            .alert-success {
+                background: #d4edda;
+                color: #155724;
+                border: 1px solid #c3e6cb;
+            }
+            
+            .alert-error {
+                background: #f8d7da;
+                color: #721c24;
+                border: 1px solid #f5c6cb;
+            }
+            
+            .alert-info {
+                background: #d1ecf1;
+                color: #0c5460;
+                border: 1px solid #bee5eb;
+            }
+            
+            .loading {
+                text-align: center;
+                padding: 20px;
+                color: #667eea;
+            }
+            
+            .database-list {
+                margin-top: 20px;
+            }
+            
+            .identity-card {
+                background: #f9f9f9;
+                padding: 15px;
+                margin-bottom: 10px;
+                border-radius: 10px;
+                border-right: 4px solid #667eea;
+            }
+            
+            .identity-card h3 {
+                color: #333;
+                margin-bottom: 5px;
+            }
+            
+            .identity-card p {
+                color: #666;
+                font-size: 14px;
+            }
+        </style>
+    </head>
+    <body>
+        <div class="container">
+            <div class="header">
+                <h1>🔒 AEFRS Ultimate</h1>
+                <p>نظام متقدم للتعرف على الوجوه</p>
+            </div>
+            
+            <div class="tabs">
+                <button class="tab-btn active" onclick="showTab('enroll')">📝 تسجيل وجه جديد</button>
+                <button class="tab-btn" onclick="showTab('search')">🔍 بحث عن وجه</button>
+                <button class="tab-btn" onclick="showTab('database')">🗄️ قاعدة البيانات</button>
+            </div>
+            
+            <div id="enroll" class="tab-content active">
+                <div class="form-group">
+                    <label>📸 صورة الوجه</label>
+                    <input type="file" id="enrollImage" accept="image/*">
+                    <div class="image-preview" id="enrollPreview"></div>
+                </div>
+                <div class="form-group">
+                    <label>👤 اسم الشخص</label>
+                    <input type="text" id="personName" placeholder="أدخل الاسم الكامل">
+                </div>
+                <button class="btn-primary" onclick="enrollFace()">✨ تسجيل الوجه</button>
+                <div id="enrollResult"></div>
+            </div>
+            
+            <div id="search" class="tab-content">
+                <div class="form-group">
+                    <label>🔍 صورة للبحث</label>
+                    <input type="file" id="searchImage" accept="image/*">
+                    <div class="image-preview" id="searchPreview"></div>
+                </div>
+                <button class="btn-primary" onclick="searchFace()">🔎 بحث</button>
+                <div id="searchResult"></div>
+            </div>
+            
+            <div id="database" class="tab-content">
+                <button class="btn-secondary" onclick="loadDatabase()">🔄 تحديث</button>
+                <div id="databaseList"></div>
+            </div>
+        </div>
+        
+        <script>
+            // معاينة الصورة
+            document.getElementById('enrollImage')?.addEventListener('change', function(e) {
+                previewImage(e, 'enrollPreview');
+            });
+            document.getElementById('searchImage')?.addEventListener('change', function(e) {
+                previewImage(e, 'searchPreview');
+            });
+            
+            function previewImage(event, previewId) {
+                const file = event.target.files[0];
+                if (file) {
+                    const reader = new FileReader();
+                    reader.onload = function(e) {
+                        const preview = document.getElementById(previewId);
+                        preview.innerHTML = `<img src="${e.target.result}" alt="Preview">`;
+                    };
+                    reader.readAsDataURL(file);
+                }
+            }
+            
+            async function enrollFace() {
+                const imageFile = document.getElementById('enrollImage').files[0];
+                const personName = document.getElementById('personName').value;
+                
+                if (!imageFile || !personName) {
+                    showResult('enrollResult', '❌ الرجاء اختيار صورة وإدخال الاسم', 'error');
+                    return;
+                }
+                
+                const formData = new FormData();
+                formData.append('file', imageFile);
+                
+                showLoading('enrollResult', 'جاري تسجيل الوجه...');
+                
+                try {
+                    const response = await fetch(`/api/enroll?name=${encodeURIComponent(personName)}`, {
+                        method: 'POST',
+                        body: formData
+                    });
+                    
+                    const result = await response.json();
+                    
+                    if (result.success) {
+                        showResult('enrollResult', `✅ ${result.message}<br>الرقم: ${result.identity_id}`, 'success');
+                        document.getElementById('personName').value = '';
+                        document.getElementById('enrollImage').value = '';
+                        document.getElementById('enrollPreview').innerHTML = '';
+                    } else {
+                        showResult('enrollResult', `❌ ${result.message}`, 'error');
+                    }
+                } catch (error) {
+                    showResult('enrollResult', '❌ حدث خطأ في الاتصال', 'error');
+                }
+            }
+            
+            async function searchFace() {
+                const imageFile = document.getElementById('searchImage').files[0];
+                
+                if (!imageFile) {
+                    showResult('searchResult', '❌ الرجاء اختيار صورة للبحث', 'error');
+                    return;
+                }
+                
+                const formData = new FormData();
+                formData.append('file', imageFile);
+                
+                showLoading('searchResult', 'جاري البحث...');
+                
+                try {
+                    const response = await fetch('/api/search', {
+                        method: 'POST',
+                        body: formData
+                    });
+                    
+                    const result = await response.json();
+                    
+                    if (result.success) {
+                        const similarityPercent = (result.similarity * 100).toFixed(2);
+                        showResult('searchResult', 
+                            `✅ تم العثور على تطابق!<br>👤 الاسم: ${result.name}<br>📊 نسبة التشابه: ${similarityPercent}%<br>🆔 الرقم: ${result.identity_id}`,
+                            'success');
+                    } else {
+                        showResult('searchResult', `❌ ${result.message}`, 'error');
+                    }
+                } catch (error) {
+                    showResult('searchResult', '❌ حدث خطأ في الاتصال', 'error');
+                }
+            }
+            
+            async function loadDatabase() {
+                showLoading('databaseList', 'جاري تحميل البيانات...');
+                
+                try {
+                    const response = await fetch('/api/identities');
+                    const result = await response.json();
+                    
+                    if (result.success && result.data.length > 0) {
+                        let html = '';
+                        result.data.forEach(person => {
+                            html += `
+                                <div class="identity-card">
+                                    <h3>${person.name}</h3>
+                                    <p>🆔 ${person.id}</p>
+                                    <p>📅 ${new Date(person.created_at).toLocaleDateString('ar-EG')}</p>
+                                </div>
+                            `;
+                        });
+                        document.getElementById('databaseList').innerHTML = html;
+                    } else {
+                        showResult('databaseList', '📭 لا يوجد أشخاص مسجلين', 'info');
+                    }
+                } catch (error) {
+                    showResult('databaseList', '❌ خطأ في تحميل البيانات', 'error');
+                }
+            }
+            
+            function showResult(elementId, message, type) {
+                const element = document.getElementById(elementId);
+                element.innerHTML = `<div class="result-message alert-${type}">${message}</div>`;
+                setTimeout(() => {
+                    if (element.innerHTML.includes('alert')) {
+                        element.innerHTML = '';
+                    }
+                }, 5000);
+            }
+            
+            function showLoading(elementId, message) {
+                document.getElementById(elementId).innerHTML = `<div class="loading">⏳ ${message}</div>`;
+            }
+            
+            function showTab(tabName) {
+                document.querySelectorAll('.tab-content').forEach(tab => {
+                    tab.classList.remove('active');
+                });
+                document.querySelectorAll('.tab-btn').forEach(btn => {
+                    btn.classList.remove('active');
+                });
+                
+                document.getElementById(tabName).classList.add('active');
+                event.target.classList.add('active');
+                
+                if (tabName === 'database') {
+                    loadDatabase();
+                }
+            }
+            
+            // تحميل قاعدة البيانات عند بدء التشغيل
+            loadDatabase();
+        </script>
+    </body>
+    </html>
+    """
+    return HTMLResponse(content=html_content)
 
-# تثبيت الملفات الثابتة
-app.mount("/static", StaticFiles(directory="static"), name="static")
+# تثبيت الملفات الثابتة (اختياري)
+app.mount("/static", StaticFiles(directory="static", html=True), name="static")
 
 if __name__ == "__main__":
     import uvicorn