Move browser installation to runtime and improve error handling

Dockerfile

@@ -53,13 +53,6 @@ RUN useradd -m -u 1000 user && \
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Install Playwright browsers and dependencies as root
-RUN mkdir -p /home/user/.cache/ms-playwright && \
-    chown -R user:user /home/user/.cache/ms-playwright && \
-    playwright install chromium && \
-    playwright install-deps && \
-    chown -R user:user /home/user/.cache/ms-playwright
-
 # Copy the rest of the application
 COPY . .
 RUN chown -R user:user /app
@@ -72,7 +65,6 @@ ENV GRADIO_ALLOW_FLAGGING=false
 ENV GRADIO_QUEUE_ENABLED=false
 ENV DISPLAY=:99
 ENV PLAYWRIGHT_BROWSERS_PATH=/home/user/.cache/ms-playwright
-ENV PLAYWRIGHT_SKIP_BROWSER_DOWNLOAD=1
 
 # Switch to non-root user
 USER user
@@ -80,5 +72,5 @@ USER user
 # Expose the port
 EXPOSE 7860
 
-# Run the application with Xvfb
-CMD Xvfb :99 -screen 0 1280x1024x24 > /dev/null 2>&1 & python app.py 
+# Run the application with Xvfb and install browsers
+CMD Xvfb :99 -screen 0 1280x1024x24 > /dev/null 2>&1 & playwright install chromium && python app.py 

app.py

@@ -1,5 +1,6 @@
 import os
 import sys
+import subprocess
 from pathlib import Path
 
 # Add the current directory to Python path
@@ -8,9 +9,18 @@ if str(current_dir) not in sys.path:
     sys.path.append(str(current_dir))
 
 # Set environment variables for Playwright
-os.environ["PLAYWRIGHT_SKIP_BROWSER_DOWNLOAD"] = "1"
 os.environ["PLAYWRIGHT_BROWSERS_PATH"] = "/home/user/.cache/ms-playwright"
 
+# Ensure browser is installed
+try:
+    from playwright.sync_api import sync_playwright
+    with sync_playwright() as p:
+        p.chromium.launch()
+except Exception as e:
+    print(f"Browser installation error: {e}")
+    print("Attempting to install browser...")
+    subprocess.run(["playwright", "install", "chromium"], check=True)
+
 from owl.webapp import create_ui
 import gradio as gr