Update app.py

app.py

@@ -56,6 +56,17 @@ def azure_login_with_playwright(username, password):
         st.error(f"An error occurred during browser simulation: {str(e)}")
         return False
 
+# Function to install Playwright
+def install_playwright():
+    try:
+        subprocess.run([sys.executable, "-m", "pip", "install", "playwright"], 
+                       capture_output=True, text=True, check=True)
+        st.success("Playwright installed successfully!")
+        return True
+    except subprocess.CalledProcessError as e:
+        st.error(f"Failed to install Playwright. Error: {e.stderr}")
+        return False
+
 # Function to install Playwright browsers
 def install_playwright_browsers():
     try:
@@ -139,8 +150,10 @@ if not st.session_state.logged_in:
         if not PLAYWRIGHT_INSTALLED:
             st.warning("Playwright is not installed. Please install it to use browser simulation.")
             if st.button("Install Playwright"):
-                if install_playwright_browsers():
-                    st.rerun()
+                if install_playwright():
+                    if install_playwright_browsers():
+                        st.success("Playwright and browsers installed successfully. Please restart the app.")
+                        st.stop()
         else:
             username = st.text_input("Azure Username/Email")
             password = st.text_input("Azure Password", type="password")