fix

app.py

@@ -43,47 +43,65 @@ def download_image_locally(image_url: str, local_path: str = "downloaded_image.p
     return local_path
 
 
-def login(oauth_token: gr.OAuthToken | None, fal_key_from_ui: str | None):
+def login(oauth_token: gr.OAuthToken | None):
     """
-    Login to Hugging Face and FAL.
+    Login to Hugging Face and check initial key statuses.
 
     Args:
         oauth_token (gr.OAuthToken | None): The OAuth token from Hugging Face.
-        fal_key_from_ui (str | None): The FAL key from the UI.
     """
-    global TOKEN, FAL_KEY
+    global TOKEN
 
     if oauth_token and oauth_token.token:
         print("Received OAuth token, logging in for Hugging Face...")
         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:
-            print("No Hugging Face OAuth token received and HF_TOKEN environment variable not set.")
+            if not TOKEN:
+                print("No Hugging Face OAuth token received and HF_TOKEN environment variable not set.")
+                gr.Warning("Hugging Face token not set. Image generation via HF Inference Providers might fail.")
+            else:  # It was already set, and no new OAuth token, so it's still configured
+                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.")
+
 
+def set_fal_key(fal_key_from_ui: str | None):
+    """
+    Sets the FAL API key from the UI.
+    Args:
+        fal_key_from_ui (str | None): The FAL key from the UI textbox.
+    """
+    global FAL_KEY
     if fal_key_from_ui and fal_key_from_ui.strip():
         FAL_KEY = fal_key_from_ui.strip()
-    elif os.environ.get("FAL_KEY"):
-        if FAL_KEY == os.environ.get("FAL_KEY"):
-            print("Using FAL_KEY from environment variable.")
+        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
+            print("FAL_KEY UI input empty, using key from environment variable.")
+            gr.Info("FAL_KEY is configured from environment variable (UI input was empty).")
         else:
-            FAL_KEY = os.environ.get("FAL_KEY")
-            print("Using FAL_KEY from environment variable (UI input was blank).")
-            gr.Info("FAL_KEY has been set from environment variable.")
-
-    else:
-        print("FAL_KEY not provided in UI or environment.")
-        FAL_KEY = None
-
-    if not TOKEN:
-        gr.Warning("Hugging Face token not set. Image generation via HF Inference Providers might fail.")
-    else:
-        gr.Info("Hugging Face token is configured.")
-
-    if not FAL_KEY:
+            FAL_KEY = None
+            print("FAL_KEY UI input empty and not found in environment. FAL_KEY is now unset.")
+            gr.Warning("FAL_KEY is not set. Video generation will not work.")
+    elif not FAL_KEY:
+        print("FAL_KEY not provided in UI and not in environment.")
         gr.Warning("FAL_KEY not set. Video generation will not work.")
     else:
         gr.Info("FAL_KEY is configured.")
@@ -207,18 +225,26 @@ css = """
 with gr.Blocks(css=css) as demo:
     demo.load(login, inputs=None, outputs=None)
     with gr.Sidebar():
-        gr.Markdown("# Inference Provider")
+        gr.Markdown("# Authentication")
         gr.Markdown(
-            "This Space showcases the black‑forest‑labs/FLUX.1‑dev model, served by the nebius API. Sign in with your Hugging Face account to use this API."
+            "Sign in with Hugging Face for image generation. Separately, set your FAL_KEY for video generation."
         )
-        hf_login_button = gr.LoginButton("Sign in")
+
+        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)
+
+        gr.Markdown("### FAL API Key (for Video Generation)")
         fal_key_input = gr.Textbox(
-            label="FAL_KEY",
+            label="FAL API Key",
             placeholder="Enter your FAL API Key here",
             type="password",
             value=FAL_KEY if FAL_KEY else "",  # Pre-fill if loaded from env
         )
-        hf_login_button.click(fn=login, inputs=[hf_login_button, fal_key_input], outputs=None)
+        set_fal_key_button = gr.Button("Set FAL Key")
+        set_fal_key_button.click(fn=set_fal_key, inputs=[fal_key_input], outputs=None)
+
     with gr.Column(elem_id="col-container"):
         gr.Markdown(
             """# FLUX.1 [schnell] with fal‑ai through HF Inference Providers ⚡\nLearn more about HF Inference Providers [here](https://huggingface.co/docs/inference-providers/index)"""
@@ -298,7 +324,7 @@ with gr.Blocks(css=css) as demo:
             outputs=[result, seed_number],
         ).then(
             lambda img_path, vid_accordion, vid_btn: {  # Make video section interactive
-                vid_accordion: gr.Accordion(open=True, interactive=True),
+                vid_accordion: gr.Accordion(open=True),
                 vid_btn: gr.Button(interactive=True),
             },
             inputs=[result],