from fastapi import FastAPI, Request import src.Paraphrase as Paraphrase import src.Translate as Translate from typing import Optional from fastapi_mcp import FastApiMCP app = FastAPI() # app = FastAPI(docs_url="/docs") MODELS = {'enro': 'BlackKakapo/opus-mt-en-ro', 'roen': 'BlackKakapo/opus-mt-ro-en', 'gemma': 'Gargaz/gemma-2b-romanian-better', 'paraphrase': 'tuner007/pegasus_paraphrase'} @app.get("/") def index(request: Request): from fastapi.responses import HTMLResponse host_url = "https://" + request.url.netloc mcp_config = '''{"mcpServers": {"fastapi-mcp": {"url": "https://tiberiucristianleon-fastapimt.hf.space/mcp"}}}''' html_content = f''' FastAPI with MCP

FastAPI URLS

Host URL: {host_url}

DOCS

REDOC

openapi.json

MCP

MCP configuration: {mcp_config}

MODELS: {list(MODELS.values())}"

''' return HTMLResponse(content=html_content) # @app.get("/") # async def get_host_url(request: Request): # host_url = request.url.scheme + "s://" + request.url.netloc # return {"host_url": host_url, 'endpoints': ['/paraphrase', '/translate', f'{host_url}/docs', f'{host_url}/redoc', f'{host_url}/openapi.json'], 'models': MODELS} @app.get("/paraphrase", operation_id="get_paraphrase", description="Paraphrase text", tags=["paraphrase"], summary="Paraphrase text") def paraphrase(text: str, model: str = MODELS['paraphrase']): resultValue, exception = Paraphrase.paraphraseParaphraseMethod(text, model) return {"input": text, "result": resultValue, "exception": exception} @app.get("/translate", operation_id="get_translate", description="Translate text", tags=["translate"], summary="Translate text") def translate(text: str, model: Optional[str] = MODELS['enro']): if 'BlackKakapo' in model: translation = Translate.paraphraseTranslateMethod(text, model) else: translation: str = Translate.gemma_direct(text, model) return {"input": text, "result": translation, "model": model} # Create an MCP server based on this app mcp = FastApiMCP( app, name="Translate and paraphrase FASTAPI MCP", description="MCP server to translate and paraphrase text", describe_all_responses=True, describe_full_response_schema=True, include_operations=["get_translate", "get_paraphrase"], include_tags=["paraphrase", "translate"] ) # Mount the MCP server directly to the FASTAPI app mcp.mount()