Update app.py

app.py

@@ -1,4 +1,3 @@
-import os
 import gradio as gr
 import fal_client
 import time
@@ -6,18 +5,17 @@ import requests
 from io import BytesIO
 from PIL import Image
 
-# Ensure the API key is set via environment variable
-if "FAL_KEY" not in os.environ:
-    os.environ["FAL_KEY"] = "YOUR_API_KEY"  # Replace with your actual API key
-
-def generate_image(prompt):
+def generate_image(api_key, prompt):
     try:
-        # Map default parameters
+        # Set the API key provided by the user
+        fal_client.api_key = api_key
+
+        # Default parameters
         image_size = "landscape_4_3"
         num_images = 1
         enable_safety_checker = True
         safety_tolerance = "2"
-        
+
         handler = fal_client.submit(
             "fal-ai/flux-pro/v1.1",
             arguments={
@@ -25,10 +23,10 @@ def generate_image(prompt):
                 "image_size": image_size,
                 "num_images": num_images,
                 "enable_safety_checker": enable_safety_checker,
-                "safety_tolerance": safety_tolerance
+                "safety_tolerance": safety_tolerance,
             },
         )
-        
+
         # Wait for the result
         while True:
             status = handler.status()
@@ -37,14 +35,14 @@ def generate_image(prompt):
             elif status["status"] == "failed":
                 return "The image generation failed."
             time.sleep(1)  # Wait for 1 second before checking again
-        
+
         result = handler.get()
         if not result["images"]:
             return "No images were generated."
-        
+
         image_info = result["images"][0]
         image_url = image_info["url"]
-        
+
         # Download the image
         response = requests.get(image_url)
         if response.status_code == 200:
@@ -58,10 +56,13 @@ def generate_image(prompt):
 # Set up the Gradio interface
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(label="Prompt", placeholder="Enter your prompt here..."),
+    inputs=[
+        gr.Textbox(label="API Key", placeholder="Enter your API key here...", type="password"),
+        gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
+    ],
     outputs=gr.Image(label="Generated Image"),
     title="FLUX1.1 [pro] Image Generation",
-    description="Generate an image using the FLUX1.1 [pro] model by providing a text prompt."
+    description="Generate an image using the FLUX1.1 [pro] model by providing your API key and a text prompt."
 )
 
 iface.launch()