Update app.py

app.py

@@ -210,8 +210,6 @@ def navigate(input_numpy_image: np.ndarray, task: str) -> Tuple[str, Optional[Im
 
     prompt = get_navigation_prompt(task, input_pil_image)
 
-    print("Prompt:")
-    print(prompt)
 
     if model is None:
         raise ValueError("Model not loaded")
@@ -252,9 +250,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(f"<h1 style='text-align: center;'>{title}</h1>")
     # gr.Markdown(description)
 
+    with gr.Row():
+        input_image_component = gr.Image(label="UI Image", height=500)
     with gr.Row():
         with gr.Column():
-            input_image_component = gr.Image(label="Input UI Image", height=400)
             task_component = gr.Textbox(
                 label="task",
                 placeholder="e.g., Search for the name of the current UK Prime Minister.",
@@ -266,12 +265,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             localization_image_component = gr.Image(label="Action Localization", height=400)
             output_coords_component = gr.Textbox(label="Agent Output", lines=10)
 
-    submit_button.click(navigate, [input_image_component, task_component], [output_coords_component, localization_image_component])
+    submit_button.click(navigate, [input_image_component, task_component], [output_coords_component, input_image_component])
 
     gr.Examples(
         examples=[[example_1_image, example_1_task], [example_2_image, example_2_task]],
         inputs=[input_image_component, task_component],
-        outputs=[output_coords_component, localization_image_component],
+        outputs=[output_coords_component, input_image_component],
         fn=navigate,
         cache_examples=True,
     )