Update app.py

app.py

@@ -24,37 +24,77 @@ from langchain_core.messages import SystemMessage, HumanMessage
 load_dotenv()
 
 # --- Playwright Browser and System Dependency Installation Workaround for Hugging Face Spaces ---
-# This block ensures Playwright browsers AND specific system dependencies are installed when the app starts.
+# This block ensures Playwright browsers AND all necessary system dependencies are installed when the app starts.
 # It checks if browsers are already installed or if an attempt has been made.
 PLAYWRIGHT_BROWSERS_PATH = os.path.expanduser("~/.cache/ms-playwright")
 PLAYWRIGHT_INSTALLED_FLAG = "PLAYWRIGHT_BROWSERS_INSTALLED"
 
-# Define problematic system packages that apt-get struggles with via packages.txt
+# Define ALL system packages required for Playwright on Debian Bookworm
 # These will be installed manually via subprocess.
-PROBLEM_APT_PACKAGES = [
-    "libwoff2-1.0",
-    "libjpeg-turbo8"
+ALL_PLAYWRIGHT_APT_PACKAGES = [
+    "chromium", # Basic chromium package for good measure
+    "libgstreamer1.0-0",
+    "libgtk-4-1",
+    "libgraphene-1.0-0",
+    "libwoff2-1.0", # Reverted to original name, will be installed via apt-get in-app
+    "libgstreamer-plugins-base1.0-0",
+    "libgstreamer-gl1.0-0",
+    "libgstreamer-plugins-bad1.0-0",
+    "libavif15",
+    "libharfbuzz-icu0",
+    "libenchant-2-2",
+    "libhyphen0",
+    "libmanette-0.2-0",
+    "libnss3",
+    "libxss1",
+    "libasound2",
+    "libgbm-dev",
+    "libatk-bridge2.0-0",
+    "libcups2",
+    "libdrm-dev",
+    "libgdk-pixbuf2.0-0",
+    "libglib2.0-0",
+    "libjpeg-turbo8", # Reverted to original name, will be installed via apt-get in-app
+    "libnspr4",
+    "libxcb-dri3-0",
+    "libxcomposite1",
+    "libxrandr2",
+    "libxshmfence1",
+    "libxtst6",
+    "xdg-utils"
 ]
 
 # Check if installation has been attempted during this container's lifetime
 if not os.path.exists(os.path.join(PLAYWRIGHT_BROWSERS_PATH, "chromium")) and \
    os.environ.get(PLAYWRIGHT_INSTALLED_FLAG, "false") == "false":
-    print("Playwright browsers and/or specific system dependencies not found. Attempting to install them now (this might take a moment)...")
+    print("Playwright browsers and/or system dependencies not found. Attempting a comprehensive installation now (this will take several minutes)...")
     try:
-        # First, try to install the problematic system dependencies via apt-get directly
-        print(f"Installing specific system dependencies via apt-get: {', '.join(PROBLEM_APT_PACKAGES)}...")
-        apt_install_command = ["sudo", "apt-get", "update", "-qq"] # -qq for quiet update
-        subprocess.run(apt_install_command, check=True, capture_output=True)
-        
-        apt_install_command = ["sudo", "apt-get", "install", "-y"] + PROBLEM_APT_PACKAGES
-        subprocess.run(apt_install_command, check=True, capture_output=True)
-        print("Specific system dependencies installed successfully!")
-
-        # Then, run the playwright install command
+        # First, update apt package lists
+        print("Updating apt package lists...")
+        # Use stdout=subprocess.PIPE, stderr=subprocess.PIPE to capture output and prevent it from filling up logs unless an error occurs
+        subprocess.run(["sudo", "apt-get", "update", "-qq"], check=True, capture_output=True)
+        print("Apt package lists updated.")
+
+        # Then, install all necessary system dependencies via apt-get directly
+        print(f"Installing all Playwright system dependencies via apt-get: {', '.join(ALL_PLAYWRIGHT_APT_PACKAGES)}...")
+        apt_install_command = ["sudo", "apt-get", "install", "-y"] + ALL_PLAYWRIGHT_APT_PACKAGES
+        result = subprocess.run(apt_install_command, check=True, capture_output=True, text=True)
+        print("All system dependencies installed successfully!")
+        if result.stdout:
+            print("APT Install STDOUT:\n", result.stdout)
+        if result.stderr:
+            print("APT Install STDERR:\n", result.stderr)
+
+        # Finally, run the playwright install command for the browsers themselves
         print("Running playwright install for browsers...")
-        subprocess.run([sys.executable, "-m", "playwright", "install"], check=True)
+        result = subprocess.run([sys.executable, "-m", "playwright", "install"], check=True, capture_output=True, text=True)
         os.environ[PLAYWRIGHT_INSTALLED_FLAG] = "true" # Mark as installed for this session
         print("Playwright browsers installed successfully!")
+        if result.stdout:
+            print("Playwright Install STDOUT:\n", result.stdout)
+        if result.stderr:
+            print("Playwright Install STDERR:\n", result.stderr)
+            
     except subprocess.CalledProcessError as e:
         print(f"Error during Playwright/apt-get installation: {e}. STDOUT: {e.stdout.decode()} STDERR: {e.stderr.decode()}")
     except Exception as e: