Update app.py

app.py

@@ -3,7 +3,8 @@ import src.Paraphrase as Paraphrase
 import src.Translate as Translate
 from typing import Optional
 
-app = FastAPI(docs_url="/")
+app = FastAPI()
+# app = FastAPI(docs_url="/")
 MTMODELS = {'enro': 'BlackKakapo/opus-mt-en-ro',
             'roen': 'BlackKakapo/opus-mt-ro-en',
             'gemma': 'Gargaz/gemma-2b-romanian-better'}
@@ -21,4 +22,11 @@ def paraphrase(text: str, model: str):
 def translate(text: str, model: Optional[str] = MTMODELS['gemma']):
     # resultValue, exception = Translate.paraphraseTranslateMethod(text, model)
     resultValue = Translate.gemma_direct(text, model)
-    return {"input": text, "result": resultValue, "model": model}
+    return {"input": text, "result": resultValue, "model": model}
+
+from fastapi_mcp import FastApiMCP
+# Create an MCP server based on this app
+mcp = FastApiMCP(app)
+
+# Mount the MCP server directly to your app
+mcp.mount()