Update app.py

app.py

@@ -6,9 +6,10 @@ from fastapi_mcp import FastApiMCP
 
 app = FastAPI()
 # app = FastAPI(docs_url="/docs")
-MTMODELS = {'enro': 'BlackKakapo/opus-mt-en-ro',
-            'roen': 'BlackKakapo/opus-mt-ro-en',
-            'gemma': 'Gargaz/gemma-2b-romanian-better'}
+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):
@@ -28,7 +29,7 @@ def index(request: Request):
                 <p><a href="{host_url}/openapi.json" target="_blank">openapi.json</a></p>
                 <p><a href="{host_url}/mcp" target="_blank">MCP</a></p>
                 <p>MCP configuration: {mcp_config}</a></p>
-                <p>MTMODELS: {list(MTMODELS.values())}"</p>
+                <p>MODELS: {list(MODELS.values())}"</p>
             </body>
         </html>
         '''
@@ -37,20 +38,20 @@ def index(request: Request):
 # @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'], 'mtmodels': MTMODELS}
+#     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):
+def paraphrase(text: str, model: str  = MODELS['paraphrase']):
     resultValue, exception = Paraphrase.paraphraseParaphraseMethod(text, model)
-    return {"input": text, "translation": resultValue, "exception": exception}
+    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] = MTMODELS['enro']):
+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, "translation": translation, "model": model}
+    return {"input": text, "result": translation, "model": model}
 
 # Create an MCP server based on this app
 mcp = FastApiMCP(