Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,238 @@
+# app.py - الكود النهائي المعدل ليتوافق مع بنية measurements.json الحقيقية
+
+import os
+import json
+import gradio as gr
+import base64
+from PIL import Image
+import io
+import requests
+
+# ==============================================================================
+#           1. إعدادات الـ API
+# ==============================================================================
+API_URL_BASE = "https://BaseerAI-baseer-server.hf.space"
+API_START_SESSION_URL = f"{API_URL_BASE}/start_session"
+API_RUN_STEP_URL = f"{API_URL_BASE}/run_step"
+API_END_SESSION_URL = f"{API_URL_BASE}/end_session"
+
+EXAMPLES_DIR = "examples" 
+
+# ==============================================================================
+#           2. الدوال المساعدة والدوال التي تتصل بالـ API
+# ==============================================================================
+
+def image_to_base64(pil_image):
+    buffered = io.BytesIO()
+    pil_image.save(buffered, format="JPEG")
+    return base64.b64encode(buffered.getvalue()).decode('utf-8')
+
+def base64_to_image(b64_string):
+    img_data = base64.b64decode(b64_string)
+    return Image.open(io.BytesIO(img_data))
+
+def start_new_session():
+    """يبدأ جلسة جديدة على الخادم ويعيد معرف الجلسة وأمر إظهار الواجهة."""
+    print("Requesting to start a new session...")
+    try:
+        response = requests.post(API_START_SESSION_URL, timeout=30)
+        response.raise_for_status()
+        session_data = response.json()
+        session_id = session_data.get("session_id")
+        if not session_id:
+            raise gr.Error("لم يتمكن الخادم من بدء جلسة جديدة.")
+        
+        status_message = f"✅ تم بدء جلسة جديدة بنجاح. أنت الآن متصل بالخادم."
+        print(f"Session started successfully: {session_id}")
+        return session_id, status_message, gr.update(visible=True)
+    except requests.exceptions.RequestException as e:
+        error_message = f"فشل الاتصال بخادم الـ API: {e}"
+        print(error_message)
+        raise gr.Error(error_message)
+
+def handle_end_session(session_id):
+    """ينهي الجلسة الحالية على الخادم."""
+    if not session_id:
+        return "لا توجد جلسة نشطة لإنهائها.", None, gr.update(visible=False)
+    
+    print(f"Requesting to end session: {session_id}")
+    try:
+        response = requests.post(f"{API_END_SESSION_URL}?session_id={session_id}", timeout=30)
+        response.raise_for_status()
+        
+        status_message = f"🔌 تم إنهاء الجلسة بنجاح. انقطع الاتصال بالخادم."
+        print(status_message)
+        return status_message, None, gr.update(visible=False)
+    except requests.exceptions.RequestException as e:
+        error_message = f"حدث خطأ أثناء إنهاء الجلسة: {e}"
+        print(error_message)
+        raise gr.Error(error_message)
+
+# --- [تم تعديل هذه الدالة] ---
+def run_single_frame_via_api(session_id, rgb_image_path, measurements_path):
+    if not session_id:
+        raise gr.Error("لا توجد جلسة نشطة. يرجى بدء جلسة جديدة أولاً.")
+    if not (rgb_image_path and measurements_path):
+        raise gr.Error("الرجاء توفير الصورة الأمامية وملف القياسات على الأقل.")
+    try:
+        rgb_image_pil = Image.open(rgb_image_path).convert("RGB")
+        with open(measurements_path, 'r') as f:
+            measurements_dict = json.load(f)
+        image_b64 = image_to_base64(rgb_image_pil)
+        
+        # --- [التعديل الرئيسي هنا] ---
+        # بناء الحمولة (Payload) بناءً على بنية الملف الحقيقية
+        payload = {
+            "session_id": session_id,
+            "image_b64": image_b64,
+            "measurements": {
+                "pos_global": measurements_dict.get('pos_global', [0.0, 0.0]),
+                "theta": measurements_dict.get('theta', 0.0),
+                "speed": measurements_dict.get('speed', 0.0),
+                "target_point": measurements_dict.get('target_point', [0.0, 100.0])
+            }
+        }
+        
+        print(f"Sending run_step request for session: {session_id}")
+        response = requests.post(API_RUN_STEP_URL, json=payload, timeout=60)
+        response.raise_for_status()
+        api_result = response.json()
+        dashboard_b64 = api_result.get("dashboard_b64")
+        control_commands = api_result.get("control_commands", {})
+        if not dashboard_b64:
+            raise gr.Error("لم يتم إرجاع صورة لوحة التحكم من الـ API.")
+        output_image = base64_to_image(dashboard_b64)
+        return output_image, control_commands
+    except requests.exceptions.RequestException as e:
+        raise gr.Error(f"حدث خطأ أثناء الاتصال بالـ API: {e}")
+    except Exception as e:
+        raise gr.Error(f"حدث خطأ غير متوقع: {e}")
+
+# ==============================================================================
+#           4. تعريف واجهة Gradio
+# ==============================================================================
+
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"), css=".gradio-container {max-width: 100% !important;}") as demo:
+    session_id_state = gr.State(value=None)
+    
+    gr.Markdown("# 🚗 واجهة التحكم لنظام Baseer للقيادة الذاتية")
+    gr.Markdown("هذه الواجهة تعمل كـ **عميل (Client)** يتصل بـ **[Baseer Self-Driving API](https://huggingface.co/spaces/BaseerAI/baseer-server)** لمعالجة البيانات.")
+
+    gr.Markdown("---")
+
+    with gr.Group():
+        gr.Markdown("## ⚙️ التحكم بالجلسة")
+        with gr.Row():
+            start_session_button = gr.Button("🟢 بدء جلسة جديدة", variant="primary")
+            end_session_button = gr.Button("🔴 إنهاء الجلسة الحالية")
+        status_textbox = gr.Textbox(label="حالة الجلسة", interactive=False, value="في انتظار بدء الجلسة...")
+
+    gr.Markdown("")
+
+    with gr.Group(visible=False) as main_controls_group:
+        with gr.Row():
+            # -- العمود الأيسر: الإعدادات والمدخلات --
+            with gr.Column(scale=2):
+                with gr.Group():
+                    gr.Markdown("### 🗂️ ارفع ملفات السيناريو")
+                    api_rgb_image_path = gr.Image(type="filepath", label="صورة الكاميرا الأمامية (RGB)")
+                    api_measurements_path = gr.File(label="ملف القياسات (JSON)", type="filepath")
+                    
+                    # --- [تم حذف gr.JSON من هنا] ---
+                    
+                    api_run_button = gr.Button("🚀 أرسل البيانات للمعالجة", variant="primary")
+
+                gr.Markdown("")
+                
+# ... (داخل with gr.Group(visible=False) as main_controls_group:)
+                with gr.Group():
+                    gr.Markdown("### ✨ أمثلة جاهزة")
+                    
+                    # التحقق من وجود مجلد الأمثلة لتجنب الأخطاء
+                    if os.path.isdir(EXAMPLES_DIR) and \
+                       os.path.exists(os.path.join(EXAMPLES_DIR, "sample1", "measurements.json")) and \
+                       os.path.exists(os.path.join(EXAMPLES_DIR, "sample2", "measurements.json")):
+                        
+                        # قراءة بيانات المثال الأول
+                        with open(os.path.join(EXAMPLES_DIR, "sample1", "measurements.json"), 'r') as f:
+                            data1 = json.load(f)
+                        
+                        # قراءة بيانات المثال الثاني
+                        with open(os.path.join(EXAMPLES_DIR, "sample2", "measurements.json"), 'r') as f:
+                            data2 = json.load(f)
+                        
+                        # --- [تعديل] 1: إنشاء مكونات جديدة لكل البيانات المستخدمة ---
+                        # سيتم استخدام عناوينها كعناوين للأعمدة في الجدول
+                        with gr.Column(visible=False):
+                            example_speed = gr.Textbox(label="السرعة (m/s)")
+                            example_theta = gr.Textbox(label="الاتجاه (Theta)")
+                            example_pos = gr.Textbox(label="الموقع العام")
+                            example_target = gr.Textbox(label="النقطة المستهدفة")
+
+                        gr.Examples(
+                            examples=[
+                                # --- [تعديل] 2: بناء البيانات للأعمدة الجديدة ---
+                                [
+                                    # المدخلات الأساسية
+                                    os.path.join(EXAMPLES_DIR, "sample1", "rgb.jpg"), 
+                                    os.path.join(EXAMPLES_DIR, "sample1", "measurements.json"),
+                                    # البيانات للعرض فقط
+                                    f"{data1.get('speed', 0.0):.2f}",
+                                    f"{data1.get('theta', 0.0):.2f}",
+                                    str(data1.get('pos_global', 'N/A')),
+                                    str(data1.get('target_point', 'N/A'))
+                                ],
+                                [
+                                    # المدخلات الأساسية
+                                    os.path.join(EXAMPLES_DIR, "sample2", "rgb.jpg"),
+                                    os.path.join(EXAMPLES_DIR, "sample2", "measurements.json"),
+                                    # البيانات للعرض فقط
+                                    f"{data2.get('speed', 0.0):.2f}",
+                                    f"{data2.get('theta', 0.0):.2f}",
+                                    str(data2.get('pos_global', 'N/A')),
+                                    str(data2.get('target_point', 'N/A'))
+                                ]
+                            ],
+                            # --- [تعديل] 3: تحديث المدخلات لتشمل كل الأعمدة ---
+                            inputs=[
+                                api_rgb_image_path,
+                                api_measurements_path,
+                                example_speed,
+                                example_theta,
+                                example_pos,
+                                example_target
+                            ],
+                            label="اختر سيناريو اختبار",
+                        )
+                    else:
+                        gr.Markdown("لم يتم العثور على مجلد الأمثلة (`examples`) أو محتوياته.")
+
+            # -- العمود الأيمن: المخرجات --
+            with gr.Column(scale=3):
+                with gr.Group():
+                    gr.Markdown("### 📊 النتائج من الخادم")
+                    api_output_image = gr.Image(label="لوحة التحكم المرئية (من الـ API)", type="pil", interactive=False, height=600)
+                    api_control_json = gr.JSON(label="أوامر التحكم (من الـ API)")
+
+    # --- ربط منطق الواجهة ---
+    start_session_button.click(
+        fn=start_new_session,
+        inputs=None,
+        outputs=[session_id_state, status_textbox, main_controls_group]
+    )
+    
+    end_session_button.click(
+        fn=handle_end_session,
+        inputs=[session_id_state],
+        outputs=[status_textbox, session_id_state, main_controls_group]
+    )
+
+    # --- [تم تعديل هذا الجزء] ---
+    api_run_button.click(
+        fn=run_single_frame_via_api,
+        inputs=[session_id_state, api_rgb_image_path, api_measurements_path], # تم حذف target_point_list
+        outputs=[api_output_image, api_control_json],
+    )
+
+if __name__ == "__main__":
+    demo.queue().launch(debug=True)