authentication

app.py

@@ -20,16 +20,6 @@ from library.utils_prompt import *
 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 [])
 
-# New function to check if current user is authorized
-def check_authorization(profile: gr.OAuthProfile | None):
-    if profile is None:
-        return False #"Not logged in. Please Login for full access."
-    
-    if profile.username in AUTHORIZED_USER_IDS:
-        return True
-    else:
-        return False
-
 #Api sdk
 OR = OpenRouterAPI()
 
@@ -421,6 +411,15 @@ def create_demo():
 
     return demo
 
+# New function to check if current user is authorized
+def check_authorization(profile: gr.OAuthProfile | None):
+    if profile is None:
+        return False #"Not logged in. Please Login for full access."
+    
+    if profile.username in AUTHORIZED_USER_IDS:
+        return True
+    else:
+        return False
 
 def hello(profile: gr.OAuthProfile | None) -> str:
     if profile is None:
@@ -439,7 +438,8 @@ def create_demo_free():
     
 # Launch the app
 if __name__ == "__main__":
-    if check_authorization(None):
+    profile = gr.OAuthProfile()
+    if check_authorization(profile):
         app = create_demo()
     else:
         app = create_demo_free()