Update app.py

app.py

@@ -4,21 +4,18 @@ 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__)
 
 def get_hidden_code(url):
-    # Setup Selenium headless Chrome
     chrome_options = Options()
     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()}")
+    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")
 
     driver = webdriver.Chrome(options=chrome_options)
 
@@ -26,44 +23,39 @@ def get_hidden_code(url):
 
     try:
         driver.get(url)
-        time.sleep(3)  # wait for page load
+        time.sleep(3)
 
-        # Try clicking Start Challenge button
+        # Try clicking the button
         try:
             start_button = driver.find_element(By.XPATH, "//button[contains(text(), 'Start Challenge')]")
             start_button.click()
-            time.sleep(3)  # wait for challenge to load
+            time.sleep(3)
         except Exception as e:
             print("⚠️ Could not click Start Challenge button:", e)
 
-        # Get updated HTML after click
         html = driver.page_source
         soup = BeautifulSoup(html, "html.parser")
 
-        # Find hidden elements
+        # 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))
 
-        # Find hidden input fields
         for hidden in soup.find_all("input", {"type": "hidden"}):
             if hidden.get("value"):
                 token_found.append(hidden["value"])
 
-        # Look for "Hidden Code" text
         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
@@ -75,7 +67,6 @@ def mission():
 
     tokens = get_hidden_code(url)
 
-    # Print in logs for debugging
     print("\n===== Mission Request =====")
     print("URL received:", url)
     print("Questions received:", questions)
@@ -88,7 +79,6 @@ def mission():
         "token_found": tokens
     })
 
-
 if __name__ == "__main__":
     import os
     port = int(os.environ.get("PORT", 7860))