Add application file4

app.py

@@ -4,31 +4,56 @@ import json
 import time
 import streamlit as st
 import base64
+import subprocess  # Add subprocess import
 from io import BytesIO
 from dotenv import load_dotenv
-# Remove Selenium imports
-# from selenium import webdriver
-# from selenium.webdriver.chrome.options import Options
-# from selenium.webdriver.common.by import By
-# from selenium.webdriver.common.keys import Keys
-# from selenium.webdriver.support.ui import WebDriverWait
-# from selenium.webdriver.support import expected_conditions as EC
 from google import genai
-# Add Playwright imports
 from playwright.sync_api import sync_playwright
 import platform
+import sys
 
 # Load environment variables from .env
 load_dotenv()
 GEMINI_API_KEY = 'AIzaSyAOK9vRTSRQzd22B2gmbiuIePbZTDyaGYs'
 
-# --- Playwright Browser Setup ---
+# --- Playwright Setup ---
+
+if "PLAYWRIGHT_BROWSERS_INSTALLED" not in os.environ:
+    st.info("Installing Playwright browsers...")
+    os.system("playwright install chromium")
+    os.environ["PLAYWRIGHT_BROWSERS_INSTALLED"] = "1"
+    st.success("Browser installation complete!")
+    
+def install_playwright_browsers():
+    """Install Playwright browsers programmatically"""
+    try:
+        st.info("Installing Playwright browsers...")
+        result = subprocess.run(
+            [sys.executable, "-m", "playwright", "install", "chromium"],
+            capture_output=True,
+            text=True,
+            check=True
+        )
+        st.success("Playwright browsers installed successfully!")
+        return True
+    except subprocess.CalledProcessError as e:
+        st.error(f"Failed to install Playwright browsers: {e}")
+        st.error(f"Error output: {e.stderr}")
+        return False
+    except Exception as e:
+        st.error(f"Unexpected error installing browsers: {e}")
+        return False
 
 def get_browser():
     """Get a Playwright browser instance that works in Hugging Face Spaces"""
     try:
+        # Make sure browsers are installed first
+        if "playwright_browsers_installed" not in st.session_state:
+            if not install_playwright_browsers():
+                raise Exception("Failed to install browser binaries")
+            st.session_state.playwright_browsers_installed = True
+            
         st.info("Initializing Playwright browser...")
-        # This will download the browser if needed
         playwright_instance = sync_playwright().start()
         browser = playwright_instance.chromium.launch(
             headless=True,
@@ -434,13 +459,18 @@ def login_to_google(browser_data, email, password):
         return False
 
 # --- Streamlit App ---
-
 st.title("Google Form Auto Filler with Gemini")
 st.write("""
 This app uses a headless browser to help you fill Google Forms automatically with AI-generated answers.
 You'll be able to see screenshots of what's happening in the browser as it progresses.
 """)
 
+# Install Playwright browsers when app starts (if not already installed)
+if "playwright_browsers_installed" not in st.session_state:
+    with st.spinner("Setting up browser environment (this might take a few minutes the first time)..."):
+        install_playwright_browsers()
+        st.session_state.playwright_browsers_installed = True
+
 # Initialize session state variables
 if "browser_data" not in st.session_state:
     st.session_state.browser_data = None
@@ -450,7 +480,6 @@ if "form_filled" not in st.session_state:
     st.session_state.form_filled = False
 if "screenshot" not in st.session_state:
     st.session_state.screenshot = None
-
 # Step 1: Login to Google Account
 st.header("Step 1: Login to Google Account")