Rafs-an09002 commited on
Commit
de3cc06
·
verified ·
1 Parent(s): 5527d30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -30
app.py CHANGED
@@ -36,36 +36,13 @@ class HealthResponse(BaseModel):
36
  @app.on_event("startup")
37
  async def startup_event():
38
  global engine
39
- logger.info("🚀 Starting Nexus-Core...")
40
-
41
- # FIXED: Check both possible paths
42
- possible_paths = [
43
- "/app/app/models/nexus-core.onnx", # When uploaded to app/models/
44
- "/app/models/nexus-core.onnx" # When uploaded to models/
45
- ]
46
-
47
- model_path = None
48
- for path in possible_paths:
49
- if os.path.exists(path):
50
- model_path = path
51
- logger.info(f"✅ Found model at: {path}")
52
- break
53
-
54
- if not model_path:
55
- logger.error("❌ Model not found in any expected location")
56
- logger.error(f"Checked paths: {possible_paths}")
57
- # List all .onnx files
58
- for root, dirs, files in os.walk("/app"):
59
- for file in files:
60
- if file.endswith('.onnx'):
61
- logger.error(f"Found .onnx at: {os.path.join(root, file)}")
62
- raise FileNotFoundError("Model not found")
63
-
64
  try:
65
  engine = NexusCoreEngine(model_path=model_path, num_threads=2)
66
- logger.info("✅ Engine loaded!")
67
  except Exception as e:
68
- logger.error(f"❌ Failed: {e}", exc_info=True)
69
  raise
70
 
71
  @app.get("/health", response_model=HealthResponse)
@@ -74,15 +51,15 @@ async def health_check():
74
 
75
  @app.post("/get-move", response_model=MoveResponse)
76
  async def get_move(request: MoveRequest):
77
- if not engine: raise HTTPException(503, "Not loaded")
78
  if not engine.validate_fen(request.fen): raise HTTPException(400, "Invalid FEN")
79
  start = time.time()
80
  try:
81
  result = engine.get_best_move(request.fen, request.depth, request.time_limit)
82
- logger.info(f" {result['best_move']} | {result['evaluation']:+.2f} | {result['time_taken']}ms")
83
  return MoveResponse(**result)
84
  except Exception as e:
85
- logger.error(f"Error: {e}", exc_info=True)
86
  raise HTTPException(500, str(e))
87
 
88
  @app.get("/")
 
36
  @app.on_event("startup")
37
  async def startup_event():
38
  global engine
39
+ logger.info("🚀 Starting Nexus-Core API...")
40
+ model_path = "/app/models/nexus-core.onnx"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
41
  try:
42
  engine = NexusCoreEngine(model_path=model_path, num_threads=2)
43
+ logger.info("✅ Nexus-Core engine loaded")
44
  except Exception as e:
45
+ logger.error(f"❌ Failed to load engine: {e}")
46
  raise
47
 
48
  @app.get("/health", response_model=HealthResponse)
 
51
 
52
  @app.post("/get-move", response_model=MoveResponse)
53
  async def get_move(request: MoveRequest):
54
+ if not engine: raise HTTPException(503, "Engine not loaded")
55
  if not engine.validate_fen(request.fen): raise HTTPException(400, "Invalid FEN")
56
  start = time.time()
57
  try:
58
  result = engine.get_best_move(request.fen, request.depth, request.time_limit)
59
+ logger.info(f"Move: {result['best_move']} | Eval: {result['evaluation']:+.2f} | Time: {result['time_taken']}ms")
60
  return MoveResponse(**result)
61
  except Exception as e:
62
+ logger.error(f"Error: {e}")
63
  raise HTTPException(500, str(e))
64
 
65
  @app.get("/")