Update app.py

app.py

@@ -8,6 +8,8 @@ from PIL import Image
 from io import BytesIO
 import asyncio
 from gradio_client import Client
+import time
+import threading
 
 HF_TOKEN = os.getenv("HF_TOKEN")
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=HF_TOKEN)
@@ -119,6 +121,8 @@ def take_screenshot(url):
         print(f"Screenshot error: {str(e)}")
         return Image.new('RGB', (1080, 720), color='lightgray')
             
+
+
 def create_ui():
     try:
         spaces_list = get_most_liked_spaces()
@@ -151,7 +155,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)
+                    update_trigger = gr.Button("Update Screenshot", visible=False)
 
                 for _, button, space in space_rows:
                     button.click(
@@ -164,16 +168,19 @@ def create_ui():
                         outputs=[screenshot_output, last_url_state]
                     )
 
-                def increment(x):
-                    return (x + 1) % 1000000
-
-                update_trigger.change(
+                update_trigger.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)
+        # Start a background thread to trigger updates
+        def trigger_updates():
+            while True:
+                time.sleep(5)
+                update_trigger.click()
+
+        threading.Thread(target=trigger_updates, daemon=True).start()
 
         return demo