updatE

app.py

@@ -115,9 +115,9 @@ if __name__ == '__main__':
             fn=handle_image_selection, inputs=[image_selector, genie], outputs=image_display, show_progress='hidden'
         )
 
-        up.click(fn=lambda state: handle_input("up", state), inputs=[genie], outputs=image_display, show_progress='hidden')
-        down.click(fn=lambda state: handle_input("down", state), inputs=[genie], outputs=image_display, show_progress='hidden')
-        left.click(fn=lambda state: handle_input("left", state), inputs=[genie], outputs=image_display, show_progress='hidden')
-        right.click(fn=lambda state: handle_input("right", state), inputs=[genie], outputs=image_display, show_progress='hidden')
+        up.click(fn=lambda genie: handle_input("up", genie), inputs=[genie], outputs=image_display, show_progress='hidden')
+        down.click(fn=lambda genie: handle_input("down", genie), inputs=[genie], outputs=image_display, show_progress='hidden')
+        left.click(fn=lambda genie: handle_input("left", genie), inputs=[genie], outputs=image_display, show_progress='hidden')
+        right.click(fn=lambda genie: handle_input("right", genie), inputs=[genie], outputs=image_display, show_progress='hidden')
 
     demo.launch(share=True)