Upload game_engine.py

game_engine.py

@@ -25,10 +25,14 @@ from typing import Dict, Tuple, Optional
 ocr_module = None
 ocr_info = {"model_name": "Unknown", "device": "Unknown"}
 
+# Forcer GPU en mode HF Spaces avec hardware GPU
 try:
-    # Test GPU : torch + CUDA disponible
-    import torch
-    if torch.cuda.is_available():
+    # Détecter si on est sur HF Spaces avec GPU allocé
+    import os
+    is_hf_gpu_space = os.getenv("SPACE_ID") and os.getenv("SPACE_HARDWARE") == "gpu"
+    
+    if is_hf_gpu_space:
+        print("🎯 HF Spaces GPU détecté - Force mode TrOCR")
         from image_processing_gpu import (
             recognize_number_fast_with_image, 
             create_thumbnail_fast, 
@@ -38,9 +42,39 @@ try:
             get_ocr_model_info
         )
         ocr_module = "gpu"
-        print("✅ Game Engine: Mode GPU - TrOCR activé")
+        print("✅ Game Engine: Mode GPU - TrOCR forcé sur HF Spaces")
     else:
-        # Torch installé mais pas de GPU → CPU
+        raise ImportError("Pas en mode HF GPU")
+        
+except (ImportError, Exception) as e:
+    print(f"⚠️ Mode GPU non disponible: {e}")
+    # Fallback CPU normal
+    try:
+        import torch
+        if torch.cuda.is_available():
+            from image_processing_gpu import (
+                recognize_number_fast_with_image, 
+                create_thumbnail_fast, 
+                create_white_canvas,
+                cleanup_memory,
+                log_memory_usage,
+                get_ocr_model_info
+            )
+            ocr_module = "gpu"
+            print("✅ Game Engine: Mode GPU - TrOCR activé")
+        else:
+            from image_processing_cpu import (
+                recognize_number_fast_with_image, 
+                create_thumbnail_fast, 
+                create_white_canvas,
+                cleanup_memory,
+                log_memory_usage,
+                get_ocr_model_info
+            )
+            ocr_module = "cpu"
+            print("✅ Game Engine: Mode CPU - EasyOCR activé")
+            
+    except ImportError:
         from image_processing_cpu import (
             recognize_number_fast_with_image, 
             create_thumbnail_fast, 
@@ -51,19 +85,6 @@ try:
         )
         ocr_module = "cpu"
         print("✅ Game Engine: Mode CPU - EasyOCR activé")
-        
-except ImportError:
-    # Torch pas installé → CPU obligatoire
-    from image_processing_cpu import (
-        recognize_number_fast_with_image, 
-        create_thumbnail_fast, 
-        create_white_canvas,
-        cleanup_memory,
-        log_memory_usage,
-        get_ocr_model_info
-    )
-    ocr_module = "cpu"
-    print("✅ Game Engine: Mode CPU - EasyOCR activé")
 
 # Récupérer les infos du modèle sélectionné
 try: