Update app.py

app.py

@@ -4,8 +4,7 @@ import torch
 import uvicorn
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from fastapi_poe import PoeBot, make_app
-from fastapi_poe.types import QueryRequest
-from sse_starlette.sse import ServerSentEvent
+from fastapi_poe.types import QueryRequest, PartialResponse
 
 # --- CONFIGURATION DU MODÈLE ---
 MODEL_ID = "Finisha-F-scratch/Charlotte-amity"
@@ -52,20 +51,17 @@ class CharlottePoeBot(PoeBot):
         full_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
         response_text = full_text[len(last_message):].strip()
         
-        # Envoi de la réponse sous forme de flux (streaming)
-        yield ServerSentEvent(event="text", data=response_text)
+        # CORRECTION : Utilisation du format natif PartialResponse de Poe
+        yield PartialResponse(text=response_text)
 
 # --- CREATION DE L'APPLICATION ---
 bot = CharlottePoeBot()
-# Correction ici : on enlève l'argument obsolète
 app = make_app(bot)
 
 # --- LANCEMENT DU SERVEUR ---
 if __name__ == "__main__":
-    # Hugging Face impose le port 7860
     port_to_use = int(os.environ.get("PORT", 7860))
     print(f"--> Démarrage d'Uvicorn sur le port {port_to_use}...")
     
-    # On lance le serveur web sur l'adresse 0.0.0.0 pour qu'il soit accessible de l'extérieur
     uvicorn.run(app, host="0.0.0.0", port=port_to_use)