Upload 2 files

mcp_server.py

@@ -2,6 +2,7 @@ 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()
@@ -412,8 +413,111 @@ async def wait_5_seconds() -> dict:
             "message": f"Failed to wait: {str(e)}"
         }
 
-# Note: Web UI served separately by Flask app
-# This MCP server focuses on MCP tools for LeChat integration
+# 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__":

requirements.txt

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