Update app.py

app.py

@@ -1,17 +1,17 @@
-import gradio as gr
-import mss
+from fastapi import FastAPI, UploadFile, File
+from fastapi.responses import JSONResponse
 from PIL import Image
-import numpy as np
 import io
 
-def capture_screen():
-    with mss.mss() as sct:
-        monitor = sct.monitors[1]  # full screen
-        screenshot = sct.grab(monitor)
-        img = Image.frombytes("RGB", screenshot.size, screenshot.rgb)
-        return img
+app = FastAPI()
 
-demo = gr.Interface(fn=capture_screen, inputs=[], outputs="image")
-
-if __name__ == "__main__":
-    demo.launch()
+@app.post("/upload")
+async def upload(file: UploadFile = File(...)):
+    try:
+        contents = await file.read()
+        image = Image.open(io.BytesIO(contents))
+        # You can run your AI analysis here
+        print("Screenshot received")
+        return JSONResponse({"status": "success"})
+    except Exception as e:
+        return JSONResponse({"status": "error", "details": str(e)})