Create app.py

app.py

@@ -0,0 +1,135 @@
+from flask import Flask, request, jsonify, send_from_directory
+import undetected_chromedriver as uc
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+import os
+import time
+import base64
+from datetime import datetime
+
+app = Flask(__name__)
+PUBLIC_DIR = 'public'
+os.makedirs(PUBLIC_DIR, exist_ok=True)
+
+def get_driver():
+    options = uc.ChromeOptions()
+    options.add_argument('--headless')
+    options.add_argument('--no-sandbox')
+    options.add_argument('--disable-dev-shm-usage')
+    options.add_argument('--disable-gpu')
+    return uc.Chrome(options=options, use_subprocess=False)
+
+@app.route('/')
+def index():
+    return jsonify({
+        'message': 'Undetected Chrome Scraper API is running',
+        'endpoints': {
+            'POST /api/scrape': 'Get HTML content from URL',
+            'POST /api/execute': 'Execute Python code with Selenium'
+        }
+    })
+
+@app.route('/api/scrape', methods=['POST'])
+def scrape():
+    data = request.get_json()
+    url = data.get('url')
+    wait_time = data.get('wait', 3)
+    screenshot = data.get('screenshot', False)
+    
+    if not url:
+        return jsonify({'success': False, 'error': 'URL is required'}), 400
+    
+    driver = None
+    try:
+        driver = get_driver()
+        driver.get(url)
+        time.sleep(wait_time)
+        
+        html = driver.page_source
+        title = driver.title
+        current_url = driver.current_url
+        
+        result = {
+            'success': True,
+            'data': {
+                'html': html,
+                'title': title,
+                'url': current_url,
+                'timestamp': datetime.now().isoformat()
+            }
+        }
+        
+        if screenshot:
+            filename = f'screenshot-{int(time.time())}.png'
+            filepath = os.path.join(PUBLIC_DIR, filename)
+            driver.save_screenshot(filepath)
+            result['data']['screenshot'] = f"{request.host_url}files/{filename}"
+        
+        return jsonify(result)
+        
+    except Exception as e:
+        return jsonify({'success': False, 'error': str(e)}), 500
+    finally:
+        if driver:
+            driver.quit()
+
+@app.route('/api/execute', methods=['POST'])
+def execute():
+    data = request.get_json()
+    code = data.get('code')
+    
+    if not code:
+        return jsonify({'success': False, 'error': 'Code is required'}), 400
+    
+    driver = None
+    try:
+        driver = get_driver()
+        
+        local_vars = {
+            'driver': driver,
+            'By': By,
+            'WebDriverWait': WebDriverWait,
+            'EC': EC,
+            'time': time,
+            'PUBLIC_DIR': PUBLIC_DIR,
+            'os': os
+        }
+        
+        exec(code, {}, local_vars)
+        
+        result = local_vars.get('result', None)
+        
+        files = []
+        if os.path.exists(PUBLIC_DIR):
+            for f in os.listdir(PUBLIC_DIR):
+                if f.startswith('screenshot-') and f.endswith('.png'):
+                    files.append({
+                        'name': f,
+                        'publicURL': f"{request.host_url}files/{f}"
+                    })
+        
+        return jsonify({
+            'success': True,
+            'data': {
+                'result': result,
+                'files': files,
+                'timestamp': int(time.time())
+            }
+        })
+        
+    except Exception as e:
+        return jsonify({
+            'success': False,
+            'error': str(e)
+        }), 500
+    finally:
+        if driver:
+            driver.quit()
+
+@app.route('/files/<path:filename>')
+def serve_file(filename):
+    return send_from_directory(PUBLIC_DIR, filename)
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=int(os.environ.get('PORT', 7860)))