Update app.py

app.py

@@ -103,6 +103,15 @@ def load_pipeline():
         pipeline_loaded = True
     return pipeline
 
+def load_default_image():
+    """Load the default man.webp image"""
+    if os.path.exists("man.webp"):
+        try:
+            return Image.open("man.webp")
+        except Exception as e:
+            print(f"Error loading default image: {e}")
+    return None
+
 @spaces.GPU(duration=120)
 def style_transfer(input_image, style_name, prompt_suffix, num_inference_steps, guidance_scale, seed):
     """
@@ -181,10 +190,6 @@ def style_transfer(input_image, style_name, prompt_suffix, num_inference_steps,
         torch.cuda.empty_cache()
         return None
 
-def select_style(style_name):
-    """Handler for thumbnail clicks"""
-    return style_name, style_descriptions.get(style_name, "")
-
 def create_thumbnail_grid():
     """Create a gallery of style thumbnails"""
     thumbnails = []
@@ -209,18 +214,10 @@ def create_thumbnail_grid():
     return thumbnails
 
 # Create Gradio interface
-with gr.Blocks(title="FLUX.1 Kontext Style Transfer", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("""
-    # 🎨 FLUX.1 Kontext Style Transfer
-    
-    Transform your images into various artistic styles using FLUX.1-Kontext-dev and high-quality style LoRAs.
-    
-    This demo uses the official Owen777/Kontext-Style-Loras collection with 22 different artistic styles!
-    """)
+with gr.Blocks(title="Flux Kontext Style LoRA", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🎨 Flux Styler : Flux Kontext Style LoRA")
     
     # Thumbnail Grid Section
-    gr.Markdown("### 🖼️ Click a style thumbnail to select it:")
-    
     with gr.Row():
         style_gallery = gr.Gallery(
             value=create_thumbnail_grid(),
@@ -235,21 +232,19 @@ with gr.Blocks(title="FLUX.1 Kontext Style Transfer", theme=gr.themes.Soft()) as
             show_download_button=False
         )
     
-    gr.Markdown("---")
-    
     with gr.Row():
         with gr.Column(scale=1):
             input_image = gr.Image(
-                label="Upload Image",
+                label="Input Image",
                 type="pil",
-                height=400
+                height=400,
+                value=load_default_image()
             )
             
             style_dropdown = gr.Dropdown(
                 choices=list(style_type_lora_dict.keys()),
                 value="Ghibli",
                 label="Selected Style",
-                info="Choose from 22 different artistic styles or click a thumbnail above",
                 elem_id="style_dropdown"
             )
             
@@ -262,7 +257,7 @@ with gr.Blocks(title="FLUX.1 Kontext Style Transfer", theme=gr.themes.Soft()) as
             
             prompt_suffix = gr.Textbox(
                 label="Additional Instructions (Optional)",
-                placeholder="Add extra details like 'make it more colorful' or 'add dramatic lighting'...",
+                placeholder="Add extra details...",
                 lines=2
             )
             
@@ -272,8 +267,7 @@ with gr.Blocks(title="FLUX.1 Kontext Style Transfer", theme=gr.themes.Soft()) as
                     maximum=50,
                     value=24,
                     step=1,
-                    label="Inference Steps",
-                    info="More steps = better quality but slower"
+                    label="Inference Steps"
                 )
                 
                 guidance = gr.Slider(
@@ -281,15 +275,13 @@ with gr.Blocks(title="FLUX.1 Kontext Style Transfer", theme=gr.themes.Soft()) as
                     maximum=5.0,
                     value=2.5,
                     step=0.1,
-                    label="Guidance Scale",
-                    info="How closely to follow the prompt (2.5 recommended)"
+                    label="Guidance Scale"
                 )
                 
                 seed = gr.Number(
                     label="Seed",
                     value=42,
-                    precision=0,
-                    info="Set to 0 for random results"
+                    precision=0
                 )
             
             generate_btn = gr.Button("🎨 Transform Image", variant="primary", size="lg")
@@ -300,16 +292,6 @@ with gr.Blocks(title="FLUX.1 Kontext Style Transfer", theme=gr.themes.Soft()) as
                 type="pil",
                 height=400
             )
-            
-            gr.Markdown("""
-            ### 💡 Tips:
-            - Click any thumbnail above to quickly select a style
-            - All images are resized to 1024x1024
-            - First run downloads the model (~12GB)
-            - Each style transformation takes ~30-60 seconds
-            - Try different styles to find the best match!
-            - Use additional instructions for fine control
-            """)
     
     # Handle gallery selection
     def on_gallery_select(evt: gr.SelectData):
@@ -337,20 +319,6 @@ with gr.Blocks(title="FLUX.1 Kontext Style Transfer", theme=gr.themes.Soft()) as
         outputs=[style_info]
     )
     
-    # Examples
-    gr.Examples(
-        examples=[
-            ["https://huggingface.co/datasets/black-forest-labs/kontext-bench/resolve/main/test/images/0003.jpg", "Ghibli", ""],
-            ["https://huggingface.co/datasets/black-forest-labs/kontext-bench/resolve/main/test/images/0003.jpg", "3D_Chibi", "make it extra cute"],
-            ["https://huggingface.co/datasets/black-forest-labs/kontext-bench/resolve/main/test/images/0003.jpg", "Van_Gogh", "with swirling sky"],
-            ["https://huggingface.co/datasets/black-forest-labs/kontext-bench/resolve/main/test/images/0003.jpg", "Pixel", "8-bit retro game style"],
-        ],
-        inputs=[input_image, style_dropdown, prompt_suffix],
-        outputs=output_image,
-        fn=style_transfer,
-        cache_examples=False
-    )
-    
     # Connect the generate button
     generate_btn.click(
         fn=style_transfer,
@@ -360,16 +328,6 @@ with gr.Blocks(title="FLUX.1 Kontext Style Transfer", theme=gr.themes.Soft()) as
     
     gr.Markdown("""
     ---
-    ### 📚 Available Styles:
-    
-    **Anime/Cartoon**: Ghibli, American Cartoon, Jojo, Snoopy, Rick & Morty, Irasutoya  
-    **3D/Geometric**: 3D Chibi, Poly, LEGO, Clay Toy  
-    **Traditional Art**: Chinese Ink, Oil Painting, Van Gogh, Picasso, Pop Art  
-    **Craft/Material**: Fabric, Origami, Paper Cutting, Macaron  
-    **Digital/Modern**: Pixel, Line, Vector
-    
-    ---
-    
     Powered by ❤️ https://discord.gg/openfreeai
     """)