Update app.py

app.py

@@ -23,28 +23,44 @@ from langchain_core.messages import SystemMessage, HumanMessage
 # On Hugging Face Spaces, secrets are automatically available as environment variables.
 load_dotenv()
 
-# --- Playwright Browser Installation Workaround for Hugging Face Spaces ---
-# This block ensures Playwright browsers are installed when the app starts,
-# especially useful when setup.sh is not reliably executed or for cold starts.
+# --- 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.
 # It checks if browsers are already installed or if an attempt has been made.
 PLAYWRIGHT_BROWSERS_PATH = os.path.expanduser("~/.cache/ms-playwright")
-# Use a custom environment variable to track if installation has been attempted during this container's lifetime
-# This prevents it from running on every single request.
+PLAYWRIGHT_INSTALLED_FLAG = "PLAYWRIGHT_BROWSERS_INSTALLED"
+
+# Define problematic system packages that apt-get struggles with via packages.txt
+# These will be installed manually via subprocess.
+PROBLEM_APT_PACKAGES = [
+    "libwoff2-1.0",
+    "libjpeg-turbo8"
+]
+
+# 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_BROWSERS_INSTALLED", "false") == "false":
-    print("Playwright browsers not found. Attempting to install them now (this might take a moment)...")
+   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)...")
     try:
-        # Run the playwright install command as a subprocess
-        # sys.executable ensures we use the correct Python interpreter's playwright
+        # 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
+        print("Running playwright install for browsers...")
         subprocess.run([sys.executable, "-m", "playwright", "install"], check=True)
-        os.environ["PLAYWRIGHT_BROWSERS_INSTALLED"] = "true" # Mark as installed for this session
+        os.environ[PLAYWRIGHT_INSTALLED_FLAG] = "true" # Mark as installed for this session
         print("Playwright browsers installed successfully!")
     except subprocess.CalledProcessError as e:
-        print(f"Error installing Playwright browsers: {e}. Output: {e.stdout.decode() if e.stdout else ''} {e.stderr.decode() if e.stderr else ''}")
+        print(f"Error during Playwright/apt-get installation: {e}. STDOUT: {e.stdout.decode()} STDERR: {e.stderr.decode()}")
     except Exception as e:
-        print(f"Unexpected error during Playwright browser installation: {e}")
+        print(f"Unexpected error during Playwright browser or system dependency installation: {e}")
 else:
-    print("Playwright browsers already appear to be installed or installation skipped.")
+    print("Playwright browsers and required system dependencies already appear to be installed or installation skipped.")
 
 
 # --- 1. API KEY CHECK ---
@@ -173,7 +189,7 @@ async def scrape_flight_vouchers(details):
                     })
                     continue
     except Exception as e_browser_launch: # Catching general Exception for browser launch errors
-        print(f"Playwright browser launch error: {e_browser_launch}. Ensure `playwright install` was run.")
+        print(f"Playwright browser launch error: {e_browser_launch}. Ensure `playwright install` was run and system dependencies are met.")
         return [{
             "source": "Scraper Initialization Error",
             "type": "voucher",
@@ -339,7 +355,7 @@ async def ask_bot(question):
 with gr.Blocks(theme=gr.themes.Soft()) as iface:
     gr.Markdown(
         """
-        # ✨ The Ultimate Global Travel Planner Bot - Bazinga Boi Edition ✨
+        # ✨ The Ultimate Global Travel Planner Bot - Bazinga Edition ✨
         Your smart AI assistant for finding the best flights, hotels, and activities worldwide BAZINGA!
         """
     )