Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import sys
 import zipfile
 import uvicorn
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Request
 from fastapi.responses import JSONResponse
 
 # --- 1️⃣ Décompression du zip MCP ---
@@ -17,15 +17,14 @@ if os.path.exists(ZIP_PATH) and not os.path.exists(EXTRACT_DIR):
 # --- 2️⃣ Ajouter le dossier MCP au PYTHONPATH ---
 sys.path.append(EXTRACT_DIR)
 
-# Patch rapide pour corriger les f-strings avec strftime
-main_path = "app/MistralHackathonMCP/main.py"
+# --- Patch rapide pour corriger les f-strings avec strftime ---
+main_path = os.path.join(EXTRACT_DIR, "main.py")
 if os.path.exists(main_path):
     with open(main_path, "r", encoding="utf-8") as f:
         lines = f.readlines()
 
     fixed_lines = []
     for line in lines:
-        # Remplace f"[{current_time.strftime("%Y-%m-%d %H:%M:%S")}] ..." par f"[{current_time.strftime('%Y-%m-%d %H:%M:%S')}] ..."
         fixed_lines.append(line.replace('strftime("%Y-%m-%d %H:%M:%S")', "strftime('%Y-%m-%d %H:%M:%S')"))
 
     with open(main_path, "w", encoding="utf-8") as f:
@@ -45,6 +44,7 @@ except Exception as e:
 # --- 4️⃣ Créer l'application FastAPI ---
 app = FastAPI(title="Atlas-MCP Server")
 
+# --- Route GET racine ---
 @app.get("/")
 def root():
     return {
@@ -52,19 +52,18 @@ def root():
         "endpoints": ["/mcp/list_tools", "/mcp/call_tool", "/health"]
     }
 
-from fastapi import Request
-
+# --- Route POST racine (pour front HF) ---
 @app.post("/")
 async def root_post(request: Request):
     return {"status": "MCP Atlas actif 🚀 (POST OK)"}
 
 # --- 5️⃣ Route pour lister les outils MCP ---
 @app.get("/mcp/list_tools")
-def list_tools():
+async def list_tools():
     if not mcp:
         raise HTTPException(status_code=500, detail="MCP non initialisé")
     try:
-        tools = mcp.list_tools()  # FastMCP gère les outils enregistrés
+        tools = await mcp.list_tools()  # await la coroutine
         return JSONResponse(tools)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Erreur list_tools(): {e}")
@@ -75,18 +74,22 @@ def call_tool(name: str):
     if not mcp:
         raise HTTPException(status_code=500, detail="MCP non initialisé")
     try:
-        # FastMCP call_tool() peut prendre args/kwargs si nécessaire
-        result = mcp.call_tool(name)
+        # passer arguments vides si outil n'en requiert pas
+        result = mcp.call_tool(name, arguments=[])
         return JSONResponse(result)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Erreur call_tool(): {e}")
 
 # --- 7️⃣ Route healthcheck ---
 @app.get("/health")
-def health():
-    if mcp:
-        return {"status": "ok", "mcp_tools_count": len(mcp.list_tools())}
-    return {"status": "error", "detail": "MCP non initialisé"}
+async def health():
+    if not mcp:
+        return {"status": "error", "detail": "MCP non initialisé"}
+    try:
+        tools = await mcp.list_tools()
+        return {"status": "ok", "mcp_tools_count": len(tools)}
+    except Exception as e:
+        return {"status": "error", "detail": f"Healthcheck failed: {e}"}
 
 # --- 8️⃣ Point d'entrée pour lancer l'API ---
 if __name__ == "__main__":