Update app.py

app.py

@@ -11,353 +11,6 @@ from pydantic import BaseModel
 from torchvision import transforms
 from typing import List, Dict, Any, Optional
 import logging
-
-# # استيراد من ملفاتنا المحلية
-# from model_definition import InterfuserModel, load_and_prepare_model, create_model_config
-# from simulation_modules import (
-#     InterfuserController, ControllerConfig, Tracker, DisplayInterface, 
-#     render, render_waypoints, render_self_car, WAYPOINT_SCALE_FACTOR, 
-#     T1_FUTURE_TIME, T2_FUTURE_TIME
-# )
-
-# # إعداد التسجيل
-# logging.basicConfig(level=logging.INFO)
-# logger = logging.getLogger(__name__)
-
-# # ================== إعدادات عامة وتحميل النموذج ==================
-# app = FastAPI(
-#     title="Baseer Self-Driving API",
-#     description="API للقيادة الذاتية باستخدام نموذج InterFuser",
-#     version="1.0.0"
-# )
-
-# device = torch.device("cpu")
-# logger.info(f"Using device: {device}")
-
-# # تحميل النموذج باستخدام الدالة المحسنة
-# try:
-#     # إنشاء إعدادات النموذج باستخدام الإعدادات الصحيحة من التدريب
-#     model_config = create_model_config(
-#         model_path="model/best_model.pth"
-#         # الإعدادات الصحيحة من التدريب ستطبق تلقائياً:
-#         # embed_dim=256, rgb_backbone_name='r50', waypoints_pred_head='gru'
-#         # with_lidar=False, with_right_left_sensors=False, with_center_sensor=False
-#     )
-    
-#     # تحميل النموذج مع الأوزان
-#     model = load_and_prepare_model(model_config, device)
-#     logger.info("✅ تم تحميل النموذج بنجاح")
-    
-# except Exception as e:
-#     logger.error(f"❌ خطأ في تحميل النموذج: {e}")
-#     logger.info("🔄 محاولة إنشاء نموذج بأوزان عشوائية...")
-#     try:
-#         model = InterfuserModel()
-#         model.to(device)
-#         model.eval()
-#         logger.warning("⚠️ تم إنشاء النموذج بأوزان عشوائية")
-#     except Exception as e2:
-#         logger.error(f"❌ فشل في إنشاء النموذج: {e2}")
-#         model = None
-
-# # تهيئة واجهة العرض
-# display = DisplayInterface()
-
-# # قاموس لتخزين جلسات المستخدمين
-# SESSIONS: Dict[str, Dict] = {}
-
-# # ================== هياكل بيانات Pydantic ==================
-# class Measurements(BaseModel):
-#     pos: List[float] = [0.0, 0.0]  # [x, y] position
-#     theta: float = 0.0              # orientation angle
-#     speed: float = 0.0              # current speed
-#     steer: float = 0.0              # current steering
-#     throttle: float = 0.0           # current throttle
-#     brake: bool = False             # brake status
-#     command: int = 4                # driving command (4 = FollowLane)
-#     target_point: List[float] = [0.0, 0.0]  # target point [x, y]
-
-# class ModelOutputs(BaseModel):
-#     traffic: List[List[List[float]]]  # 20x20x7 grid
-#     waypoints: List[List[float]]      # Nx2 waypoints
-#     is_junction: float
-#     traffic_light_state: float
-#     stop_sign: float
-
-# class ControlCommands(BaseModel):
-#     steer: float
-#     throttle: float
-#     brake: bool
-
-# class RunStepInput(BaseModel):
-#     session_id: str
-#     image_b64: str
-#     measurements: Measurements
-
-# class RunStepOutput(BaseModel):
-#     model_outputs: ModelOutputs
-#     control_commands: ControlCommands
-#     dashboard_image_b64: str
-
-# class SessionResponse(BaseModel):
-#     session_id: str
-#     message: str
-
-# # ================== دوال المساعدة ==================
-# def get_image_transform():
-#     """إنشاء تحويلات الصورة كما في PDMDataset"""
-#     return transforms.Compose([
-#         transforms.ToTensor(),
-#         transforms.Resize((224, 224), antialias=True),
-#         transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
-#     ])
-
-# # إنشاء كائن التحويل مرة واحدة
-# image_transform = get_image_transform()
-
-# def preprocess_input(frame_rgb: np.ndarray, measurements: Measurements, device: torch.device) -> Dict[str, torch.Tensor]:
-#     """
-#     تحاكي ما يفعله PDMDataset.__getitem__ لإنشاء دفعة (batch) واحدة.
-#     """
-#     # 1. معالجة الصورة الرئيسية
-#     from PIL import Image
-#     if isinstance(frame_rgb, np.ndarray):
-#         frame_rgb = Image.fromarray(frame_rgb)
-    
-#     image_tensor = image_transform(frame_rgb).unsqueeze(0).to(device)  # إضافة بُعد الدفعة
-
-#     # 2. إنشاء مدخلات الكاميرات الأخرى عن طريق الاستنساخ
-#     batch = {
-#         'rgb': image_tensor,
-#         'rgb_left': image_tensor.clone(),
-#         'rgb_right': image_tensor.clone(),
-#         'rgb_center': image_tensor.clone(),
-#     }
-
-#     # 3. إنشاء مدخل ليدار وهمي (أصفار)
-#     batch['lidar'] = torch.zeros(1, 3, 224, 224, dtype=torch.float32).to(device)
-
-#     # 4. تجميع القياسات بنفس ترتيب PDMDataset
-#     m = measurements
-#     measurements_tensor = torch.tensor([[
-#         m.pos[0], m.pos[1], m.theta,
-#         m.steer, m.throttle, float(m.brake),
-#         m.speed, float(m.command)
-#     ]], dtype=torch.float32).to(device)
-#     batch['measurements'] = measurements_tensor
-
-#     # 5. إنشاء نقطة هدف
-#     batch['target_point'] = torch.tensor([m.target_point], dtype=torch.float32).to(device)
-    
-#     # لا نحتاج إلى قيم ground truth (gt_*) أثناء التنبؤ
-#     return batch
-
-# def decode_base64_image(image_b64: str) -> np.ndarray:
-#     """
-#     فك تشفير صورة Base64
-#     """
-#     try:
-#         image_bytes = base64.b64decode(image_b64)
-#         nparr = np.frombuffer(image_bytes, np.uint8)
-#         image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
-#         return image
-#     except Exception as e:
-#         raise HTTPException(status_code=400, detail=f"Invalid image format: {str(e)}")
-
-# def encode_image_to_base64(image: np.ndarray) -> str:
-#     """
-#     تشفير صورة إلى Base64
-#     """
-#     _, buffer = cv2.imencode('.jpg', image, [cv2.IMWRITE_JPEG_QUALITY, 85])
-#     return base64.b64encode(buffer).decode('utf-8')
-
-# # ================== نقاط نهاية الـ API ==================
-# @app.get("/", response_class=HTMLResponse)
-# async def root():
-#     """
-#     الصفحة الرئيسية للـ API
-#     """
-#     html_content = f"""
-#     <!DOCTYPE html>
-#     <html dir="rtl" lang="ar">
-#     <head>
-#         <meta charset="UTF-8">
-#         <meta name="viewport" content="width=device-width, initial-scale=1.0">
-#         <title>🚗 Baseer Self-Driving API</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;
-#                 display: flex;
-#                 align-items: center;
-#                 justify-content: center;
-#                 padding: 20px;
-#             }}
-#             .container {{
-#                 background: rgba(255, 255, 255, 0.95);
-#                 backdrop-filter: blur(10px);
-#                 border-radius: 20px;
-#                 padding: 40px;
-#                 box-shadow: 0 20px 40px rgba(0, 0, 0, 0.1);
-#                 text-align: center;
-#                 max-width: 600px;
-#                 width: 100%;
-#             }}
-#             .logo {{
-#                 font-size: 4rem;
-#                 margin-bottom: 20px;
-#                 animation: bounce 2s infinite;
-#             }}
-#             @keyframes bounce {{
-#                 0%, 20%, 50%, 80%, 100% {{ transform: translateY(0); }}
-#                 40% {{ transform: translateY(-10px); }}
-#                 60% {{ transform: translateY(-5px); }}
-#             }}
-#             h1 {{
-#                 color: #333;
-#                 margin-bottom: 10px;
-#                 font-size: 2.5rem;
-#             }}
-#             .subtitle {{
-#                 color: #666;
-#                 margin-bottom: 30px;
-#                 font-size: 1.2rem;
-#             }}
-#             .status {{
-#                 display: inline-block;
-#                 background: #4CAF50;
-#                 color: white;
-#                 padding: 8px 16px;
-#                 border-radius: 20px;
-#                 margin: 10px 0;
-#                 font-weight: bold;
-#             }}
-#             .stats {{
-#                 display: grid;
-#                 grid-template-columns: repeat(auto-fit, minmax(150px, 1fr));
-#                 gap: 20px;
-#                 margin: 30px 0;
-#             }}
-#             .stat-card {{
-#                 background: #f8f9fa;
-#                 padding: 20px;
-#                 border-radius: 15px;
-#                 border-left: 4px solid #667eea;
-#             }}
-#             .stat-number {{
-#                 font-size: 2rem;
-#                 font-weight: bold;
-#                 color: #667eea;
-#             }}
-#             .stat-label {{
-#                 color: #666;
-#                 margin-top: 5px;
-#             }}
-#             .buttons {{
-#                 display: flex;
-#                 gap: 15px;
-#                 justify-content: center;
-#                 flex-wrap: wrap;
-#                 margin-top: 30px;
-#             }}
-#             .btn {{
-#                 display: inline-block;
-#                 padding: 12px 24px;
-#                 border-radius: 25px;
-#                 text-decoration: none;
-#                 font-weight: bold;
-#                 transition: all 0.3s ease;
-#                 border: none;
-#                 cursor: pointer;
-#             }}
-#             .btn-primary {{
-#                 background: #667eea;
-#                 color: white;
-#             }}
-#             .btn-secondary {{
-#                 background: #6c757d;
-#                 color: white;
-#             }}
-#             .btn:hover {{
-#                 transform: translateY(-2px);
-#                 box-shadow: 0 5px 15px rgba(0, 0, 0, 0.2);
-#             }}
-#             .features {{
-#                 text-align: right;
-#                 margin-top: 30px;
-#                 padding: 20px;
-#                 background: #f8f9fa;
-#                 border-radius: 15px;
-#             }}
-#             .features h3 {{
-#                 color: #333;
-#                 margin-bottom: 15px;
-#             }}
-#             .features ul {{
-#                 list-style: none;
-#                 padding: 0;
-#             }}
-#             .features li {{
-#                 padding: 5px 0;
-#                 color: #666;
-#             }}
-#             .features li:before {{
-#                 content: "✅ ";
-#                 margin-left: 10px;
-#             }}
-#         </style>
-#     </head>
-#     <body>
-#         <div class="container">
-#             <div class="logo">🚗</div>
-#             <h1>Baseer Self-Driving API</h1>
-#             <p class="subtitle">نظام القيادة الذاتية المتقدم</p>
-            
-#             <div class="status">🟢 يعمل بنجاح</div>
-            
-#             <div class="stats">
-#                 <div class="stat-card">
-#                     <div class="stat-number">{len(SESSIONS)}</div>
-#                     <div class="stat-label">الجلسات النشطة</div>
-#                 </div>
-#                 <div class="stat-card">
-#                     <div class="stat-number">v1.0</div>
-#                     <div class="stat-label">الإصدار</div>
-#                 </div>
-#                 <div class="stat-card">
-#                     <div class="stat-number">FastAPI</div>
-#                     <div class="stat-label">التقنية</div>
-#                 </div>
-#             </div>
-            
-#             <div class="buttons">
-#                 <a href="/docs" class="btn btn-primary">📚 توثيق API</a>
-#                 <a href="/sessions" class="btn btn-secondary">📊 الجلسات</a>
-#             </div>
-            
-#             <div class="features">
-#                 <h3>🌟 الميزات الرئيسية</h3>
-#                 <ul>
-#                     <li>نموذج InterFuser للقيادة الذاتية</li>
-#                     <li>معالجة الصور في الوقت الفعلي</li>
-#                     <li>اكتشاف الكائنات المرورية</li>
-#                     <li>تحديد المسارات الذكية</li>
-#                     <li>واجهة RESTful سهلة الاستخدام</li>
-#                     <li>إدارة جلسات متعددة</li>
-#                 </ul>
-#             </div>
-#         </div>
-#     </body>
-#     </html>
-#     """
-#     return html_content
-
 import uuid
 import base64
 import cv2
@@ -369,23 +22,12 @@ from fastapi.responses import HTMLResponse
 from pydantic import BaseModel, Field
 from typing import List, Dict, Tuple
 
-# ==============================================================================
-# 1. استيراد كل مكونات المشروع التي قمنا بتطويرها
-#    (تأكد من أن هذه الملفات موجودة في نفس المجلد)
-# ==============================================================================
-# من ملف النموذج (يحتوي على كلاس Interfuser والدوال المساعدة)
 from model_definition import InterfuserModel, load_and_prepare_model, get_master_config
 
-# من ملفات التحكم والعرض
+
 from simulation_modules import InterfuserController, Tracker
 from simulation_modules import DisplayInterface, render_bev, unnormalize_image, DisplayConfig
-# # استيراد من ملفاتنا المحلية
-# from model_definition import InterfuserModel, load_and_prepare_model, create_model_config
-# from simulation_modules import (
-#     InterfuserController, ControllerConfig, Tracker, DisplayInterface, 
-#     render, render_waypoints, render_self_car, WAYPOINT_SCALE_FACTOR, 
-#     T1_FUTURE_TIME, T2_FUTURE_TIME
-# )
+
 # ==============================================================================
 # 2. إعدادات عامة وتطبيق FastAPI
 # ==============================================================================
@@ -477,9 +119,7 @@ def prepare_model_input(image: np.ndarray, measurements: Measurements) -> Dict[s
         'target_point': target_point_tensor,
         'lidar': torch.zeros_like(image_tensor)
     }
-# model_config = create_model_config(
-#     model_path="model/best_model.pth"
-# )
+
 # ==============================================================================
 # 5. أحداث دورة حياة التطبيق (Startup/Shutdown)
 # ==============================================================================
@@ -494,44 +134,198 @@ async def startup_event():
         logging.info("✅ Model loaded successfully. Server is ready!")
     else:
         logging.error("❌ CRITICAL: Model could not be loaded. The API will not function correctly.")
-# # تحميل النموذج باستخدام الدالة المحسنة
-# try:
-#     # إنشاء إعدادات النموذج باستخدام الإعدادات الصحيحة من التدريب
-
-    
-#     # تحميل النموذج مع الأوزان
-#     
-#     logger.info("✅ تم تحميل النموذج بنجاح")
-    
-# except Exception as e:
-#     logger.error(f"❌ خطأ في تحميل النموذج: {e}")
-#     logger.info("🔄 محاولة إنشاء نموذج بأوزان عشوائية...")
-#     try:
-#         model = InterfuserModel()
-#         model.to(device)
-#         model.eval()
-#         logger.warning("⚠️ تم إنشاء النموذج بأوزان عشوائية")
-#     except Exception as e2:
-#         logger.error(f"❌ فشل في إنشاء النموذج: {e2}")
-#         model = None
 
 # ==============================================================================
 # 6. نقاط النهاية الرئيسية (API Endpoints)
 # ==============================================================================
-@app.get("/", response_class=HTMLResponse, include_in_schema=False)
+
+@app.get("/", response_class=HTMLResponse, include_in_schema=False, tags=["General"])
 async def root():
-    # هذا يعرض صفحة رئيسية بسيطة وجميلة للمستخدمين
-    return """
-    <html>
-        <head><title>Baseer API</title></head>
-        <body style='font-family: sans-serif; text-align: center; padding-top: 50px;'>
-            <h1>🚗 Baseer Self-Driving API</h1>
-            <p>Welcome! The API is running.</p>
-            <p>Navigate to <a href="/docs">/docs</a> for the interactive API documentation.</p>
-        </body>
+    """
+    يعرض صفحة رئيسية احترافية وتفاعلية لواجهة برمجة التطبيقات.
+    """
+    # الحصول على عدد الجلسات النشطة حاليًا
+    active_sessions_count = len(SESSIONS)
+    
+    # تحديد لون حالة النظام
+    status_color = "#4CAF50" # أخضر
+    status_text = "يعمل بنجاح"
+    if MODEL is None:
+        status_color = "#F44336" # أحمر
+        status_text = "خطأ: النموذج غير محمل"
+
+    html_content = f"""
+    <!DOCTYPE html>
+    <html dir="rtl" lang="ar">
+    <head>
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <title>🚗 Baseer Self-Driving API</title>
+        <style>
+            @import url('https://fonts.googleapis.com/css2?family=Tajawal:wght@400;700&display=swap');
+            
+            :root {{
+                --primary-color: #4a69bd;
+                --secondary-color: #6a89cc;
+                --text-color: #333;
+                --bg-color: #f4f7f6;
+                --panel-bg: #ffffff;
+                --shadow: 0 10px 30px rgba(0, 0, 0, 0.1);
+            }}
+
+            body {{
+                font-family: 'Tajawal', sans-serif;
+                background-color: var(--bg-color);
+                color: var(--text-color);
+                margin: 0;
+                display: flex;
+                justify-content: center;
+                align-items: center;
+                min-height: 100vh;
+                padding: 20px;
+            }}
+
+            .container {{
+                background: var(--panel-bg);
+                border-radius: 20px;
+                padding: 40px;
+                box-shadow: var(--shadow);
+                text-align: center;
+                max-width: 700px;
+                width: 100%;
+                border-top: 5px solid var(--primary-color);
+            }}
+
+            .logo {{
+                font-size: 4rem;
+                margin-bottom: 15px;
+                animation: car-drive 3s ease-in-out infinite;
+            }}
+
+            @keyframes car-drive {{
+                0% {{ transform: translateX(-20px); }}
+                50% {{ transform: translateX(20px); }}
+                100% {{ transform: translateX(-20px); }}
+            }}
+
+            h1 {{
+                font-size: 2.8rem;
+                font-weight: 700;
+                color: var(--primary-color);
+                margin-bottom: 10px;
+            }}
+
+            .subtitle {{
+                font-size: 1.2rem;
+                color: #777;
+                margin-bottom: 25px;
+            }}
+            
+            .status-badge {{
+                display: inline-block;
+                background-color: {status_color};
+                color: white;
+                padding: 10px 20px;
+                border-radius: 25px;
+                font-weight: bold;
+                margin-bottom: 30px;
+                font-size: 1rem;
+            }}
+
+            .stats-grid {{
+                display: grid;
+                grid-template-columns: 1fr 1fr;
+                gap: 20px;
+                margin-bottom: 30px;
+            }}
+
+            .stat-card {{
+                background: #f8f9fa;
+                padding: 20px;
+                border-radius: 15px;
+            }}
+
+            .stat-number {{
+                font-size: 2.5rem;
+                font-weight: 700;
+                color: var(--primary-color);
+            }}
+
+            .stat-label {{
+                font-size: 1rem;
+                color: #666;
+                margin-top: 5px;
+            }}
+
+            .button-group {{
+                display: flex;
+                gap: 15px;
+                justify-content: center;
+            }}
+
+            .btn {{
+                padding: 14px 28px;
+                border-radius: 30px;
+                text-decoration: none;
+                font-weight: bold;
+                font-size: 1rem;
+                transition: all 0.3s ease;
+                border: 2px solid transparent;
+                cursor: pointer;
+            }}
+            
+            .btn-primary {{
+                background: var(--primary-color);
+                color: white;
+            }}
+
+            .btn-primary:hover {{
+                background: var(--secondary-color);
+                transform: translateY(-3px);
+            }}
+
+            .btn-secondary {{
+                background: transparent;
+                color: var(--primary-color);
+                border-color: var(--primary-color);
+            }}
+            
+            .btn-secondary:hover {{
+                background: var(--primary-color);
+                color: white;
+                transform: translateY(-3px);
+            }}
+
+        </style>
+    </head>
+    <body>
+        <div class="container">
+            <div class="logo">🚗</div>
+            <h1>Baseer Self-Driving API</h1>
+            <p class="subtitle">واجهة برمجية متقدمة للقيادة الذاتية</p>
+            
+            <div class="status-badge">{status_text}</div>
+            
+            <div class="stats-grid">
+                <div class="stat-card">
+                    <div class="stat-number">{active_sessions_count}</div>
+                    <div class="stat-label">الجلسات النشطة</div>
+                </div>
+                <div class="stat-card">
+                    <div class="stat-number">1.1.0</div>
+                    <div class="stat-label">إصدار الـ API</div>
+                </div>
+            </div>
+            
+            <div class="button-group">
+                <a href="/docs" target="_blank" class="btn btn-primary">📚 التوثيق التفاعلي (Docs)</a>
+                <a href="/sessions" target="_blank" class="btn btn-secondary">📊 عرض الجلسات</a>
+            </div>
+        </div>
+    </body>
     </html>
     """
-# في ملف app.py
+    return HTMLResponse(content=html_content)
 
 @app.post("/start_session", summary="Start a new driving session", tags=["Session Management"])
 def start_session():