Update app.py

app.py

@@ -42,7 +42,11 @@ async def generate_response(prompt: str):
     if not prompt:
         raise HTTPException(status_code=400, detail="Prompt cannot be empty")
     
-    return token
+    return StreamingResponse(
+        response_with_token(),
+        media_type="text/event-stream",
+        headers={"X-Token": token}  # Optionally include token in headers
+    )
 
 def get_app():
     return app