Update app.py

app.py

@@ -132,21 +132,6 @@ def create_ui():
         .space-row {margin-bottom: 5px !important;}
         """
 
-        js = """
-        function updateScreenshot() {
-            const urlState = document.querySelector('#url_state textarea');
-            const lastUrlState = document.querySelector('#last_url_state textarea');
-            if (urlState && lastUrlState && urlState.value !== lastUrlState.value) {
-                const updateButton = document.querySelector('#update_screenshot_button');
-                if (updateButton) {
-                    updateButton.click();
-                }
-            }
-            setTimeout(updateScreenshot, 5000);
-        }
-        setTimeout(updateScreenshot, 5000);
-        """
-
         with gr.Blocks(css=css, theme="Nymbo/Nymbo_Theme") as demo:
             gr.Markdown("# 300: HuggingFace Most Liked Spaces")
             
@@ -162,27 +147,22 @@ def create_ui():
 
                 with gr.Column(scale=1):
                     info_output = gr.Textbox(label="Space 정보 및 요약", lines=20)
-                    url_state = gr.Textbox(visible=False, elem_id="url_state")
-                    last_url_state = gr.Textbox(visible=False, elem_id="last_url_state")
+                    url_state = gr.State("")
+                    last_url_state = gr.State("")
                     screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=600)
                     app_py_content = gr.Code(language="python", label="메인 소스코드")
-                    update_screenshot_button = gr.Button("Update Screenshot", visible=False, elem_id="update_screenshot_button")
 
                 for _, button, space in space_rows:
                     button.click(
                         lambda s=space: on_select(s),
                         inputs=[],
                         outputs=[info_output, app_py_content, url_state]
+                    ).then(
+                        update_screenshot,
+                        inputs=[url_state, last_url_state],
+                        outputs=[screenshot_output, last_url_state]
                     )
 
-                update_screenshot_button.click(
-                    update_screenshot,
-                    inputs=[url_state, last_url_state],
-                    outputs=[screenshot_output, last_url_state]
-                )
-
-            demo.load(None, None, None, _js=js)
-
         return demo
 
     except Exception as e: