Update app.py

app.py

@@ -2,8 +2,11 @@ import gradio as gr
 import helium
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.chrome.service import Service  # Import Service
 from io import BytesIO
 from PIL import Image
+import os
+
 
 def capture_screenshot(url):
     """
@@ -13,12 +16,20 @@ def capture_screenshot(url):
         # Chrome options for headless browsing
         chrome_options = Options()
         chrome_options.add_argument("--headless")  # Headless mode
-        chrome_options.add_argument("--no-sandbox") # For security reasons on some platforms
-        chrome_options.add_argument("--disable-dev-shm-usage") # Fixes /dev/shm out-of-memory issues
+        chrome_options.add_argument("--no-sandbox")  # For security reasons on some platforms
+        chrome_options.add_argument("--disable-dev-shm-usage")  # Fixes /dev/shm out-of-memory issues
         chrome_options.add_argument("--window-size=1024,768")  # Set a reasonable window size
+        # chrome_options.binary_location = '/usr/bin/google-chrome' # これは不要 or 間違い
+
+        # Use Service object (explicitly specify chromedriver path if needed)
+        # Hugging Face Spaces should handle chromedriver automatically,
+        # so we usually don't need to specify the path. But if it still fails,
+        # we can try to find it and specify.
+        # service = Service(executable_path="/path/to/chromedriver") # 通常は不要
+        service = Service() # これで動くはず
 
-        # Start the browser with helium (using the options)
-        driver = helium.start_chrome(options=chrome_options)
+        # Start the browser with helium (using the options and service)
+        driver = helium.start_chrome(options=chrome_options, service=service)
         helium.go_to(url)
 
         # Get screenshot as bytes
@@ -27,27 +38,24 @@ def capture_screenshot(url):
         # Convert bytes to PIL Image
         image = Image.open(BytesIO(png_bytes))
 
-        # Kill the browser
-        helium.kill_browser()
         return image
 
     except Exception as e:
         # Handle errors gracefully.
-        return f"Error: {e}"
+        return f"Error: {e}"  # Return the error message
     finally:
         try:
-            helium.kill_browser()
+            helium.kill_browser()  # Always kill the browser
         except:
             pass
 
 
-
 # Gradio Interface
 if __name__ == '__main__':
     iface = gr.Interface(
         fn=capture_screenshot,
         inputs=gr.Textbox(label="Enter URL", placeholder="https://www.example.com"),
-        outputs=gr.Image(type="pil", label="Screenshot"), # Use PIL Image output
+        outputs=gr.Image(type="pil", label="Screenshot"),
         title="HTML Screenshot Capture",
         description="Enter a URL to capture a screenshot of the webpage.",
         examples=[
@@ -56,4 +64,5 @@ if __name__ == '__main__':
             ["https://www.github.com"]
         ],
     )
-    iface.launch(server_name="0.0.0.0", server_port=7860)
+    # Disable caching (to avoid potential caching-related errors during development)
+    iface.launch(server_name="0.0.0.0", server_port=7860, cache_examples=False)