Update app.py

app.py

@@ -20,6 +20,7 @@ import json
 import pickle
 from nltk.tokenize import word_tokenize
 from nltk.stem.lancaster import LancasterStemmer
+import subprocess
 
 # Ensure necessary NLTK resources are downloaded
 nltk.download('punkt')
@@ -120,13 +121,20 @@ def install_chrome_and_driver():
     os.system("google-chrome-stable --version")
 
     # Fix ownership of /etc/sudo.conf
-    os.system("chown root:root /etc/sudo.conf")
+    os.system("sudo chown root:root /etc/sudo.conf")
 
     # Verify Chrome installation
     os.system("which google-chrome-stable")
     if not os.path.exists("/usr/bin/google-chrome-stable"):
         raise RuntimeError("Google Chrome was not installed correctly")
 
+    # Check if CUDA libraries are available and install them if present
+    try:
+        os.system("apt-get install -y cuda")
+        os.system("apt-get install -y libcudart.so.11.0")
+    except subprocess.CalledProcessError:
+        print("CUDA libraries not found or installation failed. Proceeding without GPU support.")
+
     # Install Chromedriver (if not already installed)
     chromedriver_autoinstaller.install()
 
@@ -210,7 +218,7 @@ def scrape_website_for_contact_info(website):
         if phone_match:
             phone_number = phone_match.group()
 
-            email_match = re.search(r'[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}', soup.get_text())
+        email_match = re.search(r'[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}', soup.get_text())
         if email_match:
             email = email_match.group()