Update app.py

app.py

@@ -1,37 +1,7 @@
 import streamlit as st
 st.set_page_config(page_title="Advanced File Downloader", layout="wide")
 
-import os
-import subprocess
-
-def install_playwright_dependencies():
-    try:
-        # Install system dependencies using apt-get
-        dependencies = [
-            'libnss3',
-            'libnspr4',
-            'libatk1.0-0',
-            'libatk-bridge2.0-0',
-            'libcups2',
-            'libxcomposite1',
-            'libxdamage1',
-            'libatspi2.0-0'
-        ]
-        
-        subprocess.run(['apt-get', 'update', '-y'], check=True)
-        subprocess.run(['apt-get', 'install', '-y'] + dependencies, check=True)
-        
-        # Install Playwright browser
-        subprocess.run(['playwright', 'install', 'chromium'], check=True)
-        
-    except Exception as e:
-        st.error(f"Error installing dependencies: {e}")
-        st.error("Please install the following dependencies manually:")
-        st.code("apt-get install -y " + " ".join(dependencies))
 
-# Call installation function
-with st.spinner("Installing required dependencies..."):
-    install_playwright_dependencies()
 
 # Import other required packages
 import spacy
@@ -118,6 +88,45 @@ import zipfile
 import tempfile
 import mimetypes
 import requests
+# -------------------- Playwright Setup --------------------
+def install_playwright_dependencies():
+    os.environ['PLAYWRIGHT_BROWSERS_PATH'] = os.path.expanduser("~/.cache/ms-playwright")
+    os.environ['LD_LIBRARY_PATH'] = '/usr/lib/playwright:/usr/lib/x86_64-linux-gnu'
+    try:
+        subprocess.run(['apt-get', 'update', '-y'], check=True)
+        packages = [
+            'libnss3', 'libnss3-tools', 'libnspr4', 'libatk1.0-0',
+            'libatk-bridge2.0-0', 'libatspi2.0-0', 'libcups2', 'libxcomposite1',
+            'libxdamage1', 'libdrm2', 'libgbm1', 'libpango-1.0-0'
+        ]
+        subprocess.run(['apt-get', 'install', '-y', '--no-install-recommends'] + packages, check=True)
+        os.makedirs('/usr/lib/playwright', exist_ok=True)
+        symlinks = {
+            'libnss3.so': '/usr/lib/x86_64-linux-gnu/libnss3.so',
+            'libnssutil3.so': '/usr/lib/x86_64-linux-gnu/libnssutil3.so',
+            'libsmime3.so': '/usr/lib/x86_64-linux-gnu/libsmime3.so',
+            'libnspr4.so': '/usr/lib/x86_64-linux-gnu/libnspr4.so',
+            'libatk-1.0.so.0': '/usr/lib/x86_64-linux-gnu/libatk-1.0.so.0',
+            'libatk-bridge-2.0.so.0': '/usr/lib/x86_64-linux-gnu/libatk-bridge-2.0.so.0',
+            'libcups.so.2': '/usr/lib/x86_64-linux-gnu/libcups.so.2',
+            'libatspi.so.0': '/usr/lib/x86_64-linux-gnu/libatspi.so.0',
+            'libXcomposite.so.1': '/usr/lib/x86_64-linux-gnu/libXcomposite.so.1',
+            'libXdamage.so.1': '/usr/lib/x86_64-linux-gnu/libXdamage.so.1'
+        }
+        for link_name, target in symlinks.items():
+            link_path = os.path.join('/usr/lib/playwright', link_name)
+            if not os.path.exists(link_path):
+                os.symlink(target, link_path)
+        subprocess.run(['python3', '-m', 'playwright', 'install', 'chromium'], check=True)
+        browser_path = os.path.expanduser("~/.cache/ms-playwright")
+        os.makedirs(browser_path, exist_ok=True)
+        subprocess.run(['chmod', '-R', '755', browser_path], check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Error installing dependencies: {e}")
+    except Exception as e:
+        print(f"Error: {e}")
+
+install_playwright_dependencies()
 
 # -------------------- spaCy Model Setup --------------------
 import spacy