Improving routing code.

No functional change, just using APIRouter more intelligently.

src/api/routes/audio.py

@@ -9,7 +9,7 @@ from src.services.audio_transcription import (get_supported_formats,
 											  validate_audio_format)
 from src.utils.logger import logger
 
-router = APIRouter(prefix="/audio", tags=["audio"])
+router = APIRouter(prefix="/audio", tags=["Audio"])
 
 @router.post("/transcribe")
 async def transcribe_audio(

src/api/routes/chat.py

@@ -12,7 +12,7 @@ from src.services.medical_response import generate_medical_response
 from src.services.summariser import summarise_title_with_nvidia
 from src.utils.logger import logger
 
-router = APIRouter()
+router = APIRouter(tags=["Chat"])
 
 @router.post("/chat", response_model=ChatResponse)
 async def chat_endpoint(

src/api/routes/doctors.py

@@ -7,9 +7,20 @@ from src.data.repositories.account import (create_doctor, get_all_doctors,
 from src.models.user import DoctorCreateRequest
 from src.utils.logger import logger
 
-router = APIRouter()
+router = APIRouter(prefix="/doctors", tags=["Doctors"])
 
-@router.post("/doctors")
+@router.get("/")
+async def get_all_doctors_route(limit: int = 50):
+	try:
+		logger().info(f"GET /doctors limit={limit}")
+		results = get_all_doctors(limit=limit)
+		logger().info(f"Retrieved {len(results)} doctors")
+		return {"results": results}
+	except Exception as e:
+		logger().error(f"Error getting all doctors: {e}")
+		raise HTTPException(status_code=500, detail=str(e))
+
+@router.post("/")
 async def create_doctor_profile(req: DoctorCreateRequest):
 	try:
 		logger().info(f"POST /doctors name={req.name}")
@@ -25,7 +36,7 @@ async def create_doctor_profile(req: DoctorCreateRequest):
 		logger().error(f"Error creating doctor: {e}")
 		raise HTTPException(status_code=500, detail=str(e))
 
-@router.get("/doctors/{doctor_name}")
+@router.get("/{doctor_name}")
 async def get_doctor(doctor_name: str):
 	try:
 		logger().info(f"GET /doctors/{doctor_name}")
@@ -39,7 +50,7 @@ async def get_doctor(doctor_name: str):
 		logger().error(f"Error getting doctor: {e}")
 		raise HTTPException(status_code=500, detail=str(e))
 
-@router.get("/doctors/search")
+@router.get("/search")
 async def search_doctors_route(q: str, limit: int = 10):
 	try:
 		logger().info(f"GET /doctors/search q='{q}' limit={limit}")
@@ -49,14 +60,3 @@ async def search_doctors_route(q: str, limit: int = 10):
 	except Exception as e:
 		logger().error(f"Error searching doctors: {e}")
 		raise HTTPException(status_code=500, detail=str(e))
-
-@router.get("/doctors")
-async def get_all_doctors_route(limit: int = 50):
-	try:
-		logger().info(f"GET /doctors limit={limit}")
-		results = get_all_doctors(limit=limit)
-		logger().info(f"Retrieved {len(results)} doctors")
-		return {"results": results}
-	except Exception as e:
-		logger().error(f"Error getting all doctors: {e}")
-		raise HTTPException(status_code=500, detail=str(e))

src/api/routes/patients.py

@@ -5,12 +5,35 @@ from fastapi import APIRouter, HTTPException
 from src.data.repositories.patient import (create_patient, get_patient_by_id,
                                            search_patients,
                                            update_patient_profile)
+from src.data.repositories.session import list_patient_sessions
 from src.models.user import PatientCreateRequest, PatientUpdateRequest
 from src.utils.logger import logger
 
-router = APIRouter()
+router = APIRouter(prefix="/patients", tags=["Patients"])
 
-@router.get("/patients/search")
+@router.post("/")
+async def create_patient_profile(req: PatientCreateRequest):
+	try:
+		logger().info(f"POST /patients name={req.name}")
+		patient = create_patient(
+			name=req.name,
+			age=req.age,
+			sex=req.sex,
+			address=req.address,
+			phone=req.phone,
+			email=req.email,
+			medications=req.medications,
+			past_assessment_summary=req.past_assessment_summary,
+			assigned_doctor_id=req.assigned_doctor_id
+		)
+		patient["_id"] = str(patient.get("_id")) if patient.get("_id") else None
+		logger().info(f"Created patient {patient.get('name')} id={patient.get('patient_id')}")
+		return patient
+	except Exception as e:
+		logger().error(f"Error creating patient: {e}")
+		raise HTTPException(status_code=500, detail=str(e))
+
+@router.get("/search")
 async def search_patients_route(q: str, limit: int = 20):
 	try:
 		logger().info(f"GET /patients/search q='{q}' limit={limit}")
@@ -21,7 +44,7 @@ async def search_patients_route(q: str, limit: int = 20):
 		logger().error(f"Error searching patients: {e}")
 		raise HTTPException(status_code=500, detail=str(e))
 
-@router.get("/patients/{patient_id}")
+@router.get("/{patient_id}")
 async def get_patient(patient_id: str):
 	try:
 		logger().info(f"GET /patients/{patient_id}")
@@ -36,29 +59,7 @@ async def get_patient(patient_id: str):
 		logger().error(f"Error getting patient: {e}")
 		raise HTTPException(status_code=500, detail=str(e))
 
-@router.post("/patients")
-async def create_patient_profile(req: PatientCreateRequest):
-	try:
-		logger().info(f"POST /patients name={req.name}")
-		patient = create_patient(
-			name=req.name,
-			age=req.age,
-			sex=req.sex,
-			address=req.address,
-			phone=req.phone,
-			email=req.email,
-			medications=req.medications,
-			past_assessment_summary=req.past_assessment_summary,
-			assigned_doctor_id=req.assigned_doctor_id
-		)
-		patient["_id"] = str(patient.get("_id")) if patient.get("_id") else None
-		logger().info(f"Created patient {patient.get('name')} id={patient.get('patient_id')}")
-		return patient
-	except Exception as e:
-		logger().error(f"Error creating patient: {e}")
-		raise HTTPException(status_code=500, detail=str(e))
-
-@router.patch("/patients/{patient_id}")
+@router.patch("/{patient_id}")
 async def update_patient(patient_id: str, req: PatientUpdateRequest):
 	try:
 		payload = {k: v for k, v in req.model_dump().items() if v is not None}
@@ -70,3 +71,13 @@ async def update_patient(patient_id: str, req: PatientUpdateRequest):
 	except Exception as e:
 		logger().error(f"Error updating patient: {e}")
 		raise HTTPException(status_code=500, detail=str(e))
+
+@router.get("/{patient_id}/sessions")
+async def list_sessions_for_patient(patient_id: str):
+	"""List sessions for a patient from Mongo"""
+	try:
+		logger().info(f"GET /patients/{patient_id}/sessions")
+		return {"sessions": list_patient_sessions(patient_id)}
+	except Exception as e:
+		logger().error(f"Error listing sessions: {e}")
+		raise HTTPException(status_code=500, detail=str(e))

src/api/routes/session.py

@@ -8,14 +8,13 @@ from src.core.state import MedicalState, get_state
 from src.data.repositories.message import list_session_messages
 from src.data.repositories.session import (delete_session,
                                            delete_session_messages,
-                                           ensure_session,
-                                           list_patient_sessions)
+                                           ensure_session)
 from src.models.chat import SessionRequest
 from src.utils.logger import logger
 
-router = APIRouter()
+router = APIRouter(prefix="/sessions", tags=["Session"])
 
-@router.post("/sessions")
+@router.post("/")
 async def create_chat_session(
 	request: SessionRequest,
 	state: MedicalState = Depends(get_state)
@@ -31,7 +30,7 @@ async def create_chat_session(
 		logger().error(f"Error creating session: {e}")
 		raise HTTPException(status_code=500, detail=str(e))
 
-@router.get("/sessions/{session_id}")
+@router.get("/{session_id}")
 async def get_chat_session(
 	session_id: str,
 	state: MedicalState = Depends(get_state)
@@ -60,17 +59,7 @@ async def get_chat_session(
 		logger().error(f"Error getting session: {e}")
 		raise HTTPException(status_code=500, detail=str(e))
 
-@router.get("/patients/{patient_id}/sessions")
-async def list_sessions_for_patient(patient_id: str):
-	"""List sessions for a patient from Mongo"""
-	try:
-		logger().info(f"GET /patients/{patient_id}/sessions")
-		return {"sessions": list_patient_sessions(patient_id)}
-	except Exception as e:
-		logger().error(f"Error listing sessions: {e}")
-		raise HTTPException(status_code=500, detail=str(e))
-
-@router.get("/sessions/{session_id}/messages")
+@router.get("/{session_id}/messages")
 async def list_messages_for_session(session_id: str, patient_id: str, limit: int | None = None):
 	"""List messages for a session from Mongo, verified to belong to the patient"""
 	try:
@@ -86,7 +75,7 @@ async def list_messages_for_session(session_id: str, patient_id: str, limit: int
 		logger().error(f"Error listing messages: {e}")
 		raise HTTPException(status_code=500, detail=str(e))
 
-@router.delete("/sessions/{session_id}")
+@router.delete("/{session_id}")
 async def delete_chat_session(
 	session_id: str,
 	state: MedicalState = Depends(get_state)

src/api/routes/static.py

@@ -3,7 +3,7 @@
 from fastapi import APIRouter, HTTPException
 from fastapi.responses import HTMLResponse
 
-router = APIRouter()
+router = APIRouter(tags=["Static"])
 
 @router.get("/", response_class=HTMLResponse)
 async def get_medical_chatbot():

src/api/routes/user.py

@@ -7,9 +7,9 @@ from src.data.repositories.account import create_account
 from src.models.user import UserProfileRequest
 from src.utils.logger import logger
 
-router = APIRouter()
+router = APIRouter(prefix="/users", tags=["Users"])
 
-@router.post("/users")
+@router.post("/")
 async def create_user_profile(
 	request: UserProfileRequest,
 	state: MedicalState = Depends(get_state)
@@ -38,7 +38,7 @@ async def create_user_profile(
 		logger().error(f"Error creating user profile: {e}")
 		raise HTTPException(status_code=500, detail=str(e))
 
-@router.get("/users/{user_id}")
+@router.get("/{user_id}")
 async def get_user_profile(
 	user_id: str,
 	state: MedicalState = Depends(get_state)