Update app.py

app.py

@@ -9,8 +9,15 @@ from git import Repo
 import shutil
 from datetime import datetime
 import base64
-from playwright.sync_api import sync_playwright
-import time
+import subprocess
+import sys
+
+# Check if playwright is installed
+try:
+    from playwright.sync_api import sync_playwright
+    PLAYWRIGHT_INSTALLED = True
+except ImportError:
+    PLAYWRIGHT_INSTALLED = False
 
 # Azure configuration
 ENDPOINT = "https://acae-afd.documents.azure.com:443/"
@@ -19,31 +26,46 @@ CONTAINER_NAME = os.environ.get("COSMOS_CONTAINER_NAME")
 
 # Playwright function for Azure login
 def azure_login_with_playwright(username, password):
-    with sync_playwright() as p:
-        browser = p.chromium.launch(headless=False)  # Set to True for headless mode
-        page = browser.new_page()
-        page.goto("https://portal.azure.com")
-        
-        # Wait for and fill in the email field
-        page.wait_for_selector('input[type="email"]')
-        page.fill('input[type="email"]', username)
-        page.click('input[type="submit"]')
-        
-        # Wait for and fill in the password field
-        page.wait_for_selector('input[type="password"]')
-        page.fill('input[type="password"]', password)
-        page.click('input[type="submit"]')
-        
-        # Wait for login to complete (adjust as needed)
-        page.wait_for_selector('button[id="nav_menu"]', timeout=60000)
-        
-        # At this point, the login should be complete
-        # You may want to extract any necessary tokens or cookies here
+    if not PLAYWRIGHT_INSTALLED:
+        st.error("Playwright is not installed. Please install it using 'pip install playwright'")
+        return False
+
+    try:
+        with sync_playwright() as p:
+            browser = p.chromium.launch(headless=False)  # Set to True for headless mode
+            page = browser.new_page()
+            page.goto("https://portal.azure.com")
+            
+            # Wait for and fill in the email field
+            page.wait_for_selector('input[type="email"]')
+            page.fill('input[type="email"]', username)
+            page.click('input[type="submit"]')
+            
+            # Wait for and fill in the password field
+            page.wait_for_selector('input[type="password"]')
+            page.fill('input[type="password"]', password)
+            page.click('input[type="submit"]')
+            
+            # Wait for login to complete (adjust as needed)
+            page.wait_for_selector('button[id="nav_menu"]', timeout=60000)
+            
+            browser.close()
         
-        browser.close()
-    
-    # For now, we'll just return True to indicate successful login
-    return True
+        return True
+    except Exception as e:
+        st.error(f"An error occurred during browser simulation: {str(e)}")
+        return False
+
+# Function to install Playwright browsers
+def install_playwright_browsers():
+    try:
+        result = subprocess.run([sys.executable, "-m", "playwright", "install"], 
+                                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. Error: {e.stderr}")
+        return False
 
 # GitHub configuration
 def download_github_repo(url, local_path):
@@ -114,8 +136,14 @@ if not st.session_state.logged_in:
     login_method = st.radio("Choose login method", ["Automatic", "Manual (Browser Simulation)"])
     
     if login_method == "Manual (Browser Simulation)":
-        username = st.text_input("Azure Username/Email")
-        password = st.text_input("Azure Password", type="password")
+        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()
+        else:
+            username = st.text_input("Azure Username/Email")
+            password = st.text_input("Azure Password", type="password")
     
     if st.button("🚀 Login"):
         if login_method == "Automatic":
@@ -132,6 +160,9 @@ if not st.session_state.logged_in:
                     st.error(f"Failed to authenticate automatically. Please try the Manual (Browser Simulation) method. Error: {str(e)}")
                     st.stop()
         else:
+            if not PLAYWRIGHT_INSTALLED:
+                st.error("Playwright is not installed. Please install it to use browser simulation.")
+                st.stop()
             if not username or not password:
                 st.error("Please enter both username and password for manual login.")
                 st.stop()
@@ -152,7 +183,7 @@ if not st.session_state.logged_in:
         st.success("Successfully logged in!")
         st.rerun()
 else:
-    # Main app content (unchanged)
+    # Main app content
     col1, col2 = st.columns([1, 3])
 
     with col1: