Update main.py

main.py

@@ -198,11 +198,7 @@ server_status = False
 @app.post("api/v1/chat/completions")
 async def get_completion(payload: Payload, request: Request):
     # Check server status
-    if not server_status:
-        return JSONResponse(
-            status_code=503,
-            content={"message": "Server is under maintenance. Please try again later."}
-        )
+    
     
     model_to_use = payload.model if payload.model else "gpt-4o-mini"
         
@@ -225,8 +221,13 @@ async def get_completion(payload: Payload, request: Request):
     # Current time and IP logging
     current_time = (datetime.datetime.utcnow() + datetime.timedelta(hours=5, minutes=30)).strftime("%Y-%m-%d %I:%M:%S %p")
     aaip = request.client.host
-    print(f"Time: {current_time}, {aaip} , {model_to_use}")
+    print(f"Time: {current_time}, {aaip} , {model_to_use}, server status :- {server_status}")
     print(payload_dict) 
+    if not server_status:
+        return JSONResponse(
+            status_code=503,
+            content={"message": "Server is under maintenance. Please try again later."}
+        )
     scraper = cloudscraper.create_scraper()    
     async def stream_generator(payload_dict):