Update app.py

app.py

@@ -15,13 +15,32 @@ def log_versions():
         # Print PATH for debugging
         print(f"System PATH: {os.environ.get('PATH')}")
         # Try finding Chrome binary
-        chrome_path = subprocess.check_output(["which", "google-chrome"]).decode().strip()
-        print(f"Chrome binary found at: {chrome_path}")
-        # Get versions
-        chrome_version = subprocess.check_output(["google-chrome", "--version"]).decode().strip()
-        chromedriver_version = subprocess.check_output(["chromedriver", "--version"]).decode().strip()
+        chrome_binary = None
+        for binary in ["google-chrome", "google-chrome-stable", "chromium-browser"]:
+            try:
+                chrome_path = subprocess.check_output(["which", binary]).decode().strip()
+                print(f"Found {binary} at: {chrome_path}")
+                chrome_binary = binary
+                break
+            except subprocess.CalledProcessError:
+                continue
+        if not chrome_binary:
+            print("Error: No Chrome binary (google-chrome, google-chrome-stable, chromium-browser) found")
+            return
+
+        # Get Chrome version
+        chrome_version = subprocess.check_output([chrome_binary, "--version"]).decode().strip()
         print(f"Chrome version: {chrome_version}")
-        print(f"ChromeDriver version: {chromedriver_version}")
+
+        # Try finding ChromeDriver
+        try:
+            chromedriver_path = subprocess.check_output(["which", "chromedriver"]).decode().strip()
+            print(f"Found chromedriver at: {chromedriver_path}")
+            chromedriver_version = subprocess.check_output(["chromedriver", "--version"]).decode().strip()
+            print(f"ChromeDriver version: {chromedriver_version}")
+        except subprocess.CalledProcessError as e:
+            print(f"Error finding chromedriver: {str(e)}")
+
     except Exception as e:
         print(f"Error checking versions: {str(e)}")
 
@@ -97,7 +116,7 @@ iface = gr.Interface(
     fn=gradio_interface,
     inputs=gr.Textbox(label="Enter your prompt", placeholder="e.g., A serene mountain landscape at sunset"),
     outputs=[gr.Textbox(label="Result"), gr.Image(label="Generated Image")],
-    title="Image Generator with Selenium",
+    title="Image Generator with selenium",
     description="Enter a prompt to generate an image using gptimage.ai."
 )