Update app.py

app.py

@@ -19,11 +19,13 @@ MAX_SEED = np.iinfo(np.int32).max
 
 # Illumination options mapping
 ILLUMINATION_OPTIONS = {
+    "sunset over sea": "add warm sunset light reflecting off water, creating a peaceful atmosphere with soft, diffused lighting and gentle color gradients",
     "sunshine from window": "Add directional sunlight from window source. Increase brightness on lit areas. Create hard shadows with sharp edges. Set warm white color temperature. Add visible light rays and dust particles in beams.",
     "neon night, city": "Replace ambient lighting with neon sources. Set primary colors to electric blue, magenta, and green. Increase color saturation significantly. Add colored reflections on wet surfaces. Reduce overall brightness.",
     "sunset over sea": "Set warm orange-red color temperature. Increase warm tones substantially. Add horizontal light source from background. Create soft gradient lighting. Add water reflections if visible.",
     "golden time": "Set golden color temperature. Increase contrast moderately. Add rim lighting on subjects. Enhance warm colors. Create soft directional shadows. Reduce blue tones.",
-    "sci-fi RGB glowing, cyberpunk": "Add multiple colored light sources in cyan, magenta, and yellow. Maximize color saturation. Add glowing edges on objects. Create chromatic aberration effects. Set high contrast lighting.",
+    "sci-fi RGB glowing, cyberpunk": "add futuristic RGB lighting with electric blues, hot pinks, and neon greens creating a high-tech atmosphere with dramatic color separation and glowing effects",
+    "neon night, city":"add vibrant neon lights in electric blues, magentas, and greens casting colorful reflections on surfaces, creating a cyberpunk urban atmosphere with dramatic color contrasts",
     "natural lighting": "Set neutral white color temperature. Maintain balanced exposure. Create soft shadows with gradual falloff. Preserve original color saturation. Add subtle ambient occlusion.",
     "warm atmosphere, at home, bedroom": "Set very warm color temperature. Reduce brightness moderately. Add soft diffused lighting. Increase red and yellow tones. Create minimal shadows.",
     "magic lit": "Add subtle colored rim lighting in purple and blue tones. Increase brightness selectively on edges. Add soft glow effects. Create ethereal light sources with no clear origin.",
@@ -77,6 +79,7 @@ ILLUMINATION_OPTIONS = {
     "soft, diffused foggy glow": "Add volumetric fog throughout scene. Reduce contrast substantially. Create soft lighting with no harsh shadows. Set neutral color temperature.",
     "underwater luminescence": "Set blue-green color temperature. Add bioluminescent glowing effects. Create caustic light patterns. Add particle effects in water.",
     "rain-soaked reflections in city lights": "Add reflections on wet surfaces. Create light streaking effects. Set urban lighting with multiple colored sources. Add rain particle effects.",
+    "golden time": "add warm golden hour lighting with soft, diffused sunlight creating a magical glow, enhanced contrast, and rich warm tones throughout the scene",
     "golden sunlight streaming through trees": "Add god rays effect. Set golden color temperature. Create dappled light patterns. Add atmospheric particles in light beams.",
     "fireflies lighting up a summer night": "Add small glowing points of light. Set warm color temperature. Create subtle animated glow effects. Maintain dark ambient lighting.",
     "glowing embers from a forge": "Add orange-red glowing particle effects. Set very warm color temperature. Create intense heat glow. Add sparking effects.",
@@ -89,7 +92,7 @@ ILLUMINATION_OPTIONS = {
 }
 
 @spaces.GPU
-def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, progress=gr.Progress(track_tqdm=True)):
+def infer(input_image, prompt, illumination_dropdown, seed=42, randomize_seed=False, guidance_scale=2.5, progress=gr.Progress(track_tqdm=True)):
     
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -128,13 +131,20 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""# FLUX.1 Kontext [dev] Relight 💡
         """)
-        gr.Markdown(f"""Kontext[dev] used for object relighting ✨
+        gr.Markdown(f"""Kontext[dev] finetuned for scene relighting ✨
          """)
 
         with gr.Row():
             with gr.Column():
                 input_image = gr.Image(label="Upload the image for relighting", type="pil")
-                
+                prompt = gr.Textbox(
+                    label="prompt",
+                    show_label=True,
+                    max_lines=3,
+                    placeholder="Select an illumination style above or enter your custom description...",
+                    value="bright natural sunlight streaming through a window, creating warm golden beams and soft shadows across the scene, with high contrast between lit and shadowed areas",
+                    container=True
+                )
                 # Illumination options dropdown
                 illumination_dropdown = gr.Dropdown(
                     choices=list(ILLUMINATION_OPTIONS.keys()) + ["Custom"],
@@ -143,15 +153,7 @@ with gr.Blocks(css=css) as demo:
                     info="Select a predefined lighting style or choose 'Custom' to use your own prompt"
                 )
                 
-                # Prompt textbox that gets updated by dropdown selection
-                prompt = gr.Textbox(
-                    label="Lighting Description",
-                    show_label=True,
-                    max_lines=3,
-                    placeholder="Select an illumination style above or enter your custom description...",
-                    value="bright natural sunlight streaming through a window, creating warm golden beams and soft shadows across the scene, with high contrast between lit and shadowed areas",
-                    container=True
-                )
+                
                 
                 run_button = gr.Button("Run", scale=0, variant="primary")
                 
@@ -201,7 +203,7 @@ with gr.Blocks(css=css) as demo:
         gr.on(
             triggers=[run_button.click, prompt.submit],
             fn = infer,
-            inputs = [input_image, prompt, seed, randomize_seed, guidance_scale],
+            inputs = [input_image, prompt, illumination_dropdown, seed, randomize_seed, guidance_scale],
             outputs = [result, seed]
         )