update app.py

app.py

@@ -2,52 +2,40 @@ import gradio as gr
 import chess
 import numpy as np
 import requests
-import io
+from huggingface_hub import hf_hub_download
+import os
 
+# HF_TOKEN from Space Secrets (Settings → Secrets)
 REPO = "typical-cyber/chess-model"
+TOKEN = os.environ.get("HF_TOKEN")  # Add to Secrets!
 
-def download_file(path):
-    url = f"https://huggingface.co/{REPO}/resolve/main/{path}"
-    r = requests.get(url)
-    if path.endswith('.py'):
-        return io.StringIO(r.text)
-    return io.BytesIO(r.content)
-
-# Load YOUR model code
-exec(download_file('chess_mcvs.py').read())
+# Secure download (private repo)
+def load_model_secure():
+    model_code = hf_hub_download(
+        repo_id=REPO, filename="chess_mcvs.py",
+        token=TOKEN, repo_type="model"
+    )
+    exec(open(model_code).read())
+    
+    db_path = hf_hub_download(
+        repo_id=REPO, filename="user_models/user136/chess/data/chesszonedb.npz",
+        token=TOKEN, repo_type="model"
+    )
+    return np.load(db_path, allow_pickle=True)
 
-# Load YOUR DB (user_136 path)
-db_data = np.load(download_file('user_models/user136/chess/data/chess_zone_db.npz'), allow_pickle=True)
+# Load once
+zonedb_data = load_model_secure()
+exec(open('chess_mcvs.py').read())  # Fallback if download fails
 zonedb = HilbertOrderedZoneDatabase()
-zonedb.winningmatrices = list(db_data.get('winning', []))
-zonedb.losingmatrices = list(db_data.get('losing', []))
-zonedb.drawmatrices = list(db_data.get('draw', []))
+zonedb.winningmatrices = list(zonedb_data.get('winning', []))
 
-def get_move(fen, player, style="mcvs"):
+def get_move(fen, player):
     board = chess.Board(fen)
     game = Chess()
     game.board = board
-    
-    if style == "mcvs":
-        searcher = MCVSSearcher(
-            policynet=None, valuenet=None,
-            zonedb=zonedb, device='cpu',
-            lambdazone=1.0, kzone=5, usenets=False
-        )
-        visits, _ = searcher.searchwithtimebudget(game, timebudget=1.0)
-        best_move = max(visits, key=visits.get)
-        return best_move.uci()
-    
-    # Fallback
-    return random.choice(list(board.legal_moves)).uci()
+    searcher = MCVSSearcher(None, None, zonedb)
+    visits, _ = searcher.searchwithtimebudget(game, 1.0)
+    return max(visits, key=visits.get).uci()
 
-gr.Interface(
-    fn=get_move,
-    inputs=[
-        gr.Textbox(label="FEN", value="rnbqkbnr/pppppppp/8/8/4P3/8/PPPP1PPP/RNBQKBNR b KQkq e3 0 1"),
-        gr.Dropdown(["w","b"], label="Player", value="b"),
-        gr.Dropdown(["mcvs","random"], label="Style")
-    ],
-    outputs=gr.Textbox(label="UCI Move"),
-    title="typical-cyber/chess-model LIVE API"
-).launch()
+gr.Interface(fn=get_move, inputs=gr.Textbox("FEN"), 
+             outputs="text", title="Protected ChessMCVS").launch()