Create routes.py

api/routes.py

@@ -0,0 +1,60 @@
+import json
+from fastapi import APIRouter, Depends, HTTPException, Request, Response
+from fastapi.responses import StreamingResponse, JSONResponse
+from api.auth import verify_app_secret
+from api.config import GIZAI_API_ENDPOINT, GIZAI_BASE_URL
+from api.models import ChatRequest, ImageResponseModel, ChatCompletionResponse
+from api.utils import process_gizai_response
+from api.logger import setup_logger
+
+logger = setup_logger(__name__)
+
+router = APIRouter()
+
+@router.options("/v1/gizai/chat/completions")
+@router.options("/api/v1/gizai/chat/completions")
+async def gizai_chat_completions_options():
+    return Response(
+        status_code=200,
+        headers={
+            "Access-Control-Allow-Origin": "*",
+            "Access-Control-Allow-Methods": "POST, OPTIONS",
+            "Access-Control-Allow-Headers": "Content-Type, Authorization",
+        },
+    )
+
+@router.get("/v1/gizai/models")
+@router.get("/api/v1/gizai/models")
+async def list_gizai_models():
+    return {"object": "list", "data": GizAI.models}  # Assuming GizAI.models is accessible
+
+@router.post("/v1/gizai/chat/completions")
+@router.post("/api/v1/gizai/chat/completions")
+async def gizai_chat_completions(
+    request: ChatRequest, app_secret: str = Depends(verify_app_secret)
+):
+    logger.info("Entering GizAI chat_completions route")
+    logger.info(f"Processing chat completion request for model: {request.model}")
+
+    model = GizAI.get_model(request.model)
+    if model not in GizAI.models:
+        raise HTTPException(
+            status_code=400,
+            detail=f"Model {request.model} is not supported. Supported models are: {', '.join(GizAI.models)}",
+        )
+
+    if request.stream:
+        logger.info("Streaming response")
+        return StreamingResponse(process_gizai_response(request, model), media_type="text/event-stream")
+    else:
+        logger.info("Non-streaming response")
+        return await process_gizai_response(request, model)
+
+@router.route('/')
+@router.route('/healthz')
+@router.route('/ready')
+@router.route('/alive')
+@router.route('/status')
+@router.get("/health")
+def health_check(request: Request):
+    return Response(content=json.dumps({"status": "ok"}), media_type="application/json")