Update app.py

app.py

@@ -106,8 +106,9 @@ def on_select(space):
 
 def update_screenshot(url, last_url):
     if url and url != last_url:
-        return take_screenshot(url), url
-    return None, last_url
+        screenshot = take_screenshot(url)
+        return screenshot, url
+    return gr.update(), last_url
         
 def take_screenshot(url):
     try:
@@ -150,6 +151,7 @@ def create_ui():
                     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_trigger = gr.Number(value=0, visible=False)
 
                 for _, button, space in space_rows:
                     button.click(
@@ -158,13 +160,17 @@ def create_ui():
                         outputs=[info_output, app_py_content, url_state]
                     )
 
-                demo.add_periodic_callback(
-                    lambda: update_screenshot(url_state.value, last_url_state.value), 
-                    5,
+                def increment(x):
+                    return (x + 1) % 1000000
+
+                update_trigger.change(
+                    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)
+
         return demo
 
     except Exception as e: