Update app.py

app.py

@@ -311,19 +311,6 @@ def create_ui():
                     
                     update_trigger = gr.Button("Update Screenshot", visible=False)
 
-            # JavaScript to handle space selection
-            demo.load(None, None, None, _js="""
-            function() {
-                document.addEventListener('click', function(e) {
-                    if (e.target.classList.contains('select-space')) {
-                        var spaceData = JSON.stringify(JSON.parse(e.target.getAttribute('data-space')));
-                        document.querySelector('#space-data').value = spaceData;
-                        document.querySelector('#on-select-function').click();
-                    }
-                });
-            }
-            """)
-
             # Hidden elements to pass data
             space_data = gr.Textbox(elem_id="space-data", visible=False)
             on_select_trigger = gr.Button("Hidden Select Trigger", visible=False, elem_id="on-select-function")
@@ -379,6 +366,19 @@ def create_ui():
                 outputs=[screenshot_output, last_url_state]
             )
 
+        # JavaScript 코드를 추가하기 위해 js 함수를 사용합니다.
+        demo.load(js="""
+        function() {
+            document.addEventListener('click', function(e) {
+                if (e.target.classList.contains('select-space')) {
+                    var spaceData = JSON.stringify(JSON.parse(e.target.getAttribute('data-space')));
+                    document.querySelector('#space-data').value = spaceData;
+                    document.querySelector('#on-select-function').click();
+                }
+            });
+        }
+        """)
+
         # Start a background thread to trigger updates
         def trigger_updates():
             while True:
@@ -400,4 +400,4 @@ if __name__ == "__main__":
         demo.launch()
     except Exception as e:
         print(f"Error in main: {str(e)}")
-        print(traceback.format_exc())                                                        
+        print(traceback.format_exc())