Update app.py

app.py

@@ -1,45 +1,51 @@
 import os
+import sys
 import zipfile
-from fastapi import FastAPI
-from fastapi.responses import JSONResponse
 import uvicorn
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import JSONResponse
 
-# 1. Décompression du fichier zip si nécessaire
-ZIP_PATH = "MistralHackathonMCP.zip"   # adapte au vrai nom de ton zip
-EXTRACT_DIR = "app"
-
+# --- Décompression du zip ---
+ZIP_PATH = "MistralHackathonMCP.zip"
+EXTRACT_DIR = "app/MistralHackathonMCP"
 if os.path.exists(ZIP_PATH) and not os.path.exists(EXTRACT_DIR):
     with zipfile.ZipFile(ZIP_PATH, 'r') as zip_ref:
-        zip_ref.extractall(EXTRACT_DIR)
-    print(f"✅ Décompressé {ZIP_PATH} dans {EXTRACT_DIR}")
+        zip_ref.extractall("app")
+    print(f"✅ Décompressé {ZIP_PATH} dans app/")
+
+# --- Ajouter MCP au PYTHONPATH ---
+sys.path.append("app/MistralHackathonMCP")
+
+# --- Importer FastMCP instance ---
+from main import get_mcp_instance
+mcp = get_mcp_instance()
 
-# 2. Création de l'application FastAPI
+# --- FastAPI ---
 app = FastAPI()
 
-# ✅ Route d’accueil (évite le 404 sur "/")
 @app.get("/")
 def root():
     return {
-        "status": "MCP server actif 🚀",
+        "status": "MCP Atlas actif 🚀",
         "endpoints": ["/mcp/list_tools", "/mcp/call_tool"]
     }
 
-# Exemple d'endpoint MCP "list_tools"
 @app.get("/mcp/list_tools")
 def list_tools():
-    return JSONResponse({
-        "tools": [
-            {"name": "hello_world", "description": "Renvoie un message de test"}
-        ]
-    })
+    try:
+        tools = mcp.list_tools()  # FastMCP gère les outils
+        return JSONResponse(tools)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Erreur list_tools(): {e}")
 
-# Exemple d'endpoint MCP "call_tool"
 @app.get("/mcp/call_tool")
 def call_tool(name: str):
-    if name == "hello_world":
-        return {"result": "Bonjour depuis Hugging Face MCP server 🚀"}
-    return {"error": "Outil inconnu"}
+    try:
+        result = mcp.call_tool(name)  # Appel FastMCP
+        return JSONResponse(result)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Erreur call_tool(): {e}")
 
-# 3. Point d’entrée Hugging Face
+# --- Point d'entrée ---
 if __name__ == "__main__":
     uvicorn.run("app:app", host="0.0.0.0", port=7860)