Update app.py

app.py

@@ -4,6 +4,7 @@ from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.common.by import By
 from bs4 import BeautifulSoup
 import time
+import tempfile
 
 app = Flask(__name__)
 
@@ -13,6 +14,12 @@ def get_hidden_code(url):
     chrome_options.add_argument("--headless=new")
     chrome_options.add_argument("--disable-gpu")
     chrome_options.add_argument("--no-sandbox")
+    chrome_options.add_argument("--disable-dev-shm-usage")
+
+    # Force Selenium to use /tmp (writable in Hugging Face)
+    chrome_options.add_argument(f"--user-data-dir={tempfile.mkdtemp()}")
+    chrome_options.add_argument(f"--disk-cache-dir={tempfile.mkdtemp()}")
+
     driver = webdriver.Chrome(options=chrome_options)
 
     token_found = []
@@ -21,7 +28,7 @@ def get_hidden_code(url):
         driver.get(url)
         time.sleep(3)  # wait for page load
 
-        # Click the Start Challenge button if present
+        # Try clicking Start Challenge button
         try:
             start_button = driver.find_element(By.XPATH, "//button[contains(text(), 'Start Challenge')]")
             start_button.click()
@@ -29,21 +36,21 @@ def get_hidden_code(url):
         except Exception as e:
             print("⚠️ Could not click Start Challenge button:", e)
 
-        # Get updated HTML after interaction
+        # Get updated HTML after click
         html = driver.page_source
         soup = BeautifulSoup(html, "html.parser")
 
-        # Look for hidden elements
+        # Find 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))
 
-        # Look for hidden inputs
+        # Find hidden input fields
         for hidden in soup.find_all("input", {"type": "hidden"}):
             if hidden.get("value"):
                 token_found.append(hidden["value"])
 
-        # Try regex / text search for "Hidden Code"
+        # Look for "Hidden Code" text
         if "hidden code" in html.lower():
             start = html.lower().find("hidden code")
             token_found.append(html[start:start+100])
@@ -68,6 +75,13 @@ def mission():
 
     tokens = get_hidden_code(url)
 
+    # Print in logs for debugging
+    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,
@@ -77,6 +91,5 @@ def mission():
 
 if __name__ == "__main__":
     import os
-    port = int(os.environ.get("PORT", 7860))  # HF uses 7860
+    port = int(os.environ.get("PORT", 7860))
     app.run(host="0.0.0.0", port=port, debug=False)
-