Update main.py

main.py

@@ -261,8 +261,13 @@ async def _engine_call(engine, coro, timeout_sec: float):
 
 
 # ─── Background Memory Cleanup Task ───────────────────────────────────────────
-_RAM_CLEANUP_THRESHOLD_MB = float(os.environ.get("RAM_CLEANUP_THRESHOLD_MB", "400"))
-_RAM_CLEANUP_INTERVAL_SEC = int(os.environ.get("RAM_CLEANUP_INTERVAL_SEC", "300"))
+_RAM_CLEANUP_THRESHOLD_MB = float(os.environ.get("RAM_CLEANUP_THRESHOLD_MB", "1024"))
+# Global System Limit: 18GB (All analysis sessions freed)
+_RAM_PANIC_THRESHOLD_MB = 18000.0
+# Individual Session Limit: 4GB
+_RAM_SESSION_LIMIT_MB = 4000.0
+# How often to run the cleanup in seconds (5s)
+_RAM_CLEANUP_INTERVAL_SEC = 5
 
 async def memory_cleanup_task():
     """
@@ -276,7 +281,14 @@ async def memory_cleanup_task():
             process = psutil.Process(os.getpid())
             mem_mb = process.memory_info().rss / 1024 / 1024
 
-            if mem_mb > _RAM_CLEANUP_THRESHOLD_MB:
+            # PANIC CASE: Over 18GB - total server reset
+            if mem_mb > _RAM_PANIC_THRESHOLD_MB:
+                print(f"[PANIC] RAM at {mem_mb:.1f}MB! Exceeded 18GB System limit. Resetting everything...")
+                await shutdown_engine_async()
+                force_memory_release()
+            
+            # NORMAL CLEANUP
+            elif mem_mb > _RAM_CLEANUP_THRESHOLD_MB:
                 print(f"[CLEANUP] RAM at {mem_mb:.1f}MB (threshold {_RAM_CLEANUP_THRESHOLD_MB}MB) — clearing engine hash")
                 engine = _GLOBAL_DEEPCASTLE_ENGINE
                 if engine is not None:
@@ -761,7 +773,19 @@ async def analyze_game(request: AnalyzeRequest):
         player_moves_count = 0
         current_score, _ = get_normalized_score(infos_before[0])
 
+        process = psutil.Process()
+        mem_start = process.memory_info().rss / 1024 / 1024
+        
         for i, san_move in enumerate(request.moves):
+            # Check individual session growth (4GB limit)
+            mem_now = process.memory_info().rss / 1024 / 1024
+            if (mem_now - mem_start) > _RAM_SESSION_LIMIT_MB:
+                print(f"[ANALYSIS-RESET] Individual session exceeds 4GB growth. Clearing RAM mid-analysis.")
+                async with _ENGINE_IO_LOCK:
+                    await _clear_engine_hash(engine)
+                force_memory_release()
+                mem_start = process.memory_info().rss / 1024 / 1024
+
             is_white_turn = board.turn == chess.WHITE
             is_player_turn = is_white_turn if player_is_white else not is_white_turn