Upload game_engine.py

game_engine.py

@@ -498,10 +498,15 @@ class MathGame:
             if row_data['is_correct']:
                 correct_answers += 1
 
-            # Structure pour dataset
+            # Structure pour dataset avec debug OCR
             a, b, operation, correct_result = self.operations_history[i] if i < len(self.operations_history) else (0, 0, "×", 0)
             
-            ocr_info_data = get_ocr_model_info()
+            try:
+                ocr_info_data = get_ocr_model_info()
+                print(f"🔍 Debug OCR info: {ocr_info_data}")
+            except Exception as e:
+                print(f"❌ Erreur get_ocr_model_info: {e}")
+                ocr_info_data = {"model_name": "Error", "device": "Unknown"}
             
             entry = {
                 "session_id": session_id,
@@ -523,6 +528,9 @@ class MathGame:
                 "app_version": "3.0_calcul_ocr_parallel"
             }
             
+            print(f"🔍 Debug entry OCR fields: ocr_model={entry['ocr_model']}, ocr_device={entry['ocr_device']}")
+            
+            
             if row_data['dataset_image_data']:
                 entry["handwriting_image"] = row_data['dataset_image_data']["image_base64"]
                 entry["image_width"] = int(row_data['dataset_image_data']["compressed_size"][0])
@@ -653,12 +661,26 @@ def export_to_clean_dataset(session_data: list[dict], dataset_name: str = DATASE
         print(f"\n🚀 === EXPORT VERS DATASET CALCUL OCR ===")
         print(f"📊 Dataset: {dataset_name}")
         
-        # Filtrer les entrées avec images
+        # Filtrer les entrées avec images et ajouter les infos OCR globalement
         clean_entries = []
         
+        # Récupérer une seule fois les infos OCR pour toute la session
+        try:
+            global_ocr_info = get_ocr_model_info()
+            print(f"🔍 Infos OCR globales: {global_ocr_info}")
+        except Exception as e:
+            print(f"❌ Erreur infos OCR globales: {e}")
+            global_ocr_info = {"model_name": "Unknown", "device": "Unknown"}
+        
         for entry in session_data:
             if entry.get('has_image', False):
-                clean_entries.append(entry)
+                # Ajouter explicitement les champs OCR manquants
+                entry_with_ocr = entry.copy()
+                entry_with_ocr["ocr_model"] = global_ocr_info.get("model_name", "Unknown")
+                entry_with_ocr["ocr_device"] = global_ocr_info.get("device", "Unknown")
+                
+                print(f"🔍 Entry avec OCR: ocr_model={entry_with_ocr['ocr_model']}, ocr_device={entry_with_ocr['ocr_device']}")
+                clean_entries.append(entry_with_ocr)
         
         print(f"✅ {len(clean_entries)} entrées avec images converties")