Update app.py

app.py

@@ -8,6 +8,7 @@ import base64
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 import pyautogui
+import mss
 from PIL import Image
 import io
 
@@ -17,32 +18,38 @@ CORS(app)
 SCREEN_WIDTH = 1024
 SCREEN_HEIGHT = 768
 
-# Fail-safes disable karo taake mouse corner mein jane se app band na ho
+# Fail-safes disable
 pyautogui.FAILSAFE = False
 
+# MSS (Fast Screenshot) Setup
+sct = mss.mss()
+# Monitor 1 ko pakdo
+monitor = sct.monitors[1] 
+
 @app.route('/')
 def home():
-    return "Cloud PC is Running! Use HTML Client."
+    return "Cloud PC Ready. Connect via HTML."
 
 @app.route('/snapshot', methods=['GET'])
 def get_snapshot():
     try:
-        # METHOD CHANGE: mss ki jagah pyautogui use kar rahe hain (More Stable)
-        # Yeh seedha screen capture karega
-        img = pyautogui.screenshot()
+        # MSS se screenshot lo (Yeh Scrot/DBus use nahi karta, direct memory read karta hai)
+        sct_img = sct.grab(monitor)
+        
+        # Raw bytes ko Image mein convert karo
+        img = Image.frombytes("RGB", sct_img.size, sct_img.bgra, "raw", "BGRX")
         
-        # Resize kar rahe hain taake mobile par tez chale
-        img.thumbnail((600, 600)) 
+        # Resize for speed (Mobile data bachane ke liye)
+        img.thumbnail((600, 600))
         
         buffer = io.BytesIO()
-        img.save(buffer, format="JPEG", quality=50)
+        img.save(buffer, format="JPEG", quality=60)
         b64 = base64.b64encode(buffer.getvalue()).decode('utf-8')
         
         return jsonify({"screenshot": b64})
     except Exception as e:
-        # Error console mein print hoga
-        print(f"ERROR: {e}")
-        # Agar error aye to client ko batao (Black screen ki jagah error dikhega)
+        # AB AGAR ERROR AAYA TOH LOGS MEIN DIKHEGA
+        print(f"!!! SNAPSHOT ERROR !!!: {e}")
         return jsonify({"error": str(e)}), 500
 
 @app.route('/interact', methods=['POST'])
@@ -75,7 +82,9 @@ def interact():
 
         return jsonify({"status": "ok"})
     except Exception as e:
+        print(f"INTERACT ERROR: {e}")
         return jsonify({"error": str(e)}), 500
 
 if __name__ == '__main__':
+    # Threaded=False kabhi kabhi crash rokta hai
     app.run(host='0.0.0.0', port=7860, threaded=True)