Update app.py

app.py

@@ -2,192 +2,124 @@ import streamlit as st
 import os
 import time
 import base64
-import threading
-import requests
-import json
-from flask import Flask, request, jsonify
+from datetime import datetime
 from playwright.sync_api import sync_playwright
 
-# --- PART 1: BACKGROUND FLASK SERVER (THE ENGINE) ---
-# Ye server browser ko sambhalega taake Streamlit crash na ho
-
-server = Flask(__name__)
-PORT = 5000
-API_URL = f"http://127.0.0.1:{PORT}"
-
-# Global Browser State
-backend_state = {
-    "playwright": None,
-    "browser": None,
-    "context": None,
-    "page": None,
-    "logs": []
-}
-
-def add_log(msg):
-    backend_state["logs"].append(msg)
-    if len(backend_state["logs"]) > 20: backend_state["logs"].pop(0)
-
-@server.route('/start', methods=['POST'])
-def start_browser():
-    s = backend_state
-    if s["browser"]: return jsonify({"status": "Already running"})
-    
-    try:
-        s["playwright"] = sync_playwright().start()
-        s["browser"] = s["playwright"].chromium.launch(
-            headless=True,
-            args=['--no-sandbox', '--disable-blink-features=AutomationControlled']
-        )
-        s["context"] = s["browser"].new_context(
-            viewport={'width': 375, 'height': 812},
-            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
-        )
-        s["page"] = s["context"].new_page()
-        s["page"].goto("https://www.coinpayu.com/login", timeout=60000)
-        add_log("Browser Started. On Login Page.")
-        return jsonify({"status": "Started"})
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
-
-@server.route('/interact', methods=['POST'])
-def interact():
-    s = backend_state
-    if not s["page"]: return jsonify({"error": "No Browser"}), 400
-    
-    data = request.json
-    action = data.get("action")
-    
-    try:
-        if action == "type":
-            # Clear field if needed? No, just type.
-            text = data.get("text")
-            s["page"].keyboard.type(text)
-            add_log(f"Typed: {text}")
-            
-        elif action == "enter":
-            s["page"].keyboard.press("Enter")
-            add_log("Pressed Enter")
-            
-        elif action == "click":
-            # Tap center of screen if no coordinates (generic tap)
-            # But better to tap active element or just rely on 'Type' focus
-            pass
-            
-        elif action == "scroll":
-            s["page"].evaluate("window.scrollBy(0, 300)")
-            
-        # Capture Screenshot immediately
-        b64 = base64.b64encode(s["page"].screenshot(type='jpeg', quality=50)).decode('utf-8')
-        return jsonify({"status": "ok", "image": b64, "logs": s["logs"]})
+# --- 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
         
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
-
-@server.route('/status', methods=['GET'])
-def status():
-    s = backend_state
-    img = ""
-    if s["page"]:
         try:
-            img = base64.b64encode(s["page"].screenshot(type='jpeg', quality=50)).decode('utf-8')
-        except: pass
-    return jsonify({"image": img, "logs": s["logs"]})
-
-# Run Flask in Background Thread
-def run_flask():
-    server.run(host='0.0.0.0', port=PORT, use_reloader=False)
-
-if "server_thread" not in st.session_state:
-    t = threading.Thread(target=run_flask, daemon=True)
-    t.start()
-    st.session_state.server_thread = True
-    time.sleep(2) # Wait for server to start
-
-# --- PART 2: STREAMLIT UI (THE FRONTEND) ---
-
-st.set_page_config(layout="wide", page_title="CoinPayU Controller")
-
-st.title("📱 CoinPayU Remote")
-
-# Initialize Session State
-if "image" not in st.session_state: st.session_state.image = None
-if "logs" not in st.session_state: st.session_state.logs = []
+            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
 
-# --- API HELPER ---
-def api_call(endpoint, data=None):
-    try:
-        if data:
-            r = requests.post(f"{API_URL}/{endpoint}", json=data)
-        else:
-            r = requests.get(f"{API_URL}/{endpoint}")
-        
-        if r.status_code == 200:
-            res = r.json()
-            if "image" in res and res["image"]:
-                st.session_state.image = res["image"]
-            if "logs" in res:
-                st.session_state.logs = res["logs"]
-            return True
-    except:
-        st.error("Backend Sleeping. Click Start.")
-    return False
-
-# --- LAYOUT ---
-col1, col2 = st.columns([1, 1])
+    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:
-    st.subheader("Live View")
-    
-    # Auto Refresh Logic
-    if st.button("🔄 Refresh Screen"):
-        api_call("status")
-        
-    if st.session_state.image:
-        st.image(base64.b64decode(st.session_state.image), caption="Live Browser", use_column_width=True)
-    else:
-        st.info("Browser OFF. Click 'Start Engine'.")
+    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:
-    st.subheader("Controls")
-    
-    if st.button("🚀 Start Engine"):
-        with st.spinner("Starting Browser..."):
-            requests.post(f"{API_URL}/start")
-            time.sleep(3)
-            api_call("status")
-            st.rerun()
-
-    # --- TYPING FIX ---
-    st.write("### ⌨️ Keyboard")
-    # Form use karne se Enter dabane par submit hota hai
-    with st.form(key='type_form'):
-        user_text = st.text_input("Type Email / Password here:", key="input_text")
-        c1, c2 = st.columns(2)
-        with c1:
-            submit_type = st.form_submit_button("Type Text")
-        with c2:
-            submit_enter = st.form_submit_button("Press Enter")
-            
-        if submit_type and user_text:
-            api_call("interact", {"action": "type", "text": user_text})
-            st.rerun()
-            
-        if submit_enter:
-            api_call("interact", {"action": "enter"})
-            st.rerun()
-
-    st.write("### 🖱️ Navigation")
-    c3, c4 = st.columns(2)
-    if c3.button("⬇ Scroll Down"):
-        api_call("interact", {"action": "scroll"})
+    if st.button("⬇ Scroll Down", use_container_width=True):
+        bot.interact("scroll")
+        st.rerun()
+        
+    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")
     
-    if c4.button("Login Button (Enter)"):
-        api_call("interact", {"action": "enter"})
+    if submitted and user_text:
+        bot.interact("type", user_text)
         st.rerun()
 
-    st.markdown("---")
-    st.write("Logs:")
-    st.code("\n".join(st.session_state.logs))
+# 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()