Update app.py

app.py

@@ -40,9 +40,6 @@ def query_with_auto_routing(prompt, model, custom_lora, is_negative=False, steps
         if not api_key:
             raise gr.Error("HF_READ_TOKEN not found. Please set your Hugging Face API token.")
         
-        # Initialize client with automatic provider selection (default is "auto")
-        client = InferenceClient(api_key=api_key)
-        
         # Determine the model to use
         if custom_lora.strip() != "":
             model_id = custom_lora.strip()
@@ -54,10 +51,12 @@ def query_with_auto_routing(prompt, model, custom_lora, is_negative=False, steps
         # Apply model-specific prompt enhancements
         enhanced_prompt = apply_model_prompt_enhancements(model, enhanced_prompt)
         
-        # Generate image using automatic provider routing
+        # Generate image using appropriate provider
         if input_image is not None:
-            # Image-to-image generation
+            # Image-to-image generation - use Replicate provider as it supports image-to-image
             print(f"Performing image-to-image generation with model: {model_id}")
+            print("Using Replicate provider for image-to-image generation")
+            client = InferenceClient(provider="replicate", api_key=api_key)
             image = client.image_to_image(
                 image=input_image,
                 prompt=enhanced_prompt,
@@ -69,7 +68,8 @@ def query_with_auto_routing(prompt, model, custom_lora, is_negative=False, steps
                 # Note: strength parameter is handled differently in image_to_image
             )
         else:
-            # Text-to-image generation
+            # Text-to-image generation - use automatic provider selection
+            client = InferenceClient(api_key=api_key)
             image = client.text_to_image(
                 prompt=enhanced_prompt,
                 model=model_id,
@@ -618,6 +618,9 @@ with gr.Blocks(theme='Nymbo/Alyx_Theme') as dalle:
         with gr.Accordion("Advanced Settings Overview", open=False):
             gr.Markdown(
             """
+            ## Enable Image Editing
+            ###### When enabled, this feature allows you to upload an image and use the FLUX.1-Kontext-dev model to edit it. This model specializes in image-to-image transformations, allowing you to modify existing images based on your text prompts. The system automatically uses the Replicate provider for image-to-image generation as it supports this functionality.
+
             ## Negative Prompt
             ###### This box is for telling the AI what you don't want in your images. Think of it as a way to avoid certain elements. For instance, if you don't want blurry images or extra limbs showing up, this is where you'd mention it.