Update app.py

app.py

@@ -1,85 +1,21 @@
-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
+FROM python:3.10-slim
 
-app = Flask(__name__)
+RUN apt-get update && apt-get install -y \
+    wget gnupg unzip curl \
+    chromium chromium-driver \
+    fonts-liberation \
+    && rm -rf /var/lib/apt/lists/*
 
-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")
+ENV CHROME_BIN=/usr/bin/chromium
+ENV CHROMEDRIVER=/usr/bin/chromedriver
 
-    driver = webdriver.Chrome(options=chrome_options)
+COPY requirements.txt /app/requirements.txt
+WORKDIR /app
+RUN pip install --no-cache-dir -r requirements.txt
 
-    token_found = []
+COPY app.py /app/app.py
 
-    try:
-        driver.get(url)
-        time.sleep(3)
+EXPOSE 7860
 
-        # 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)
+# Run with 1 worker for Selenium stability
+CMD ["gunicorn", "-w", "1", "-b", "0.0.0.0:7860", "app:app"]