Update app.py

app.py

@@ -357,7 +357,8 @@ DIRECTION_OPTIONS = {
 def infer(
     image,
     prompt,
-    illumination_dropdown, direction_dropdown, 
+    illumination_dropdown="custom", 
+    direction_dropdown="auto", 
     seed=42,
     randomize_seed=True,
     true_guidance_scale=1.0,
@@ -366,7 +367,41 @@ def infer(
     progress=gr.Progress(track_tqdm=True),
 ):
     """
-    Generates an edited image using the Qwen-Image-Edit pipeline with Lightning acceleration.
+    Generates an edited image using Qwen-Image-Edit pipeline model with Lightning LoRA.
+
+    This function takes an input image and generates a video animation based on the provided
+    prompt and parameters. It uses an FP8 qunatized Wan 2.2 14B Image-to-Video model in with Lightning LoRA
+    for fast generation in 4-8 steps.
+    
+    Args:
+        input_image (PIL.Image): The input image to edit. 
+        prompt (str): Text prompt describing the desired lighting conditions.
+        illumination_dropdown (str, optional): Lighting style. 
+            Defaults to 'custom' - lighting style will be determined by the prompt. 
+        direction_dropdown (str, optional): Lighting direction, can be one of [auto, left side, right side, top, top left, top right, bottom, front, back].
+            Defaults to 'auto', not specifying a specific direction. 
+        seed (int, optional): Random seed for reproducible results. Defaults to 42.
+            Range: 0 to MAX_SEED (2147483647).
+        randomize_seed (bool, optional): Whether to use a random seed instead of the provided seed.
+            Defaults to False.
+        true_guidance_scale (float, optional): Controls adherence to the prompt. Higher values = more adherence.
+            Defaults to 1.0. Range: 0.0-20.0.
+        num_inference_steps(int, optional): Number of inference steps. 
+            Defaults to 8. Range: 
+        rewrite_prompt(bool, optional): Whether or not to polish the edit prompt with Qwen/Qwen2.5-VL-72B-Instruct using HF inference. 
+            Defaults to True. Range: 1-28.
+        progress (gr.Progress, optional): Gradio progress tracker. Defaults to gr.Progress(track_tqdm=True).
+    
+    Returns:
+        tuple: A tuple containing:
+            - List[PIL.Image.Image]: a 2 element list, containting the input image and the edited image
+            - int: The seed used for generation (useful when randomize_seed=True)
+    
+    Raises:
+        gr.Error: If input_image is None (no image uploaded).
+    
+    Note:
+        - The function uses GPU acceleration via the @spaces.GPU decorator
     """
     # Hardcode the negative prompt as in the original
     negative_prompt = " "
@@ -412,7 +447,7 @@ def infer(
             num_images_per_prompt=1  # Always generate only 1 image
         ).images
         
-        # Return the first (and only) image
+        
         return [image,images[0]], seed
         
     except Exception as e:
@@ -579,4 +614,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(mcp_server=True)