Update app.py

app.py

@@ -28,24 +28,14 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
         seed = random.randint(0, MAX_SEED)
         
     input_image = input_image.convert("RGB")
-    # original_width, original_height = input_image.size
+    prompt_with_template = f"hange the lighting conditions in this image and add {prompt}. Lighting determines how bright or dark different parts of the image appear, where shadows fall, and how colors look. When you relight an image, you're simulating what the photo would look like if it were taken under different lighting conditions."
     
-    # if original_width >= original_height:
-    #     new_width = 1024
-    #     new_height = int(original_height * (new_width / original_width))
-    #     new_height = round(new_height / 64) * 64
-    # else:
-    #     new_height = 1024
-    #     new_width = int(original_width * (new_height / original_height))
-    #     new_width = round(new_width / 64) * 64
-    
-    #input_image_resized = input_image.resize((new_width, new_height), Image.LANCZOS)
     image = pipe(
         image=input_image, 
-        prompt=prompt,
+        prompt=prompt_with_template,
         guidance_scale=guidance_scale,
-        # width=new_width,
-        # height=new_height,
+        width=input_image.size[0],
+        height=input_image.size[1],
         generator=torch.Generator().manual_seed(seed),
     ).images[0]
     return image, seed, gr.update(visible=True)
@@ -60,18 +50,18 @@ css="""
 with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""# FLUX.1 Kontext [dev]
+        gr.Markdown(f"""# FLUX.1 Kontext [dev] Relight
         """)
 
         with gr.Row():
             with gr.Column():
-                input_image = gr.Image(label="Upload the image for editing", type="pil")
+                input_image = gr.Image(label="Upload the image for relighting", type="pil")
                 with gr.Row():
                     prompt = gr.Text(
                         label="Prompt",
                         show_label=False,
                         max_lines=1,
-                        placeholder="Enter your prompt for editing (e.g., 'Remove glasses', 'Add a hat')",
+                        placeholder="scifi RGB flowing, studio lighting",
                         container=False,
                     )
                     run_button = gr.Button("Run", scale=0)