Update main.py

main.py

@@ -90,6 +90,7 @@ DEVELOPER_ID = os.environ["DEVELOPER_ID"]
 REVERSE_IMAGE_API = os.environ["REVERSE_IMAGE_API"]
 OCR_API_KEY = os.environ["OCR_API_KEY"]
 ONLY_DEVELOPER_API_KEYS = os.environ["ONLY_DEVELOPER_API_KEYS"]
+ULT_ONLY_API = os.environ["ULT_ONLY_API"]
 HUGGING_TOKEN = os.environ["HUGGING_TOKEN"]
 ASSISTANT_GOOGLE_API_KEYS = os.environ["ASSISTANT_GOOGLE_API_KEYS"]
 #COOKIE_BARD_TOKEN = os.environ["COOKIE_BARD_TOKEN"]
@@ -137,6 +138,10 @@ def validate_api_key_only_devs(api_key: str = Header(...)):
     if api_key not in ONLY_DEVELOPER_API_KEYS:
         raise HTTPException(status_code=401, detail="Invalid API key")
 
+def validate_api_key_only_ULTROID(api_key: str = Header(...)):
+    if api_key not in ULT_ONLY_API:
+        raise HTTPException(status_code=401, detail="Invalid API key")
+
 
 @app.get("/UFoP/blacklist-words")
 def blacklist_words():
@@ -215,7 +220,7 @@ def sibyl_system_ban(
 @app.get("/UFoP/bans")
 def sibyl_system(
     user_id: int = Query(..., description="User ID in query parameter"),
-    api_key: None = Depends(validate_api_key) or Depends(validate_api_key_only_devs)
+    api_key: None = Depends(validate_api_key) or Depends(validate_api_key_only_devs) or Depends(validate_api_key_only_ULTROID)
 ):
     result = db.get_sibyl_system_banned(user_id)
     if result is not None:
@@ -236,7 +241,7 @@ def sibyl_system(
 @app.post("/UFoP/G-AI", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def v1beta3_google_ai(
     item: ChatgptCustom,
-    api_key: None = Depends(validate_api_key)
+    api_key: None = Depends(validate_api_key) or Depends(validate_api_key_only_ULTROID)
 ):
     url = SOURCE_ASSISTANT_GOOGLE_AI
     token = ASSISTANT_GOOGLE_API_KEYS