Update app.py

app.py

@@ -1,21 +1,85 @@
-FROM python:3.10-slim
+from flask import Flask, request, jsonify
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common.by import By
+from bs4 import BeautifulSoup
+import time
 
-RUN apt-get update && apt-get install -y \
-    wget gnupg unzip curl \
-    chromium chromium-driver \
-    fonts-liberation \
-    && rm -rf /var/lib/apt/lists/*
+app = Flask(__name__)
 
-ENV CHROME_BIN=/usr/bin/chromium
-ENV CHROMEDRIVER=/usr/bin/chromedriver
+def get_hidden_code(url):
+    chrome_options = Options()
+    chrome_options.add_argument("--headless=new")
+    chrome_options.add_argument("--no-sandbox")
+    chrome_options.add_argument("--disable-dev-shm-usage")
+    chrome_options.add_argument("--disable-gpu")
+    chrome_options.add_argument("--disable-software-rasterizer")
+    chrome_options.add_argument("--incognito")
+    chrome_options.add_argument("--remote-debugging-port=9222")
 
-COPY requirements.txt /app/requirements.txt
-WORKDIR /app
-RUN pip install --no-cache-dir -r requirements.txt
+    driver = webdriver.Chrome(options=chrome_options)
 
-COPY app.py /app/app.py
+    token_found = []
 
-EXPOSE 7860
+    try:
+        driver.get(url)
+        time.sleep(3)
 
-# Run with 1 worker for Selenium stability
-CMD ["gunicorn", "-w", "1", "-b", "0.0.0.0:7860", "app:app"]
+        # Try clicking the button
+        try:
+            start_button = driver.find_element(By.XPATH, "//button[contains(text(), 'Start Challenge')]")
+            start_button.click()
+            time.sleep(3)
+        except Exception as e:
+            print("⚠️ Could not click Start Challenge button:", e)
+
+        html = driver.page_source
+        soup = BeautifulSoup(html, "html.parser")
+
+        # Scrape hidden elements
+        for tag in soup.find_all(True):
+            if tag.has_attr("style") and "display:none" in tag["style"]:
+                token_found.append(tag.get_text(strip=True))
+
+        for hidden in soup.find_all("input", {"type": "hidden"}):
+            if hidden.get("value"):
+                token_found.append(hidden["value"])
+
+        if "hidden code" in html.lower():
+            start = html.lower().find("hidden code")
+            token_found.append(html[start:start+100])
+
+        driver.quit()
+        return list(set(token_found)) or ["No hidden code found"]
+
+    except Exception as e:
+        driver.quit()
+        return [f"Error: {str(e)}"]
+
+@app.route("/mission", methods=["POST"])
+def mission():
+    data = request.json
+    url = data.get("url")
+    questions = data.get("questions", [])
+
+    if not url:
+        return jsonify({"error": "URL is required"}), 400
+
+    tokens = get_hidden_code(url)
+
+    print("\n===== Mission Request =====")
+    print("URL received:", url)
+    print("Questions received:", questions)
+    print("Token(s) found:", tokens)
+    print("===========================\n")
+
+    return jsonify({
+        "url_received": url,
+        "questions_received": questions,
+        "token_found": tokens
+    })
+
+if __name__ == "__main__":
+    import os
+    port = int(os.environ.get("PORT", 7860))
+    app.run(host="0.0.0.0", port=port, debug=False)