File size: 2,449 Bytes
c7f95a6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
import platform
import uvicorn
from backend.device import get_device_name
from backend.models.device import DeviceInfo
from constants import APP_VERSION, DEVICE
from context import Context
from fastapi import FastAPI, Request
from fastapi_mcp import FastApiMCP
from state import get_settings
from fastapi.middleware.cors import CORSMiddleware
from models.interface_types import InterfaceType
from fastapi.staticfiles import StaticFiles
app_settings = get_settings()
app = FastAPI(
title="FastSD CPU",
description="Fast stable diffusion on CPU",
version=APP_VERSION,
license_info={
"name": "MIT",
"identifier": "MIT",
},
describe_all_responses=True,
describe_full_response_schema=True,
)
origins = ["*"]
app.add_middleware(
CORSMiddleware,
allow_origins=origins,
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
print(app_settings.settings.lcm_diffusion_setting)
context = Context(InterfaceType.API_SERVER)
app.mount("/results", StaticFiles(directory="results"), name="results")
@app.get(
"/info",
description="Get system information",
summary="Get system information",
operation_id="get_system_info",
)
async def info() -> dict:
device_info = DeviceInfo(
device_type=DEVICE,
device_name=get_device_name(),
os=platform.system(),
platform=platform.platform(),
processor=platform.processor(),
)
return device_info.model_dump()
@app.post(
"/generate",
description="Generate image from text prompt",
summary="Text to image generation",
operation_id="generate",
)
async def generate(
prompt: str,
request: Request,
) -> str:
"""
Returns URL of the generated image for text prompt
"""
app_settings.settings.lcm_diffusion_setting.prompt = prompt
images = context.generate_text_to_image(app_settings.settings)
image_names = context.save_images(
images,
app_settings.settings,
)
url = request.url_for("results", path=image_names[0])
image_url = f"The generated image available at the URL {url}"
return image_url
def start_mcp_server(port: int = 8000):
mcp = FastApiMCP(
app,
name="FastSDCPU MCP",
description="MCP server for FastSD CPU API",
base_url=f"http://localhost:{port}",
)
mcp.mount()
uvicorn.run(
app,
host="0.0.0.0",
port=port,
)
|