Update app.py

app.py

@@ -12,6 +12,11 @@ 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)
 
@@ -108,18 +113,29 @@ def on_select(space):
         print(traceback.format_exc())
         return f"오류가 발생했습니다: {str(e)}", "", Image.new('RGB', (1, 1))
 
+
+
 def take_screenshot(url):
     try:
-        with sync_playwright() as p:
-            browser = p.chromium.launch()
-            page = browser.new_page()
-            page.goto(url)
-            screenshot = page.screenshot()
-            browser.close()
+        chrome_options = Options()
+        chrome_options.add_argument("--headless")
+        chrome_options.add_argument("--no-sandbox")
+        chrome_options.add_argument("--disable-dev-shm-usage")
+        
+        service = Service(ChromeDriverManager().install())
+        driver = webdriver.Chrome(service=service, options=chrome_options)
+        
+        driver.get(url)
+        driver.set_window_size(1080, 720)
+        screenshot = driver.get_screenshot_as_png()
+        driver.quit()
+        
         return Image.open(BytesIO(screenshot))
     except Exception as e:
         print(f"Screenshot error: {str(e)}")
         return Image.new('RGB', (1, 1))
+
+
 def create_ui():
     try:
         spaces_list = get_most_liked_spaces()
@@ -170,11 +186,6 @@ def create_ui():
 
 if __name__ == "__main__":
     try:
-        # Playwright 브라우저 및 의존성 설치
-        import subprocess
-        subprocess.run(["playwright", "install", "chromium"], check=True)
-        subprocess.run(["playwright", "install-deps"], check=True)
-        
         demo = create_ui()
         demo.launch()
     except Exception as e: