Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py - النسخة النهائية والصحيحة
+# app.py - النسخة النهائية باستخدام gr.Dataset
 
 import os
 import json
@@ -23,18 +23,15 @@ EXAMPLES_DIR = "examples"
 # ==============================================================================
 
 def image_to_base64(pil_image):
-    """يحول صورة PIL إلى سلسلة نصية Base64."""
     buffered = io.BytesIO()
     pil_image.save(buffered, format="JPEG")
     return base64.b64encode(buffered.getvalue()).decode('utf-8')
 
 def base64_to_image(b64_string):
-    """يحول سلسلة نصية Base64 إلى صورة PIL."""
     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)
@@ -52,7 +49,6 @@ def start_new_session():
         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}")
@@ -68,13 +64,15 @@ def handle_end_session(session_id):
         raise gr.Error(error_message)
 
 def run_single_frame_via_api(session_id, rgb_image_path, measurements_path):
-    """يرسل إطارًا واحدًا للمعالجة عبر الـ API."""
     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")
+        # التأكد من أن المسار هو ملف صالح قبل فتحه
+        if not isinstance(measurements_path, str) or not os.path.exists(measurements_path):
+             raise gr.Error(f"مسار ملف القياسات غير صالح أو مفقود: {measurements_path}")
         with open(measurements_path, 'r') as f:
             measurements_dict = json.load(f)
         image_b64 = image_to_base64(rgb_image_pil)
@@ -97,36 +95,56 @@ def run_single_frame_via_api(session_id, rgb_image_path, measurements_path):
             raise gr.Error("لم يتم إرجاع صورة لوحة التحكم من الـ API.")
         output_image = base64_to_image(dashboard_b64)
         return output_image, control_commands
+    except FileNotFoundError:
+        raise gr.Error(f"لم يتم العثور على الملف: {measurements_path}")
     except requests.exceptions.RequestException as e:
         raise gr.Error(f"حدث خطأ أثناء الاتصال بالـ API: {e}")
     except Exception as e:
         raise gr.Error(f"حدث خطأ غير متوقع: {e}")
 
+# ==============================================================================
+#           3. إعداد بيانات الأمثلة
+# ==============================================================================
+# قائمة تحتوي على البيانات الكاملة لكل مثال
+FULL_EXAMPLES_DATA = []
+# قائمة تحتوي فقط على البيانات التي سيتم عرضها في الجدول
+SAMPLES_FOR_DISPLAY = []
+
+if os.path.isdir(EXAMPLES_DIR):
+    try:
+        for i in range(1, 4):
+            sample_name = f"sample{i}"
+            img_path = os.path.join(EXAMPLES_DIR, sample_name, "rgb.jpg")
+            json_path = os.path.join(EXAMPLES_DIR, sample_name, "measurements.json")
+            
+            with open(json_path, 'r') as f:
+                data = json.load(f)
+            
+            # إضافة البيانات الكاملة
+            FULL_EXAMPLES_DATA.append({
+                "img_path": img_path,
+                "json_path": json_path
+            })
+            
+            # إضافة بيانات العرض فقط
+            SAMPLES_FOR_DISPLAY.append([
+                img_path,
+                f"{data.get('speed', 0.0):.2f}",
+                f"{data.get('theta', 0.0):.2f}",
+                str(data.get('pos_global', 'N/A')),
+                str(data.get('target_point', 'N/A'))
+            ])
+    except FileNotFoundError:
+        print(f"تحذير: لم يتم العثور على بعض ملفات الأمثلة في مجلد '{EXAMPLES_DIR}'.")
+        FULL_EXAMPLES_DATA = []
+        SAMPLES_FOR_DISPLAY = []
+
 # ==============================================================================
 #           4. تعريف واجهة Gradio
 # ==============================================================================
 
-CUSTOM_CSS = """
-.gradio-container {max-width: 100% !important;}
-#examples-container .gradio-table {
-    table-layout: fixed !important;
-    width: 100% !important;
-}
-
-/* إخفاء العمود الثالث (عمود ملف JSON) */
-#examples-container .gradio-table th:nth-child(3),
-#examples-container .gradio-table td:nth-child(3) {
-    display: none !important;
-}
-
-/* تنسيق الأعمدة الرقمية (تبدأ من الرابع) */
-#examples-container .gradio-table th:nth-child(n+4),
-#examples-container .gradio-table td:nth-child(n+4) {
-    width: 110px !important;
-    text-align: center !important;
-    vertical-align: middle !important;
-}
-"""
+# لم نعد بحاجة إلى CSS معقد
+CUSTOM_CSS = ".gradio-container {max-width: 100% !important;}"
 
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"), css=CUSTOM_CSS) as demo:
     session_id_state = gr.State(value=None)
@@ -153,33 +171,35 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"), cs
                     api_measurements_path = gr.File(label="(JSON)", type="filepath")
                     api_run_button = gr.Button("🚀 أرسل البيانات للمعالجة", variant="primary")
                 
-                with gr.Group(elem_id="examples-container"):            
+                with gr.Group():            
                     gr.Markdown("### ✨ أمثلة جاهزة")
-                    if os.path.isdir(EXAMPLES_DIR):
-                        try:
-                            data1 = json.load(open(os.path.join(EXAMPLES_DIR, "sample1", "measurements.json")))
-                            data2 = json.load(open(os.path.join(EXAMPLES_DIR, "sample2", "measurements.json")))
-                            data3 = json.load(open(os.path.join(EXAMPLES_DIR, "sample3", "measurements.json")))
-                            
-                            with gr.Column(visible=False):
-                                example_speed = gr.Textbox()
-                                example_theta = gr.Textbox()
-                                example_pos = gr.Textbox()
-                                example_target = gr.Textbox()
-                            
-                            gr.Examples(
-                                examples=[
-                                    [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'))],
-                                    [os.path.join(EXAMPLES_DIR, "sample3", "rgb.jpg"), os.path.join(EXAMPLES_DIR, "sample3", "measurements.json"), f"{data3.get('speed', 0.0):.2f}", f"{data3.get('theta', 0.0):.2f}", str(data3.get('pos_global', 'N/A')), str(data3.get('target_point', 'N/A'))]
-                                ],
-                                inputs=[api_rgb_image_path, api_measurements_path, example_speed, example_theta, example_pos, example_target],
-                                label="اختر سيناريو اختبار"
-                            )
-                        except FileNotFoundError:
-                             gr.Markdown("لم يتم العثور على ملفات الأمثلة داخل مجلد `examples`.")
+                    if SAMPLES_FOR_DISPLAY:
+                        # --- دالة التحديد ---
+                        def on_example_select(evt: gr.SelectData):
+                            # evt.index هو رقم الصف الذي تم الضغط عليه
+                            selected_example = FULL_EXAMPLES_DATA[evt.index]
+                            return gr.update(value=selected_example["img_path"]), gr.update(value=selected_example["json_path"])
+
+                        # --- استخدام gr.Dataset ---
+                        example_dataset = gr.Dataset(
+                            components=[
+                                gr.Image(label="الصورة", type="filepath"),
+                                gr.Textbox(label="السرعة"),
+                                gr.Textbox(label="الاتجاه"),
+                                gr.Textbox(label="الموقع"),
+                                gr.Textbox(label="الهدف")
+                            ],
+                            samples=SAMPLES_FOR_DISPLAY,
+                            label="اختر سيناريو اختبار"
+                        )
+                        # --- ربط حدث التحديد بالدالة ---
+                        example_dataset.select(
+                            fn=on_example_select, 
+                            inputs=None, 
+                            outputs=[api_rgb_image_path, api_measurements_path]
+                        )
                     else:
-                        gr.Markdown("لم يتم العثور على مجلد الأمثلة (`examples`).")
+                        gr.Markdown("لم يتم العثور على أمثلة صالحة.")
            
             with gr.Column(scale=3):
                 with gr.Group():
@@ -192,4 +212,5 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"), cs
     api_run_button.click(fn=run_single_frame_via_api, inputs=[session_id_state, api_rgb_image_path, api_measurements_path], outputs=[api_output_image, api_control_json])
 
 if __name__ == "__main__":
+    # قم بإعادة تشغيل الـ Space بعد لصق هذا الكود
     demo.queue().launch(debug=True)