Bug-Fixed

Dockerfile

@@ -8,7 +8,12 @@ ENV TZ=Etc/UTC
 ENV HOME=/home/appuser
 ENV PYTHONUSERBASE=/home/appuser/.local
 
-# System deps + Chromium + ChromeDriver + common headless libs
+# HuggingFace Spaces specific optimizations
+ENV CHROME_BIN=/usr/bin/chromium
+ENV CHROMEDRIVER_PATH=/usr/bin/chromedriver
+ENV DISPLAY=:99
+
+# System deps + Chromium + ChromeDriver + essential libs for HF Spaces
 RUN apt-get update && apt-get install -y --no-install-recommends \
     wget \
     gnupg \
@@ -24,11 +29,22 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libgtk-3-0 \
     libgbm1 \
     fonts-liberation \
+    libappindicator3-1 \
+    libasound2 \
+    libdbus-glib-1-2 \
+    libdrm2 \
+    libxcomposite1 \
+    libxdamage1 \
+    libxrandr2 \
+    xvfb \
+    procps \
     && rm -rf /var/lib/apt/lists/*
 
-# Create unprivileged user
+# Create unprivileged user with proper permissions
 RUN addgroup --system appuser && adduser --system --ingroup appuser appuser
-RUN mkdir -p /home/appuser/.local && chown -R appuser:appuser /app /home/appuser
+RUN mkdir -p /home/appuser/.local /tmp/chrome-temp && \
+    chown -R appuser:appuser /app /home/appuser /tmp/chrome-temp && \
+    chmod 755 /tmp/chrome-temp
 
 USER appuser
 
@@ -39,6 +55,9 @@ ENV PATH="/home/appuser/.local/bin:${PATH}"
 
 COPY --chown=appuser:appuser . .
 
+# Create config directory if needed
+RUN mkdir -p config
+
 EXPOSE 7860
 
 CMD ["python", "server.py"]

worker.py

@@ -1,6 +1,10 @@
 # worker.py
 import time
 import threading
+import tempfile
+import shutil
+import os
+import subprocess
 
 from selenium import webdriver
 from selenium.webdriver.chrome.service import Service as ChromeService
@@ -18,42 +22,83 @@ class QuantumBot:
         self.driver = None
         self.DEFAULT_TIMEOUT = 30
         self.termination_event = threading.Event()
+        self.temp_user_dir = None
+
+    def _kill_existing_chrome_processes(self):
+        """Kill any existing Chrome processes that might be locking directories"""
+        try:
+            subprocess.run(['pkill', '-f', 'chrome'], capture_output=True, text=True)
+            subprocess.run(['pkill', '-f', 'chromium'], capture_output=True, text=True)
+            time.sleep(2)
+        except Exception:
+            pass
 
     def initialize_driver(self):
         try:
             self.micro_status("Initializing headless browser...")
+            
+            # Kill any existing Chrome processes first
+            self._kill_existing_chrome_processes()
+            
+            # Create a completely isolated temporary directory for this session
+            self.temp_user_dir = tempfile.mkdtemp(prefix="hf-chrome-", dir="/tmp")
+            os.chmod(self.temp_user_dir, 0o755)
+            
             options = ChromeOptions()
-
-            # Use system Chromium installed in the image
-            options.binary_location = "/usr/bin/chromium"
-
-            # Do NOT set --user-data-dir; let ChromeDriver create a fresh temp profile per session
-            # Use an ephemeral DevTools port to avoid collisions in constrained environments
-            options.add_argument("--remote-debugging-port=0")
-
-            # Stable/headless startup flags
-            user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/115.0.0.0 Safari/537.36"
-            options.add_argument(f"--user-agent={user_agent}")
+            
+            # Force Chrome to use our isolated temporary directory
+            options.add_argument(f"--user-data-dir={self.temp_user_dir}")
+            options.add_argument(f"--data-path={self.temp_user_dir}")
+            options.add_argument(f"--disk-cache-dir={self.temp_user_dir}/cache")
+            
+            # HuggingFace Spaces specific optimizations
             options.add_argument("--headless=new")
-            options.add_argument("--window-size=1920,1080")
-            options.add_argument("--no-first-run")
-            options.add_argument("--no-default-browser-check")
             options.add_argument("--no-sandbox")
             options.add_argument("--disable-dev-shm-usage")
             options.add_argument("--disable-gpu")
-            options.add_argument("--disable-features=Translate,AutomationControlled")
-
+            options.add_argument("--disable-software-rasterizer")
+            options.add_argument("--disable-background-timer-throttling")
+            options.add_argument("--disable-backgrounding-occluded-windows")
+            options.add_argument("--disable-renderer-backgrounding")
+            options.add_argument("--disable-features=TranslateUI,BlinkGenPropertyTrees")
+            options.add_argument("--remote-debugging-port=0")
+            options.add_argument("--no-first-run")
+            options.add_argument("--no-default-browser-check")
+            options.add_argument("--disable-extensions")
+            options.add_argument("--disable-plugins")
+            options.add_argument("--disable-images")
+            options.add_argument("--disable-javascript")
+            options.add_argument("--window-size=1920,1080")
+            
+            # Set user agent
+            user_agent = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
+            options.add_argument(f"--user-agent={user_agent}")
+            
+            # Set binary location for HF Spaces
+            options.binary_location = "/usr/bin/chromium"
+            
             service = ChromeService(executable_path="/usr/bin/chromedriver")
+            
+            # Set service args for additional stability
+            service.service_args = ["--verbose", "--whitelisted-ips="]
+            
             self.driver = webdriver.Chrome(service=service, options=options)
-
+            
             # Reduce webdriver detection
             self.driver.execute_cdp_cmd('Page.addScriptToEvaluateOnNewDocument', {
                 'source': "Object.defineProperty(navigator, 'webdriver', {get: () => undefined})"
             })
+            
             return True, None
+            
         except Exception as e:
             error_message = f"Message: {str(e)}"
             print(f"CRITICAL ERROR in WebDriver Initialization: {error_message}")
+            
+            # Cleanup on failure
+            if self.temp_user_dir and os.path.exists(self.temp_user_dir):
+                shutil.rmtree(self.temp_user_dir, ignore_errors=True)
+                
             return False, error_message
 
     def micro_status(self, message):
@@ -196,6 +241,19 @@ class QuantumBot:
             return 'Error'
 
     def shutdown(self):
-        if self.driver:
-            self.driver.quit()
+        try:
+            if self.driver:
+                self.driver.quit()
+            
+            # Kill any remaining Chrome processes
+            self._kill_existing_chrome_processes()
+            
+            # Clean up temporary directory
+            if self.temp_user_dir and os.path.exists(self.temp_user_dir):
+                shutil.rmtree(self.temp_user_dir, ignore_errors=True)
+                print(f"[Bot] Cleaned up temporary Chrome profile: {self.temp_user_dir}")
+                
+        except Exception as e:
+            print(f"[Bot] Error during shutdown: {e}")
+        
         print("[Bot] Chrome session closed.")