fix login

app.py

@@ -8,29 +8,26 @@ import requests
 from dotenv import load_dotenv
 from huggingface_hub import InferenceClient
 
-FAL_KEY = None
-
-
 load_dotenv()
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 TOKEN = None
+FAL_KEY = None
 
-
-def download_image_locally(image_url: str, local_path: str = "downloaded_image.png") -> str:
-    """Download an image from a URL to a local path.
+def download_locally(url: str, local_path: str = "downloaded_file.png") -> str:
+    """Download an image or a video from a URL to a local path.
 
     Args:
-        image_url (str): The URL of the image to download. Must be an http(s) URL.
-        local_path (str, optional): The path (including filename) where the file should be saved. Defaults to "downloaded_image.png".
+        url (str): The URL of the image to download. Must be an http(s) URL.
+        local_path (str, optional): The path (including filename) where the file should be saved. Defaults to "downloaded_file.png".
 
     Returns:
         str: The filesystem path of the saved file – suitable for returning to a **gr.File** output, or as an MCP tool response.
     """
     if local_path == "":
-        local_path = "downloaded_image.png"
-    response = requests.get(image_url, timeout=30)
+        local_path = "downloaded_file.png"
+    response = requests.get(url, timeout=30)
     response.raise_for_status()
     # If the caller passed only a filename, save into a temporary directory to avoid permission issues
     if os.path.dirname(local_path) == "":
@@ -41,7 +38,7 @@ def download_image_locally(image_url: str, local_path: str = "downloaded_image.p
     return local_path
 
 
-def login(oauth_token: gr.OAuthToken | None):
+def login_hf(oauth_token: gr.OAuthToken | None):
     """
     Login to Hugging Face and check initial key statuses.
 
@@ -49,33 +46,15 @@ def login(oauth_token: gr.OAuthToken | None):
         oauth_token (gr.OAuthToken | None): The OAuth token from Hugging Face.
     """
     global TOKEN
-
     if oauth_token and oauth_token.token:
-        print("Received OAuth token, logging in for Hugging Face...")
+        print("Received OAuth token, logging in...")
         TOKEN = oauth_token.token
-        gr.Info("Hugging Face token is configured from OAuth.")
     else:
-        env_hf_token = os.environ.get("HF_TOKEN")
-        if env_hf_token:
-            TOKEN = env_hf_token
-            print("Using environment variable HF_TOKEN for Hugging Face.")
-            gr.Info("Hugging Face token is configured from HF_TOKEN environment variable.")
-        else:
-            if not TOKEN:
-                print("No Hugging Face OAuth token received and HF_TOKEN environment variable not set.")
-            else:
-                gr.Info("Hugging Face token remains configured (no new OAuth token).")
-
-    if oauth_token is None:
-        if FAL_KEY:
-            print("FAL_KEY found in environment on app load.")
-            gr.Info("FAL_KEY is configured from environment variable.")
-        else:
-            print("FAL_KEY not found in environment on app load.")
-            gr.Warning("FAL_KEY not set. Video generation will not work until set.")
+        print("No OAuth token provided, using environment variable HF_TOKEN.")
+        TOKEN = os.environ.get("HF_TOKEN")
 
 
-def set_fal_key(fal_key_from_ui: str | None):
+def login_fal(fal_key_from_ui: str | None):
     """
     Sets the FAL API key from the UI.
     Args:
@@ -86,20 +65,9 @@ def set_fal_key(fal_key_from_ui: str | None):
         FAL_KEY = fal_key_from_ui.strip()
         os.environ["FAL_KEY"] = FAL_KEY
         print("FAL_KEY has been set from UI input.")
-        gr.Info("FAL_KEY has been updated from UI input.")
-    elif not fal_key_from_ui and FAL_KEY:
-        print("FAL_KEY UI input cleared. Relying on previous value (if any) or environment variable.")
-        env_fal_key = os.environ.get("FAL_KEY")
-        if env_fal_key:
-            FAL_KEY = env_fal_key
-            gr.Info("FAL_KEY is configured from environment variable (UI input was empty).")
-        else:
-            FAL_KEY = None
-            print("FAL_KEY UI input empty and not found in environment. FAL_KEY is now unset.")
-    elif not FAL_KEY:
-        print("FAL_KEY not provided in UI and not in environment.")
     else:
-        gr.Info("FAL_KEY is configured.")
+        FAL_KEY = os.environ.get("FAL_KEY")
+        print("FAL_KEY is configured from environment variable.")
 
 
 def generate_image(prompt: str, seed: int = 42, width: int = 1024, height: int = 1024, num_inference_steps: int = 25):
@@ -219,8 +187,8 @@ css = """
 """
 
 with gr.Blocks(css=css) as demo:
-    demo.load(login, inputs=None, outputs=None)
-    demo.load(set_fal_key, inputs=None, outputs=None)
+    demo.load(login_hf, inputs=None, outputs=None)
+    demo.load(login_fal, inputs=None, outputs=None)
     with gr.Sidebar():
         gr.Markdown("# Authentication")
         gr.Markdown(
@@ -230,7 +198,7 @@ with gr.Blocks(css=css) as demo:
         gr.Markdown("### Hugging Face Login")
         hf_login_button = gr.LoginButton("Sign in with Hugging Face")
         # When hf_login_button is clicked, it provides an OAuthToken or None to the login function.
-        hf_login_button.click(fn=login, inputs=[hf_login_button], outputs=None)
+        hf_login_button.click(fn=login_hf, inputs=[hf_login_button], outputs=None)
 
         gr.Markdown("### FAL Login (for Image to Video)")
         fal_key_input = gr.Textbox(
@@ -240,7 +208,7 @@ with gr.Blocks(css=css) as demo:
             value=os.environ.get("FAL_KEY", ""),  # Pre-fill if loaded from env
         )
         set_fal_key_button = gr.Button("Set FAL Key")
-        set_fal_key_button.click(fn=set_fal_key, inputs=[fal_key_input], outputs=None)
+        set_fal_key_button.click(fn=login_fal, inputs=[fal_key_input], outputs=None)
 
     with gr.Column(elem_id="col-container"):
         gr.Markdown(