Update Gradio app with multiple files

app.py

@@ -46,6 +46,7 @@ MAX_SEED = np.iinfo(np.int32).max
 @spaces.GPU
 def convert_to_anime(
     image,
+    light_direction,
     seed,
     randomize_seed,
     true_guidance_scale,
@@ -54,7 +55,7 @@ def convert_to_anime(
     width,
     progress=gr.Progress(track_tqdm=True)
 ):
-    prompt = "Relight Figure 1 using the luminance map from Figure 2 (light source from the front)"
+    prompt = f"Relight Figure 1 using the luminance map from Figure 2 (light source from the {light_direction})"
     
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -197,6 +198,13 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                     elem_classes="image-container"
                 )
                 
+                light_direction = gr.Dropdown(
+                    choices=["Front", "Left Front", "Left", "Left Rear", "Rear", "Right Rear", "Right", "Right Front", "Above", "Below"],
+                    value="Front",
+                    label="Light Source Direction",
+                    info="Select the direction of the light source"
+                )
+                
                 with gr.Accordion("⚙️ Advanced Settings", open=False):
                     seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
                     randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
@@ -215,7 +223,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                 )
     
     inputs = [
-        image, seed, randomize_seed, true_guidance_scale, 
+        image, light_direction, seed, randomize_seed, true_guidance_scale, 
         num_inference_steps, height, width
     ]
     outputs = [result, seed]