Run FastMCP HTTP on 7860; optionalize ESM imports; expose health

Dockerfile

@@ -9,4 +9,7 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY --chown=user . /app
 EXPOSE 7860
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Run FastMCP HTTP server on port 7860
+ENV MCP_TRANSPORT=http
+ENV MCP_PORT=7860
+CMD ["python", "-m", "esm.mcp_output.start_mcp"]

app.py

@@ -1,17 +1,15 @@
-from fastapi import FastAPI, WebSocket
+from fastapi import FastAPI
+import os
 
 app = FastAPI()
 
 
 @app.get("/")
 async def root():
-    return {"status": "ok", "service": "Code2MCP-esm"}
-
-
-@app.websocket("/ws")
-async def websocket_endpoint(ws: WebSocket):
-    await ws.accept()
-    await ws.send_text("WebSocket is up. Replace with your MCP/ESM handler.")
-    await ws.close()
+    return {
+        "status": "ok",
+        "service": "Code2MCP-esm",
+        "transport": os.environ.get("MCP_TRANSPORT", "stdio"),
+    }
 
 

esm/mcp_output/mcp_plugin/mcp_service.py

@@ -5,10 +5,17 @@ source_path = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__fil
 sys.path.insert(0, source_path)
 
 from fastmcp import FastMCP
-from esm import pretrained, data, inverse_folding, model
-# from examples.lm_design.lm_design import lm_design
-# from examples.variant_prediction.predict import predict
-# from scripts import extract, fold
+try:
+    from esm import pretrained, data, inverse_folding, model  # type: ignore
+    # from examples.lm_design.lm_design import lm_design
+    # from examples.variant_prediction.predict import predict
+    # from scripts import extract, fold
+except Exception as import_error:  # pragma: no cover
+    pretrained = None  # type: ignore
+    data = None  # type: ignore
+    inverse_folding = None  # type: ignore
+    model = None  # type: ignore
+    print(f"[warn] Optional ESM modules unavailable: {import_error}")
 
 mcp = FastMCP("esm_service")