Update app.py

app.py

@@ -199,50 +199,32 @@ async def chat(request: Request):
     data = await request.json()
     user_message = data.get("message", "").strip()
     if not user_message:
-        from fastapi import HTTPException
         raise HTTPException(status_code=400, detail="Le champ 'message' est requis.")
 
     try:
-        # Utiliser l'API HF standard sans provider
+        # Utiliser le provider novita comme demandé
         response = hf_client.text_generation(
             model="mistralai/Mistral-7B-Instruct-v0.3",
             prompt=f"<s>[INST] Tu es un assistant médical spécialisé en schizophrénie. Réponds à cette question: {user_message} [/INST]",
             max_new_tokens=512,
-            temperature=0.7
+            temperature=0.7,
+            provider="novita"  # Spécifier le provider novita ici
         )
         
         return {"response": response}
     
     except Exception as e:
-        from fastapi import HTTPException
         import traceback
-        print(f"Erreur détaillée: {traceback.format_exc()}")  # Log détaillé de l'erreur
+        print(f"Erreur détaillée: {traceback.format_exc()}")
         raise HTTPException(status_code=502, detail=f"Erreur d'inférence HF : {str(e)}")
 
 
+
 @app.get("/data")
 async def get_data():
     data = {"data": np.random.rand(100).tolist()}
     return JSONResponse(data)
 
-
-
-app.mount("/", StaticFiles(directory="static", html=True), name="static")
-
-if __name__ == "__main__":
-    import uvicorn
-
-    print(args)
-    uvicorn.run(
-        "app:app",
-        host=args.host,
-        port=args.port,
-        reload=args.reload,
-        ssl_certfile=args.ssl_certfile,
-        ssl_keyfile=args.ssl_keyfile,
-    )
-
-
 # Endpoint pour récupérer toutes les conversations d'un utilisateur
 @app.get("/api/conversations")
 async def get_conversations(current_user: dict = Depends(get_current_user)):
@@ -376,4 +358,21 @@ async def delete_conversation(conversation_id: str, current_user: dict = Depends
         
         return {"success": True}
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Erreur serveur: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Erreur serveur: {str(e)}")
+
+app.mount("/", StaticFiles(directory="static", html=True), name="static")
+
+if __name__ == "__main__":
+    import uvicorn
+
+    print(args)
+    uvicorn.run(
+        "app:app",
+        host=args.host,
+        port=args.port,
+        reload=args.reload,
+        ssl_certfile=args.ssl_certfile,
+        ssl_keyfile=args.ssl_keyfile,
+    )
+
+