Update app.py

app.py

@@ -5,12 +5,24 @@ from selenium.webdriver.common.keys import Keys
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.chrome.options import Options
-from selenium.webdriver.chrome.service import Service
-from webdriver_manager.chrome import ChromeDriverManager
 import time
+import subprocess
+
+# Function to log Chrome and ChromeDriver versions
+def log_versions():
+    try:
+        chrome_version = subprocess.check_output(["google-chrome", "--version"]).decode().strip()
+        chromedriver_version = subprocess.check_output(["chromedriver", "--version"]).decode().strip()
+        print(f"Chrome version: {chrome_version}")
+        print(f"ChromeDriver version: {chromedriver_version}")
+    except Exception as e:
+        print(f"Error checking versions: {str(e)}")
 
 # Function to generate image URL
 def generate_image(prompt):
+    # Log versions for debugging
+    log_versions()
+
     # Configure Chrome options for headless mode
     chrome_options = Options()
     chrome_options.add_argument("--headless")
@@ -21,14 +33,9 @@ def generate_image(prompt):
 
     # Initialize the WebDriver
     try:
-        # Try using system-installed ChromeDriver
         driver = webdriver.Chrome(options=chrome_options)
     except Exception as e:
-        print(f"System ChromeDriver failed: {e}. Falling back to webdriver-manager.")
-        try:
-            driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=chrome_options)
-        except Exception as e2:
-            return f"Error initializing ChromeDriver: {str(e2)}"
+        return f"Error initializing ChromeDriver: {str(e)}"
 
     try:
         # Open the website
@@ -89,4 +96,4 @@ iface = gr.Interface(
 
 # Launch the app
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(server_name="0.0.0.0", server_port=7860)  # No share=True