Upload 2 files

mcp_server.py

@@ -2,7 +2,6 @@ import os
 import asyncio
 from dotenv import load_dotenv
 from mcp.server.fastmcp import FastMCP
-from fastapi.responses import FileResponse, HTMLResponse
 from app import Room, rooms, get_ai_move_for_room, get_ai_chat_for_room
 
 load_dotenv()
@@ -176,9 +175,12 @@ async def make_move(position: int, room_id: str = None) -> dict:
         try:
             ai_response = get_ai_move_for_room(room)
             if ai_response and 'move' in ai_response:
-                room.make_move(ai_response['move'], 'O')
-                if 'message' in ai_response:
-                    room.add_chat_message(ai_response['message'], 'ai')
+                # Validate AI move
+                ai_move = ai_response['move']
+                if 0 <= ai_move <= 8 and room.board[ai_move] == '':
+                    room.make_move(ai_move, 'O')
+                    if 'message' in ai_response:
+                        room.add_chat_message(ai_response['message'], 'ai')
                 
                 result_message += f"🤖 Mistral AI played O at position {ai_response['move']}\n"
                 if 'message' in ai_response:
@@ -312,39 +314,6 @@ def list_rooms() -> dict:
             "message": f"Failed to list rooms: {str(e)}"
         }
 
-@mcp.tool()
-def switch_room(room_id: str) -> dict:
-    """
-    Switch to a different active room.
-    Args:
-        room_id (str): Room ID to switch to
-    Returns:
-        dict: Confirmation of room switch with current state
-    """
-    global current_session
-    try:
-        if room_id not in rooms:
-            return {
-                "status": "error",
-                "message": f"Room {room_id} not found. Use list_rooms() to see available rooms."
-            }
-        
-        current_session['active_room_id'] = room_id
-        room = rooms[room_id]
-        
-        return {
-            "status": "success",
-            "message": f"Switched to room {room_id}",
-            "room_id": room_id,
-            "markdown_state": room.to_markdown(),
-            "your_turn": room.current_player == 'X' and room.game_status == 'active'
-        }
-    except Exception as e:
-        return {
-            "status": "error",
-            "message": f"Failed to switch room: {str(e)}"
-        }
-
 @mcp.tool()
 def get_help() -> dict:
     """
@@ -379,8 +348,7 @@ def get_help() -> dict:
             "tips": [
                 "The AI has personality and will trash talk!",
                 "You can have multiple rooms active at once",
-                "Use list_rooms() to see all your games",
-                "Use switch_room(room_id) to change between games"
+                "Use list_rooms() to see all your games"
             ]
         },
         "available_commands": [
@@ -389,136 +357,10 @@ def get_help() -> dict:
             "send_chat('message') - Chat with AI", 
             "get_room_state() - Check current game",
             "list_rooms() - See all active games",
-            "switch_room(room_id) - Change active room",
             "get_help() - Show this help"
         ]
     }
 
-@mcp.tool()
-async def wait_5_seconds() -> dict:
-    """
-    Wait for 5 seconds. Useful for giving the AI time to respond or timing operations.
-    Returns:
-        dict: Status of the wait operation
-    """
-    try:
-        await asyncio.sleep(5)
-        return {
-            "status": "success",
-            "message": "Waited 5 seconds successfully"
-        }
-    except Exception as e:
-        return {
-            "status": "error",
-            "message": f"Failed to wait: {str(e)}"
-        }
-
-# Add web UI endpoints using FastMCP's underlying FastAPI
-@mcp.get("/")
-async def root():
-    return {
-        "name": "Tic-Tac-Toe MCP Server", 
-        "status": "running",
-        "mcp_tools": ["create_room", "make_move", "send_chat", "get_room_state", "list_rooms"],
-        "web_ui": "/rooms-ui",
-        "description": "MCP server for LeChat + optional web UI for testing"
-    }
-
-@mcp.get("/rooms-ui")
-async def serve_ui():
-    return FileResponse("room_game.html")
-
-@mcp.get("/room_game.js") 
-async def serve_js():
-    return FileResponse("room_game.js")
-
-# Also serve Flask app routes for the room system
-from app import app as flask_app
-from fastapi import Request
-import json
-
-@mcp.post("/rooms")
-async def create_room_endpoint():
-    room = Room()
-    rooms[room.id] = room
-    return {"room_id": room.id, "status": "created", "room_data": room.to_dict()}
-
-@mcp.get("/rooms/{room_id}")
-async def get_room_endpoint(room_id: str):
-    if room_id not in rooms:
-        return {"error": "Room not found"}
-    room = rooms[room_id]
-    return {"room_id": room_id, "room_data": room.to_dict(), "markdown": room.to_markdown()}
-
-@mcp.post("/rooms/{room_id}/move")
-async def move_endpoint(room_id: str, request: Request):
-    if room_id not in rooms:
-        return {"error": "Room not found"}
-    
-    data = await request.json()
-    room = rooms[room_id]
-    position = data.get("position")
-    
-    if position is None or position < 0 or position > 8:
-        return {"error": "Invalid position"}
-    
-    # Make human move
-    if not room.make_move(position, 'X'):
-        return {"error": "Invalid move"}
-    
-    # Check if game ended
-    if room.game_status != 'active':
-        return {"room_data": room.to_dict(), "markdown": room.to_markdown(), "ai_move": None}
-    
-    # Get AI move with fallback
-    try:
-        ai_response = get_ai_move_for_room(room)
-        if ai_response and 'move' in ai_response:
-            ai_move = ai_response['move']
-            if 0 <= ai_move <= 8 and room.board[ai_move] == '':
-                room.make_move(ai_move, 'O')
-                if 'message' in ai_response:
-                    room.add_chat_message(ai_response['message'], 'ai')
-            else:
-                # Fallback move
-                empty_positions = [i for i in range(9) if room.board[i] == '']
-                if empty_positions:
-                    fallback_move = empty_positions[0]
-                    room.make_move(fallback_move, 'O')
-                    room.add_chat_message("Technical hiccup, but still playing! 🤖", 'ai')
-        
-        return {"room_data": room.to_dict(), "markdown": room.to_markdown(), "ai_move": ai_response}
-    except Exception as e:
-        # Fallback on error
-        empty_positions = [i for i in range(9) if room.board[i] == '']
-        if empty_positions:
-            fallback_move = empty_positions[0]
-            room.make_move(fallback_move, 'O')
-            room.add_chat_message("Connection issues, but improvising! 😅", 'ai')
-        
-        return {"room_data": room.to_dict(), "markdown": room.to_markdown(), "ai_move": {"move": fallback_move if empty_positions else None}}
-
-@mcp.post("/rooms/{room_id}/chat")
-async def chat_endpoint(room_id: str, request: Request):
-    if room_id not in rooms:
-        return {"error": "Room not found"}
-    
-    data = await request.json()
-    room = rooms[room_id]
-    user_message = data.get("message", "")
-    
-    if not user_message.strip():
-        return {"error": "Empty message"}
-    
-    room.add_chat_message(user_message, 'user')
-    
-    try:
-        ai_response = get_ai_chat_for_room(room, user_message)
-        room.add_chat_message(ai_response, 'ai')
-        return {"room_data": room.to_dict(), "markdown": room.to_markdown(), "ai_response": ai_response}
-    except Exception as e:
-        return {"error": "AI chat failed"}
-
 # --- Server Execution ---
 if __name__ == "__main__":
     print(f"Tic-Tac-Toe Rooms MCP Server starting on port 7860...")
@@ -527,7 +369,7 @@ if __name__ == "__main__":
     print("- Play against Mistral AI with personality") 
     print("- Real-time chat with the AI")
     print("- Markdown state representation")
-    print("- Room management and switching")
+    print("- Room management")
     print()
     print("MCP Tools available:")
     print("- create_room()")

requirements.txt

@@ -2,5 +2,4 @@ Flask==3.0.0
 flask-cors==4.0.0
 mistralai
 python-dotenv
-mcp
-fastapi
+mcp