Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import os
 import pandas as pd
-from azure.identity import ClientSecretCredential
+from azure.identity import AzureCliCredential, DefaultAzureCredential
 from azure.mgmt.resource import ResourceManagementClient
 from azure.cosmos import CosmosClient, exceptions
 from github import Github
@@ -9,12 +9,42 @@ from git import Repo
 import shutil
 from datetime import datetime
 import base64
+from playwright.sync_api import sync_playwright
+import time
 
 # Azure configuration
 ENDPOINT = "https://acae-afd.documents.azure.com:443/"
 DATABASE_NAME = os.environ.get("COSMOS_DATABASE_NAME")
 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
+        
+        browser.close()
+    
+    # For now, we'll just return True to indicate successful login
+    return True
+
 # GitHub configuration
 def download_github_repo(url, local_path):
     if os.path.exists(local_path):
@@ -78,28 +108,51 @@ if 'logged_in' not in st.session_state:
 # Login section
 if not st.session_state.logged_in:
     st.subheader("🔐 Login")
-    tenant_id = st.text_input("Azure Tenant ID")
-    client_id = st.text_input("Azure Client ID")
-    client_secret = st.text_input("Azure Client Secret", type="password")
-    subscription_id = st.text_input("Azure Subscription ID")
+    st.write("Please enter your Azure Subscription ID. You can find this in the Azure Portal under Subscriptions.")
+    subscription_id = st.text_input("Azure Subscription ID", value="003fba60-5b3f-48f4-ab36-3ed11bc40816")
+    
+    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 st.button("🚀 Login"):
-        try:
-            credential = ClientSecretCredential(
-                tenant_id=tenant_id,
-                client_id=client_id,
-                client_secret=client_secret
-            )
-            # Try to get a token to verify the credential works
-            credential.get_token("https://management.azure.com/.default")
-            st.session_state.credential = credential
-            st.session_state.subscription_id = subscription_id
-            st.session_state.logged_in = True
-            st.success("Successfully logged in!")
-            st.rerun()
-        except Exception as e:
-            st.error(f"Failed to authenticate: {str(e)}")
+        if login_method == "Automatic":
+            try:
+                # First, try AzureCliCredential
+                credential = AzureCliCredential()
+                credential.get_token("https://management.azure.com/.default")
+            except Exception:
+                # If AzureCliCredential fails, fall back to DefaultAzureCredential
+                try:
+                    credential = DefaultAzureCredential()
+                    credential.get_token("https://management.azure.com/.default")
+                except Exception as e:
+                    st.error(f"Failed to authenticate automatically. Please try the Manual (Browser Simulation) method. Error: {str(e)}")
+                    st.stop()
+        else:
+            if not username or not password:
+                st.error("Please enter both username and password for manual login.")
+                st.stop()
+            try:
+                if azure_login_with_playwright(username, password):
+                    credential = DefaultAzureCredential()
+                    credential.get_token("https://management.azure.com/.default")
+                else:
+                    st.error("Failed to authenticate using browser simulation.")
+                    st.stop()
+            except Exception as e:
+                st.error(f"Failed to authenticate using browser simulation. Error: {str(e)}")
+                st.stop()
+        
+        st.session_state.credential = credential
+        st.session_state.subscription_id = subscription_id
+        st.session_state.logged_in = True
+        st.success("Successfully logged in!")
+        st.rerun()
 else:
-    # Main app content
+    # Main app content (unchanged)
     col1, col2 = st.columns([1, 3])
 
     with col1: