Spaces:

Akwbw
/

Botisbot

Sleeping

App Files Files Community

Akwbw commited on 29 days ago

Commit

0511a46

verified ·

1 Parent(s): 138c8a0

Update app.py

Browse files

Files changed (1) hide show

app.py +42 -74

app.py CHANGED Viewed

@@ -1,14 +1,17 @@
 import os
 import time
-# Display Setup
 os.environ["DISPLAY"] = ":99"
-os.environ["XAUTHORITY"] = "/root/.Xauthority"
 import base64
-import subprocess
-from flask import Flask, request, jsonify, Response
 from flask_cors import CORS
 import pyautogui
 app = Flask(__name__)
 CORS(app)
@@ -16,99 +19,64 @@ CORS(app)
 SCREEN_WIDTH = 1024
 SCREEN_HEIGHT = 768
-# PyAutoGUI Config
 pyautogui.FAILSAFE = False
 pyautogui.PAUSE = 0.1
-# --- EMBEDDED UI ---
-HTML_TEMPLATE = """
-<!DOCTYPE html>
-<html>
-<head>
-    <meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no">
-    <title>Cloud PC</title>
-    <style>
-        body { margin: 0; background: #111; height: 100vh; display: flex; flex-direction: column; overflow: hidden; user-select: none; }
-        .screen-container { flex: 1; display: flex; justify-content: center; background: black; }
-        #desktop { width: 100%; height: 100%; object-fit: contain; }
-        .toolbar { position: absolute; bottom: 20px; left: 50%; transform: translateX(-50%); background: rgba(255,255,255,0.2); padding: 10px; border-radius: 30px; display: flex; gap: 15px; }
-        .btn { width: 40px; height: 40px; border-radius: 50%; background: #333; color: white; display: flex; align-items: center; justify-content: center; cursor: pointer; }
-    </style>
-</head>
-<body>
-    <div class="screen-container">
-        <img id="desktop" draggable="false">
-        <div class="toolbar">
-            <div class="btn" onclick="sendKey('win')">⊞</div>
-            <div class="btn" onclick="sendKey('enter')">↵</div>
-            <div class="btn" onclick="launchApp('chrome')">🌐</div>
-            <div class="btn" onclick="launchApp('terminal')">_></div>
-        </div>
-    </div>
-    <script>
-        const img = document.getElementById('desktop');
-        function refresh() { img.src = '/snapshot?' + new Date().getTime(); }
-        setInterval(refresh, 800);
-        img.addEventListener('click', async (e) => {
-            const rect = img.getBoundingClientRect();
-            if (e.clientX < rect.left || e.clientX > rect.right) return;
-            const x = (e.clientX - rect.left) / rect.width;
-            const y = (e.clientY - rect.top) / rect.height;
-            await fetch('/interact', { method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify({action: 'tap', x: x, y: y}) });
-            refresh();
-        });
-        async function sendKey(k) {
-            await fetch('/interact', { method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify({action: 'key', key: k}) });
-            refresh();
-        }
-        async function launchApp(app) {
-            await fetch('/interact', { method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify({action: 'launch', app: app}) });
-        }
-    </script>
-</body>
-</html>
-"""
 @app.route('/')
-def home(): return HTML_TEMPLATE
 @app.route('/snapshot', methods=['GET'])
 def get_snapshot():
     try:
-        subprocess.run(["scrot", "/tmp/screen.jpg", "-q", "20", "-o"], check=True)
-        with open("/tmp/screen.jpg", "rb") as image_file:
-            return Response(image_file.read(), mimetype='image/jpeg')
-    except: return "Error", 500
 @app.route('/interact', methods=['POST'])
 def interact():
     data = request.json
     action = data.get('action')
     try:
         if action == 'tap':
             x = int(data.get('x') * SCREEN_WIDTH)
             y = int(data.get('y') * SCREEN_HEIGHT)
             pyautogui.click(x, y)
         elif action == 'key':
-            k = data.get('key')
-            if k == 'win': pyautogui.press('super')
-            else: pyautogui.press(k)
-        elif action == 'launch':
-            app_name = data.get('app')
-            if app_name == 'chrome':
-                # FIXED: Use google-chrome instead of chromium-browser
-                subprocess.Popen("google-chrome --no-sandbox --start-maximized &", shell=True)
-            elif app_name == 'terminal':
-                subprocess.Popen("xfce4-terminal &", shell=True)
         return jsonify({"status": "ok"})
-    except Exception as e:
-        print(e)
-        return jsonify({"error": "err"}), 500
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, threaded=True)

 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
 from flask_cors import CORS
 import pyautogui
+import mss
+from PIL import Image
+import io
 app = Flask(__name__)
 CORS(app)
 SCREEN_WIDTH = 1024
 SCREEN_HEIGHT = 768
+# PyAutoGUI Settings
 pyautogui.FAILSAFE = False
 pyautogui.PAUSE = 0.1
+sct = mss.mss()
+monitor = {"top": 0, "left": 0, "width": SCREEN_WIDTH, "height": SCREEN_HEIGHT}
 @app.route('/')
+def home():
+    return "Cloud PC Ready. Connect via HTML."
 @app.route('/snapshot', methods=['GET'])
 def get_snapshot():
     try:
+        sct_img = sct.grab(monitor)
+        img = Image.frombytes("RGB", sct_img.size, sct_img.bgra, "raw", "BGRX")
+        img.thumbnail((600, 600)) # Resize for speed
+        buffer = io.BytesIO()
+        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'])
 def interact():
     data = request.json
     action = data.get('action')
     try:
         if action == 'tap':
             x = int(data.get('x') * SCREEN_WIDTH)
             y = int(data.get('y') * SCREEN_HEIGHT)
             pyautogui.click(x, y)
+        elif action == 'type':
+            pyautogui.write(data.get('text'))
         elif action == 'key':
+            k = data.get('key').lower()
+            if k == 'enter': pyautogui.press('enter')
+            elif k == 'backspace': pyautogui.press('backspace')
+        elif action == 'scroll':
+            pyautogui.scroll(-5)
+        elif action == 'open_terminal':
+            os.system("xfce4-terminal &")
+        elif action == 'open_browser':
+            os.system("chromium-browser --no-sandbox &")
         return jsonify({"status": "ok"})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, threaded=True)