Update app.py

app.py

@@ -1,9 +1,97 @@
+import os
+import json
+import requests
 from fastapi import FastAPI
+from mcp.server.lowlevel import Server, NotificationOptions
+from mcp.server.sse import SSEServerTransport
+from mcp import types as mcp_types
 import uvicorn
-import os
 
 app = FastAPI()
 
+# Load environment variables
+AIRTABLE_API_TOKEN = os.getenv("AIRTABLE_API_TOKEN")
+AIRTABLE_BASE_ID = os.getenv("AIRTABLE_BASE_ID")
+TABLE_ID = "tblQECi5f7m4y2NEV"
+AIRTABLE_API_URL = f"https://api.airtable.com/v0/{AIRTABLE_BASE_ID}/{TABLE_ID}"
+
+# Helper function for Airtable API requests
+def airtable_request(method, endpoint="", data=None):
+    headers = {
+        "Authorization": f"Bearer {AIRTABLE_API_TOKEN}",
+        "Content-Type": "application/json"
+    }
+    url = f"{AIRTABLE_API_URL}/{endpoint}" if endpoint else AIRTABLE_API_URL
+    response = requests.request(method, url, headers=headers, json=data)
+    response.raise_for_status()
+    return response.json()
+
+# Tool to list records
+async def list_records_tool(request: mcp_types.CallToolRequest):
+    try:
+        records = airtable_request("GET")
+        return {
+            "success": True,
+            "result": json.dumps(records)
+        }
+    except Exception as e:
+        return {
+            "success": False,
+            "error": str(e)
+        }
+
+# Tool to create a record
+async def create_record_tool(request: mcp_types.CallToolRequest):
+    try:
+        record_data = request.input.get("record_data", {})
+        data = {"records": [{"fields": record_data}]}
+        response = airtable_request("POST", data=data)
+        return {
+            "success": True,
+            "result": json.dumps(response)
+        }
+    except Exception as e:
+        return {
+            "success": False,
+            "error": str(e)
+        }
+
+# MCP Server setup
+async def start_mcp_server():
+    tools = [
+        mcp_types.ToolDefinition(
+            name="list_airtable_records",
+            description="Lists all records in the specified Airtable table",
+            input_schema={},
+            output_schema={"type": "string"}
+        ),
+        mcp_types.ToolDefinition(
+            name="create_airtable_record",
+            description="Creates a new record in the specified Airtable table",
+            input_schema={"record_data": {"type": "object"}},
+            output_schema={"type": "string"}
+        )
+    ]
+
+    server = Server(
+        tools=tools,
+        tool_handlers={
+            "list_airtable_records": list_records_tool,
+            "create_airtable_record": create_record_tool
+        },
+        notification_options=NotificationOptions()
+    )
+
+    transport = SSEServerTransport(path="/airtable/mcp")
+    await server.run(transport)
+
+# Start the MCP server when the FastAPI app starts
+@app.on_event("startup")
+async def startup_event():
+    import asyncio
+    asyncio.create_task(start_mcp_server())
+
+# Health check endpoint
 @app.get("/health")
 async def health_check():
     return {"status": "healthy"}