Update app.py

app.py

@@ -103,26 +103,29 @@ def on_select(space):
         print(traceback.format_exc())
         return f"오류가 발생했습니다: {str(e)}", "", Image.new('RGB', (1, 1))
         
-
 def take_screenshot(url):
     options = webdriver.ChromeOptions()
     options.add_argument('--headless')
     options.add_argument('--no-sandbox')
     options.add_argument('--disable-dev-shm-usage')
 
+    wd = None
     try:
         wd = webdriver.Chrome(options=options)
         wd.set_window_size(1080, 720)
         wd.get(url)
         wd.implicitly_wait(10)
         screenshot = wd.get_screenshot_as_png()
+        return Image.open(BytesIO(screenshot))
     except WebDriverException as e:
+        print(f"WebDriver error: {str(e)}")
+        return Image.new('RGB', (1, 1))
+    except Exception as e:
+        print(f"Unexpected error in take_screenshot: {str(e)}")
         return Image.new('RGB', (1, 1))
     finally:
         if wd:
             wd.quit()
-
-    return Image.open(BytesIO(screenshot))
     
 def create_ui():
     spaces_list = get_most_liked_spaces()
@@ -173,6 +176,7 @@ def create_ui():
             )
 
     return demo
+    
 if __name__ == "__main__":
     demo = create_ui()
     demo.launch()