Update endpoints.py

endpoints.py

@@ -3,17 +3,38 @@ from fastapi.responses import StreamingResponse, JSONResponse
 from fastapi.encoders import jsonable_encoder
 from typing import Optional
 from models import ChatRequest, VoiceOutputRequest, RiskLevel
-from auth import get_current_user
+from datetime import datetime
+from bson import ObjectId
+import asyncio
+from config import notifications_collection, send_push_notification, decode_access_token, AUTH_SPACE_URL, logger
+import httpx
 from utils import clean_text_response
 from analysis import analyze_patient_report
 from voice import recognize_speech, text_to_speech, extract_text_from_pdf
 from docx import Document
 import re
 import io
-from datetime import datetime
-from bson import ObjectId
-import asyncio
-from config import notifications_collection, send_push_notification
+
+async def get_current_user(token: str = Depends(lambda: None)):
+    if not token:
+        raise HTTPException(status_code=401, detail="Authorization token required")
+    if token.startswith("Bearer "):
+        token = token.split(" ")[1]
+    payload = decode_access_token(token)
+    if not payload:
+        raise HTTPException(status_code=401, detail="Invalid token")
+    email = payload.get("sub")
+    if not email:
+        raise HTTPException(status_code=401, detail="Invalid token payload")
+
+    # Fetch user from auth Space
+    async with httpx.AsyncClient() as client:
+        headers = {"Authorization": f"Bearer {token}"}
+        response = await client.get(f"{AUTH_SPACE_URL}/me", headers=headers)
+        if response.status_code != 200:
+            raise HTTPException(status_code=response.status_code, detail="Could not validate credentials")
+        user = response.json()
+        return user
 
 def create_router(agent, logger, patients_collection, analysis_collection, users_collection, notifications_collection):
     router = APIRouter()