Update src/streamlit_app.py

src/streamlit_app.py

@@ -25,11 +25,10 @@ from huggingface_hub import hf_hub_download
 # Hugging Face model config
 MODEL_REPO = "FlameF0X/o2"
 MODEL_FILENAME = "o2_agent_latest.pth"
-MODEL_CACHE_DIR = HF_HOME  # Updated to match writable directory
+MODEL_CACHE_DIR = HF_HOME
 
 def ensure_model():
     try:
-        # Use HF_HOME for cache_dir
         model_path = hf_hub_download(
             repo_id=MODEL_REPO,
             filename=MODEL_FILENAME,
@@ -68,28 +67,38 @@ def parse_move_input(move_input, board):
     if not move_input:
         return None
     move_input = move_input.strip()
+
+    # Try UCI
     try:
         move = chess.Move.from_uci(move_input.lower())
         if move in board.legal_moves:
             return move
-    except:
-        pass
+    except Exception as e:
+        st.info(f"UCI parsing failed: {e}")
+
+    # Try SAN
     try:
         move = board.parse_san(move_input)
         if move in board.legal_moves:
             return move
-    except:
-        pass
+    except Exception as e:
+        st.info(f"SAN parsing failed: {e}")
+
+    # Try castling aliases
     castling_variations = {
-        '0-0': 'O-O', '0-0-0': 'O-O-O', 'oo': 'O-O', 'ooo': 'O-O-O', 'o-o': 'O-O', 'o-o-o': 'O-O-O',
+        '0-0': 'O-O', '0-0-0': 'O-O-O',
+        'oo': 'O-O', 'ooo': 'O-O-O',
+        'o-o': 'O-O', 'o-o-o': 'O-O-O'
     }
-    if move_input.lower() in castling_variations:
+    normalized = castling_variations.get(move_input.lower())
+    if normalized:
         try:
-            move = board.parse_san(castling_variations[move_input.lower()])
+            move = board.parse_san(normalized)
             if move in board.legal_moves:
                 return move
-        except:
-            pass
+        except Exception as e:
+            st.info(f"Castling parsing failed: {e}")
+
     return None
 
 # --- Session state ---
@@ -180,7 +189,7 @@ if not board.is_game_over() and board.turn == chess.WHITE:
             render_pgn()
             st.rerun()
         else:
-            st.warning("Invalid move.")
+            st.warning("Invalid move. Try again using SAN (e.g., `e4`, `Nf3`, `O-O`) or UCI (e.g., `e2e4`).")
     if st.button("Random Move"):
         move = random.choice(legal_moves)
         board.push(move)