Update app.py

app.py

@@ -1,5 +1,9 @@
 import os
 import time
+
+# --- CRITICAL FIX: Set Display before importing PyAutoGUI ---
+os.environ["DISPLAY"] = ":99"
+
 import base64
 import threading
 from flask import Flask, request, jsonify, send_file
@@ -12,42 +16,34 @@ import io
 app = Flask(__name__)
 CORS(app)
 
-# --- DISPLAY SETUP ---
-os.environ["DISPLAY"] = ":99"
 SCREEN_WIDTH = 1024
 SCREEN_HEIGHT = 768
 
-# Optimization: PyAutoGUI speed up
+# PyAutoGUI Settings
 pyautogui.FAILSAFE = False
-pyautogui.PAUSE = 0.05
+pyautogui.PAUSE = 0.1
 
-# Screenshot Manager (MSS is faster than Scrot)
 sct = mss.mss()
 monitor = {"top": 0, "left": 0, "width": SCREEN_WIDTH, "height": SCREEN_HEIGHT}
 
-# --- ROUTES ---
-
 @app.route('/')
 def home():
-    return "Cloud PC Running. Use HTML Client."
+    return "Cloud PC Ready. Connect via HTML."
 
 @app.route('/snapshot', methods=['GET'])
 def get_snapshot():
     try:
-        # Capture full desktop
         sct_img = sct.grab(monitor)
         img = Image.frombytes("RGB", sct_img.size, sct_img.bgra, "raw", "BGRX")
+        img.thumbnail((600, 600)) # Resize for speed
         
-        # Resize for Mobile Speed (Width 400px is enough for control)
-        img.thumbnail((500, 500)) 
-        
-        # Compress to JPEG
         buffer = io.BytesIO()
-        img.save(buffer, format="JPEG", quality=40) # Low quality for speed
+        img.save(buffer, format="JPEG", quality=50)
         b64 = base64.b64encode(buffer.getvalue()).decode('utf-8')
         
         return jsonify({"screenshot": b64})
     except Exception as e:
+        print(f"Snapshot Error: {e}")
         return jsonify({"error": str(e)}), 500
 
 @app.route('/interact', methods=['POST'])
@@ -57,44 +53,30 @@ def interact():
     
     try:
         if action == 'tap':
-            # Map mobile coordinates to PC resolution
             x = int(data.get('x') * SCREEN_WIDTH)
             y = int(data.get('y') * SCREEN_HEIGHT)
             pyautogui.click(x, y)
             
         elif action == 'type':
-            text = data.get('text')
-            pyautogui.write(text)
+            pyautogui.write(data.get('text'))
             
         elif action == 'key':
-            key = data.get('key')
-            pyautogui.press(key)
+            k = data.get('key').lower()
+            if k == 'enter': pyautogui.press('enter')
+            elif k == 'backspace': pyautogui.press('backspace')
             
         elif action == 'scroll':
-            pyautogui.scroll(-5) # Linux scroll down
+            pyautogui.scroll(-5)
             
         elif action == 'open_terminal':
-            # Shortcut to open terminal in XFCE
-            pyautogui.hotkey('ctrl', 'alt', 't')
+            os.system("xfce4-terminal &")
 
         elif action == 'open_browser':
-            # Command to run chromium
             os.system("chromium-browser --no-sandbox &")
 
         return jsonify({"status": "ok"})
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-# --- FILE TRANSFER ---
-# Agar aap Antigravity se App banate hain, to wo '/app/output.apk' mein save karein
-@app.route('/download', methods=['GET'])
-def download_app():
-    # Example path (Change logic later based on where Antigravity saves files)
-    file_path = "/app/output.apk"
-    if os.path.exists(file_path):
-        return send_file(file_path, as_attachment=True)
-    else:
-        return "File not found yet. Build the app first!", 404
-
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, threaded=True)