Update app.py

app.py

@@ -132,6 +132,21 @@ 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")
             
@@ -147,11 +162,11 @@ def create_ui():
 
                 with gr.Column(scale=1):
                     info_output = gr.Textbox(label="Space 정보 및 요약", lines=20)
-                    url_state = gr.State("")
-                    last_url_state = gr.State("")
+                    url_state = gr.Textbox(visible=False, elem_id="url_state")
+                    last_url_state = gr.Textbox(visible=False, elem_id="last_url_state")
                     screenshot_output = gr.Image(type="pil", label="Live 화면", height=360, width=600)
                     app_py_content = gr.Code(language="python", label="메인 소스코드")
-                    update_trigger = gr.Number(value=0, visible=False)
+                    update_screenshot_button = gr.Button("Update Screenshot", visible=False, elem_id="update_screenshot_button")
 
                 for _, button, space in space_rows:
                     button.click(
@@ -160,16 +175,13 @@ def create_ui():
                         outputs=[info_output, app_py_content, url_state]
                     )
 
-                def increment(x):
-                    return (x + 1) % 1000000
-
-                update_trigger.change(
+                update_screenshot_button.click(
                     update_screenshot,
                     inputs=[url_state, last_url_state],
                     outputs=[screenshot_output, last_url_state]
                 )
 
-                demo.load(increment, inputs=update_trigger, outputs=update_trigger, every=5)
+            demo.load(None, None, None, _js=js)
 
         return demo