Update app.py

app.py

@@ -4,18 +4,8 @@ import os
 from typing import Optional, List
 from huggingface_hub import whoami
 
-# It is recommended to create this as a Secret on your Hugging Face Space
-# For example: FAL_KEY = "fal_key_..."
-FAL_KEY = os.getenv("FAL_KEY", "")
 
-# Set the key for the fal_client
-if FAL_KEY:
-    fal_client.api_key = FAL_KEY
-
-def get_fal_key():
-    """Checks for the FAL_KEY and raises a Gradio error if it's not set."""
-    if not FAL_KEY:
-        raise gr.Error("FAL_KEY is not set. Please add it to your Hugging Face Space secrets.")
+fal_client.api_key = FAL_KEY
 
 def verify_pro_status(token: Optional[gr.OAuthToken]) -> bool:
     """Verifies if the user is a Hugging Face PRO user using their token."""
@@ -32,7 +22,6 @@ def verify_pro_status(token: Optional[gr.OAuthToken]) -> bool:
 
 def run_single_image_logic(prompt: str, image: Optional[str] = None) -> str:
     """Handles text-to-image or single image-to-image and returns a single URL string."""
-    get_fal_key()
     if image:
         image_url = fal_client.upload_file(image)
         result = fal_client.run(
@@ -50,7 +39,6 @@ def run_multi_image_logic(prompt: str, images: List[str]) -> str:
     """
     Handles multi-image editing by sending a list of URLs in a single API call.
     """
-    get_fal_key()
     if not images:
         raise gr.Error("Please upload at least one image in the 'Multiple Images' tab.")
 
@@ -75,7 +63,7 @@ with gr.Blocks(theme=gr.themes.Citrus()) as demo:
     main_interface = gr.Column(visible=False)
 
     with main_interface:
-        gr.Markdown("## Thanks for being a PRO User! 🤗")
+        gr.Markdown("## Thanks for being a PRO! 🤗")
         with gr.Row():
             with gr.Column(scale=1):
                 active_tab_state = gr.State(value="single")
@@ -96,13 +84,11 @@ with gr.Blocks(theme=gr.themes.Citrus()) as demo:
                 )
                 generate_button = gr.Button("Generate", variant="primary")
 
-            # RIGHT COLUMN: Outputs
             with gr.Column(scale=1):
                 output_image = gr.Image(label="Output", interactive=False)
                 use_image_button = gr.Button("♻️ Use this Image for Next Edit")
 
     # --- Event Handlers ---
-
     def unified_generator(
         prompt: str,
         single_image: Optional[str],
@@ -126,10 +112,8 @@ with gr.Blocks(theme=gr.themes.Citrus()) as demo:
         outputs=[output_image],
     )
 
-    # Corrected handler for the continuous editing loop.
-    # It takes the output image and directly returns it to be used as the input.
     use_image_button.click(
-        lambda img: img, # A simple function that returns its input
+        lambda img: img, 
         inputs=[output_image],
         outputs=[image_input]
     )