Update app.py

app.py

@@ -7,6 +7,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from typing import List, Dict
 import uvicorn
+import json
 
 # Initialiser l'application FastAPI
 app = FastAPI()
@@ -26,7 +27,8 @@ app.mount("/static", StaticFiles(directory="static"), name="static")
 class MockRequest(BaseModel):
     """Définit la structure attendue pour le corps de la requête POST."""
     parameter: str
-    secret: str  # Nouveau champ pour le secret
+    model: str = "gemini-2.5-pro"
+    secret: str  
 
 class ConnectionManager:
     """Gère les connexions WebSocket actives."""
@@ -89,6 +91,7 @@ async def mock_endpoint(payload: MockRequest):
     """
     try:
         input_string = payload.parameter
+        provided_model = payload.model
         provided_secret = payload.secret
 
         # Vérification du secret AVANT tout traitement
@@ -107,6 +110,15 @@ async def mock_endpoint(payload: MockRequest):
         if not manager.active_connections:
             raise HTTPException(status_code=503, detail="Aucun client WebSocket n'est connecté.")
 
+        # Créer un dictionnaire avec les données à envoyer
+        message_data = {
+            "prompt": input_string,
+            "model": selected_model
+        }
+        
+        # Envoyer le message via WebSocket (sérialiser en JSON)
+        response_future = await manager.broadcast(json.dumps(message_data))
+        
         # Envoyer le message via WebSocket et obtenir un "future" pour la réponse
         print("Envoi du message au client WebSocket...")
         response_future = await manager.broadcast(input_string)