Update app.py

app.py

@@ -11,12 +11,9 @@ from PIL import Image
 from io import BytesIO
 
 from playwright.sync_api import sync_playwright
-
-
-from selenium.webdriver.chrome.service import Service
-from webdriver_manager.chrome import ChromeDriverManager
 from selenium.webdriver.chrome.options import Options
 
+
 HF_TOKEN = os.getenv("HF_TOKEN")
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=HF_TOKEN)
 
@@ -113,20 +110,19 @@ def on_select(space):
         print(traceback.format_exc())
         return f"오류가 발생했습니다: {str(e)}", "", Image.new('RGB', (1, 1))
 
-
-
 def take_screenshot(url):
     try:
         chrome_options = Options()
         chrome_options.add_argument("--headless")
         chrome_options.add_argument("--no-sandbox")
         chrome_options.add_argument("--disable-dev-shm-usage")
+        chrome_options.add_argument("--disable-gpu")
+        chrome_options.add_argument("--window-size=1080,720")
         
-        service = Service(ChromeDriverManager().install())
-        driver = webdriver.Chrome(service=service, options=chrome_options)
+        # 컨테이너 환경에 맞는 ChromeDriver 경로 설정
+        driver = webdriver.Chrome(options=chrome_options)
         
         driver.get(url)
-        driver.set_window_size(1080, 720)
         screenshot = driver.get_screenshot_as_png()
         driver.quit()