Update app.py

app.py

@@ -1,13 +1,12 @@
+import streamlit as st
+st.set_page_config(page_title="Advanced File Downloader", layout="wide")
+
 import os
 import subprocess
-import sys
 
 def install_playwright_dependencies():
     try:
-        # Update package list
-        os.system('apt-get update -y')
-        
-        # Install required dependencies including GTK
+        # Install system dependencies using apt-get
         dependencies = [
             'libnss3',
             'libnspr4',
@@ -16,28 +15,23 @@ def install_playwright_dependencies():
             'libcups2',
             'libxcomposite1',
             'libxdamage1',
-            'libatspi2.0-0',
-            'libgtk-3-0',  # Add GTK dependencies
-            'libgdk-3-0'
+            'libatspi2.0-0'
         ]
         
-        dependency_command = f"apt-get install -y {' '.join(dependencies)}"
-        os.system(dependency_command)
+        subprocess.run(['apt-get', 'update', '-y'], check=True)
+        subprocess.run(['apt-get', 'install', '-y'] + dependencies, check=True)
         
-        # Install playwright and its browsers
-        os.system('playwright install')
-        os.system('python -m playwright install')
+        # Install Playwright browser
+        subprocess.run(['playwright', 'install', 'chromium'], check=True)
         
-        st.success("Successfully installed Playwright dependencies")
     except Exception as e:
         st.error(f"Error installing dependencies: {e}")
-        st.error("Please run these commands manually in your terminal:")
-        st.code("apt-get update")
-        st.code(f"apt-get install -y {' '.join(dependencies)}")
-        st.code("playwright install")
-import streamlit as st
-# Must be the first Streamlit command
-st.set_page_config(page_title="Advanced File Downloader", layout="wide")
+        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
@@ -137,16 +131,37 @@ def dummy_roberta_transformer(nlp, name):
     return dummy
 
 @st.cache_resource
-def load_nlp_model():
+def load_models():
     try:
-        nlp_model = spacy.load("en_core_web_sm")
-    except OSError:
-        st.write("Model en_core_web_sm not found. Downloading it now...")
-        spacy.cli.download("en_core_web_sm")
-        nlp_model = spacy.load("en_core_web_sm")
-    return nlp_model
-
-nlp_model = load_nlp_model()
+        # Load spaCy model
+        try:
+            nlp = spacy.load("en_core_web_sm")
+        except OSError:
+            st.info("Downloading spaCy model...")
+            spacy.cli.download("en_core_web_sm")
+            nlp = spacy.load("en_core_web_sm")
+
+        # Load SentenceTransformer
+        try:
+            from sentence_transformers import SentenceTransformer
+            semantic_model = SentenceTransformer('all-MiniLM-L6-v2')
+        except Exception as e:
+            st.error(f"Error loading SentenceTransformer: {e}")
+            semantic_model = None
+
+        # Load Transformers pipeline with correct import
+        try:
+            from transformers import pipeline, AutoModelForSeq2SeqGenerationWithLMHead, AutoTokenizer
+            summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+        except Exception as e:
+            st.error(f"Error loading Transformers: {e}")
+            summarizer = None
+
+        return nlp, semantic_model, summarizer
+
+    except Exception as e:
+        st.error(f"Error loading models: {e}")
+        return None, None, None
 
 # Also load SentenceTransformer for semantic re-ranking.
 from sentence_transformers import SentenceTransformer, util