OAuth

app.py

@@ -18,6 +18,11 @@ from library.utils_prompt import *
 
 OR = OpenRouterAPI()
 
+# Get authorized users from environment variable/secret
+authorized_users_str = os.environ.get("AUTHORIZED_USER_IDS", "")
+AUTHORIZED_USER_IDS = set(authorized_users_str.split(',') if authorized_users_str and authorized_users_str.strip() else [])
+
+
 # Define model pricing information (approximate costs per 100 image API calls)
 MODEL_PRICING = {
     "google/gemini-2.0-flash-001": "$0.03",
@@ -177,7 +182,7 @@ def create_demo():
                         text = f"""**Current Model**: Gemini 2.0 Flash (cheap) 
                     **Estimated cost per 100 Images**: {MODEL_PRICING["google/gemini-2.0-flash-001"]}"""
                         
-                        return gr.Update(choices=preferred_models, label="Select Model",value="google/gemini-2.0-flash-001"),False,text
+                        return gr.Update(choices=preferred_models_auth, label="Select Model",value="google/gemini-2.0-flash-001"),False,text
                     else:
                         all_models_list = preferred_models