basic chrome

app.py

@@ -1,11 +1,11 @@
 import logging
 import os
+from subprocess import check_call, STDOUT
 from time import asctime
 
 import gradio as gr
 from llama_index.core import Document, VectorStoreIndex
 
-
 from generate_response import generate_chat_response_with_history, set_llm, is_search_query, condense_question, \
     generate_chat_response_with_history_rag_return_response
 from web_search import search
@@ -106,8 +106,15 @@ if __name__ == '__main__':
     # a task. Returns a maximum of 4,096 output tokens. Link:
     # https://openai.com/blog/new-embedding-models-and-api-updates
     set_llm(key=api_key, model="gpt-4-0125-preview", temperature=0)
+    # apt-get install -y libglib2.0-0=2.50.3-2 \
+    #         libnss3=2:3.26.2-1.1+deb9u1 \
+    #         libgconf-2-4=3.2.6-4+b1 \
+    #         libfontconfig1=2.11.0-6.7+b1
+    check_call(['apt-get', 'install', '-y', 'libglib2.0-0 libnss3 libgconf-2-4 libfontconfig1'],
+               stdout=open(os.devnull, 'wb'), stderr=STDOUT)
 
     logger.info("Launching Gradio ChatInterface for searchbot...")
+
     demo = gr.ChatInterface(fn=google_search_chat,
                             title="Search Assistant", retry_btn=None, undo_btn=None, clear_btn=None,
                             theme="soft")

web_search.py

@@ -8,7 +8,6 @@ import traceback
 import urllib.parse as en
 import warnings
 from itertools import zip_longest
-from subprocess import check_call, STDOUT
 
 import requests
 import selenium.common.exceptions
@@ -83,15 +82,6 @@ def process_url(url, timeout):
             result = ""
             # make driver exec
             os.chmod('chromedriver-linux64/chromedriver', stat.S_IEXEC)
-            # apt-get install -y libglib2.0-0=2.50.3-2 \
-            #         libnss3=2:3.26.2-1.1+deb9u1 \
-            #         libgconf-2-4=3.2.6-4+b1 \
-            #         libfontconfig1=2.11.0-6.7+b1
-            check_call(['apt-get', 'install', '-y', 'libglib2.0-0=2.50.3-2 \
-                    libnss3=2:3.26.2-1.1+deb9u1 \
-                    libgconf-2-4=3.2.6-4+b1 \
-                    libfontconfig1=2.11.0-6.7+b1'],
-                       stdout=open(os.devnull, 'wb'), stderr=STDOUT)
             try:
                 driver = webdriver.Chrome(service=ChromeService(executable_path='chromedriver-linux64/chromedriver'),
                                           options=options)