Update app.py

app.py

@@ -1,125 +1,100 @@
-import streamlit as st
 import os
 import time
 import base64
-from datetime import datetime
-from playwright.sync_api import sync_playwright
-
-# --- GLOBAL BOT CLASS (CACHED) ---
-# Ye decorator browser ko restart hone se rokta hai
-@st.cache_resource
-class BrowserBot:
-    def __init__(self):
-        self.playwright = None
-        self.browser = None
-        self.context = None
-        self.page = None
-        self.logs = []
-        self.add_log("Bot initialized.")
-
-    def add_log(self, msg):
-        ts = datetime.now().strftime("%H:%M:%S")
-        self.logs.insert(0, f"[{ts}] {msg}")
-        if len(self.logs) > 20: self.logs.pop()
-
-    def start(self):
-        if self.browser: return # Already running
+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)
+
+# --- DISPLAY SETUP ---
+os.environ["DISPLAY"] = ":99"
+SCREEN_WIDTH = 1024
+SCREEN_HEIGHT = 768
+
+# Optimization: PyAutoGUI speed up
+pyautogui.FAILSAFE = False
+pyautogui.PAUSE = 0.05
+
+# 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."
+
+@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")
         
-        try:
-            self.playwright = sync_playwright().start()
-            self.browser = self.playwright.chromium.launch(
-                headless=True,
-                args=['--no-sandbox', '--disable-blink-features=AutomationControlled']
-            )
-            self.context = self.browser.new_context(
-                viewport={'width': 375, 'height': 812}, # Mobile View
-                user_agent='Mozilla/5.0 (iPhone; CPU iPhone OS 14_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/14.0 Mobile/15E148 Safari/604.1',
-                is_mobile=True, has_touch=True
-            )
-            self.page = self.context.new_page()
-            self.page.goto("https://www.coinpayu.com/login", timeout=60000)
-            self.add_log("Browser Started. On Login Page.")
-        except Exception as e:
-            self.add_log(f"Start Error: {e}")
-
-    def get_screenshot(self):
-        if not self.page: return None
-        try:
-            return self.page.screenshot(type='jpeg', quality=50)
-        except: return None
-
-    def interact(self, action, text=None):
-        if not self.page: return
-        try:
-            if action == "type":
-                self.page.keyboard.type(text)
-                self.add_log(f"Typed: {text}")
-            elif action == "enter":
-                self.page.keyboard.press("Enter")
-                self.add_log("Pressed Enter")
-            elif action == "scroll":
-                self.page.mouse.wheel(0, 300)
-                self.add_log("Scrolled Down")
-            elif action == "refresh":
-                self.page.reload()
-                self.add_log("Page Refreshed")
-        except Exception as e:
-            self.add_log(f"Error: {e}")
-
-# --- INITIALIZE ---
-bot = BrowserBot()
-
-# --- UI LAYOUT ---
-st.set_page_config(layout="centered", page_title="CoinPayU Bot")
-
-st.title("📱 CoinPayU Manager")
-
-# 1. LIVE VIEW
-st.subheader("Live Screen")
-screenshot = bot.get_screenshot()
-
-if screenshot:
-    st.image(screenshot, caption="Live Browser Feed", use_column_width=True)
-else:
-    st.info("Browser is OFF. Click 'Start Browser' below.")
-
-# 2. CONTROLS
-col1, col2 = st.columns(2)
-
-with col1:
-    if st.button("🚀 Start Browser", use_container_width=True):
-        bot.start()
-        st.rerun()
-
-    if st.button("🔄 Refresh Page", use_container_width=True):
-        bot.interact("refresh")
-        st.rerun()
-
-with col2:
-    if st.button("⬇ Scroll Down", use_container_width=True):
-        bot.interact("scroll")
-        st.rerun()
+        # Resize for Mobile Speed (Width 400px is enough for control)
+        img.thumbnail((500, 500)) 
         
-    if st.button("↵ Press Enter", use_container_width=True):
-        bot.interact("enter")
-        st.rerun()
-
-# 3. TYPING AREA
-st.markdown("### ⌨️ Keyboard")
-with st.form("type_form", clear_on_submit=True):
-    user_text = st.text_input("Enter Email / Password here:")
-    submitted = st.form_submit_button("Type Text")
+        # Compress to JPEG
+        buffer = io.BytesIO()
+        img.save(buffer, format="JPEG", quality=40) # Low quality for speed
+        b64 = base64.b64encode(buffer.getvalue()).decode('utf-8')
+        
+        return jsonify({"screenshot": b64})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+@app.route('/interact', methods=['POST'])
+def interact():
+    data = request.json
+    action = data.get('action')
     
-    if submitted and user_text:
-        bot.interact("type", user_text)
-        st.rerun()
-
-# 4. LOGS
-st.markdown("---")
-st.text("Logs:")
-st.code("\n".join(bot.logs))
-
-# Auto Refresh hack (keeps image live)
-if bot.page:
-    time.sleep(1)
-    st.rerun()
+    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)
+            
+        elif action == 'key':
+            key = data.get('key')
+            pyautogui.press(key)
+            
+        elif action == 'scroll':
+            pyautogui.scroll(-5) # Linux scroll down
+            
+        elif action == 'open_terminal':
+            # Shortcut to open terminal in XFCE
+            pyautogui.hotkey('ctrl', 'alt', 't')
+
+        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)