Update main.py

main.py

@@ -32,7 +32,20 @@ if not secret_api_endpoint or not secret_api_endpoint_2 or not secret_api_endpoi
 alternate_models = {"gpt-4o-mini", "claude-3-haiku", "llama-3.1-70b", "mixtral-8x7b"}
 
 available_model_ids = []
-
+def check_server_status():
+    server_down = True
+    
+    def decorator(func):
+        async def wrapper(*args, **kwargs):
+            if server_down:
+                raise HTTPException(
+                    status_code=503, 
+                    detail="Server is currently unavailable. All services are temporarily down."
+                )
+            return await func(*args, **kwargs)
+        return wrapper
+    return decorator
+    
 class Payload(BaseModel):
     model: str
     messages: list
@@ -146,6 +159,7 @@ async def fetch_models():
 
 @app.post("/chat/completions")
 @app.post("/v1/chat/completions")
+@check_server_status()
 async def get_completion(payload: Payload,request: Request):
     model_to_use = payload.model if payload.model else "gpt-4o-mini"
     # Validate model availability