Update app.py

app.py

@@ -1,28 +1,17 @@
 import gradio as gr
 import mss
+from PIL import Image
 import numpy as np
-import PIL.Image
-import threading
-import time
+import io
 
-latest_frame = None
-
-def capture_loop():
-    global latest_frame
+def capture_screen():
     with mss.mss() as sct:
-        monitor = sct.monitors[1]
-        while True:
-            screenshot = sct.grab(monitor)
-            img = PIL.Image.frombytes("RGB", screenshot.size, screenshot.rgb)
-            latest_frame = img
-            time.sleep(0.2)  # ~5 times per second
-
-def get_latest_frame():
-    return latest_frame
+        monitor = sct.monitors[1]  # full screen
+        screenshot = sct.grab(monitor)
+        img = Image.frombytes("RGB", screenshot.size, screenshot.rgb)
+        return img
 
-# Start the capture loop in the background
-threading.Thread(target=capture_loop, daemon=True).start()
+demo = gr.Interface(fn=capture_screen, inputs=[], outputs="image")
 
-# UI
-demo = gr.Interface(fn=get_latest_frame, inputs=[], outputs="image", live=True)
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()