OAuth

app.py

@@ -22,7 +22,6 @@ OR = OpenRouterAPI()
 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",
@@ -81,6 +80,43 @@ def get_base_filename(filepath):
     filename = os.path.splitext(basename)[0]
     return filename
 
+ # Function to check authorization
+def check_authorization(profile: gr.OAuthProfile | None):
+    
+    is_authorized = profile.username in AUTHORIZED_USER_IDS
+    if is_authorized:
+        # Define preferred and additional models directly in the function
+        preferred_models_auth = [
+            ("Gemini 2.0 Flash (cheap)", "google/gemini-2.0-flash-001"),
+            ("GPT-4.1 Mini", "gpt-4.1-mini"),
+            ("GPT-4.1 (Recommended)", "gpt-4.1"),
+            ("Claude 3.7 Sonnet", "anthropic/claude-3.7-sonnet"),
+            ("Gemini 2.5 Pro", "google/gemini-2.5-pro-preview-03-25"),
+            ("Gemini 2.5 Flash Thinking (Recommended)","google/gemini-2.5-flash-preview")
+        ]
+        
+        additional_models = [
+            ("GPT-4.1 Nano", "gpt-4.1-nano"),
+            ("ChatGPT Latest", "openai/chatgpt-4o-latest"),
+            ("Llama 4 Maverick", "meta-llama/llama-4-maverick")
+        ]
+        
+        # Calculate all models once
+        all_models_list = preferred_models + additional_models
+
+        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_auth, label="Select Model",value="google/gemini-2.0-flash-001"),False,text
+    else:
+        all_models_list = preferred_models
+
+        # Default model value
+        default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
+        text = f"""**Current Model**: Llama 4 Maverick (free)
+    **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
+        return gr.Update(choices=preferred_models, label="Select Model",value=default_model),True,text
+
 # Define the Gradio interface
 def create_demo():
     custom_css = """
@@ -154,43 +190,6 @@ def create_demo():
                     label="Select Model",
                     value="meta-llama/llama-4-maverick:free"
                 )
-
-                # Function to check authorization
-                def check_authorization(profile: gr.OAuthProfile | None):
-                    
-                    is_authorized = profile.username in AUTHORIZED_USER_IDS
-                    if is_authorized:
-                        # Define preferred and additional models directly in the function
-                        preferred_models_auth = [
-                            ("Gemini 2.0 Flash (cheap)", "google/gemini-2.0-flash-001"),
-                            ("GPT-4.1 Mini", "gpt-4.1-mini"),
-                            ("GPT-4.1 (Recommended)", "gpt-4.1"),
-                            ("Claude 3.7 Sonnet", "anthropic/claude-3.7-sonnet"),
-                            ("Gemini 2.5 Pro", "google/gemini-2.5-pro-preview-03-25"),
-                            ("Gemini 2.5 Flash Thinking (Recommended)","google/gemini-2.5-flash-preview")
-                        ]
-                        
-                        additional_models = [
-                            ("GPT-4.1 Nano", "gpt-4.1-nano"),
-                            ("ChatGPT Latest", "openai/chatgpt-4o-latest"),
-                            ("Llama 4 Maverick", "meta-llama/llama-4-maverick")
-                        ]
-                        
-                        # Calculate all models once
-                        all_models_list = preferred_models + additional_models
-
-                        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_auth, label="Select Model",value="google/gemini-2.0-flash-001"),False,text
-                    else:
-                        all_models_list = preferred_models
-            
-                        # Default model value
-                        default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
-                        text = f"""**Current Model**: Llama 4 Maverick (free)
-                    **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
-                        return gr.Update(choices=preferred_models, label="Select Model",value=default_model),True,text
             
                 length_choice = gr.Radio(
                     choices=["short", "medium", "long"],