feat: add HF authentication setup and testing infrastructure - Add save_hf_storage_state.py for auth state management - Add check_hf_login.py for login verification - Update test_local.sh with Playwright setup - Add .cursor rules for test-first development - Update requirements.txt with new dependencies

.cursor/rules/test-first.mdc

@@ -0,0 +1,8 @@
+---
+description: test before commiting. 
+globs: 
+---
+
+# YOU MUST RUN THE LOCAL TESTS BEFORE PUSHING
+
+[test_local.sh](mdc:test_local.sh)

.gitignore

@@ -27,4 +27,6 @@ Thumbs.db
 
 test-image.png
 
-/chrome_profile/ 
+/chrome_profile/
+
+auth_state.json 

check_hf_login.py

@@ -0,0 +1,26 @@
+import os
+from playwright.sync_api import sync_playwright
+
+def check_huggingface_login():
+    # This assumes your copied Chrome profile is in "./chrome_profile"
+    user_data_dir = os.path.join(os.getcwd(), "chrome_profile")
+    
+    with sync_playwright() as p:
+        # Launch a persistent context using your Chrome profile
+        context = p.chromium.launch_persistent_context(
+            user_data_dir=user_data_dir,
+            headless=False  # Launch in non-headless mode so you can visually inspect the page
+        )
+        page = context.new_page()
+        
+        # Navigate to Hugging Face homepage
+        page.goto("https://huggingface.co")
+        page.wait_for_load_state("networkidle")
+        
+        print("Browser opened. Please inspect the page to verify that you are logged in (e.g., look for your profile avatar or username).")
+        input("Press Enter to close the browser once you've confirmed the login status...")
+        
+        context.close()
+
+if __name__ == "__main__":
+    check_huggingface_login() 

requirements.txt

@@ -3,4 +3,5 @@ huggingface-hub>=0.19.0
 Pillow>=10.0.0
 pytest>=7.0.0
 pytest-playwright>=0.4.0
+pytest-asyncio>=0.23.0
 playwright>=1.40.0 

save_hf_storage_state.py

@@ -0,0 +1,60 @@
+import asyncio
+import os
+import sys
+from playwright.async_api import async_playwright
+
+async def wait_for_enter():
+    # Print the prompt and flush immediately so it's visible.
+    print("Press Enter to exit...", flush=True)
+    loop = asyncio.get_event_loop()
+    # Run sys.stdin.readline in the default executor so we don't block the event loop.
+    await loop.run_in_executor(None, sys.stdin.readline)
+
+async def save_storage_state():
+    auth_state_path = "auth_state.json"
+    print("\n=== Starting Authentication State Check ===")
+    
+    # Check if auth state exists and print the result
+    has_auth_state = os.path.exists(auth_state_path)
+    if has_auth_state:
+        print(f"✓ Found existing auth_state.json at: {os.path.abspath(auth_state_path)}")
+    else:
+        print("✗ No existing auth_state.json found. Will create new authentication state.")
+    
+    print("\n=== Launching Browser ===")
+    async with async_playwright() as playwright:
+        print("• Launching Chromium...")
+        browser = await playwright.chromium.launch(headless=False)
+        
+        if has_auth_state:
+            print("• Loading stored authentication state...")
+            context = await browser.new_context(storage_state=auth_state_path)
+        else:
+            print("• Starting fresh browser context...")
+            context = await browser.new_context()
+            
+        print("• Opening new page...")
+        page = await context.new_page()
+        print("• Navigating to Hugging Face...")
+        await page.goto("https://huggingface.co")
+        await page.wait_for_load_state("networkidle")
+        
+        print("\n=== Instructions ===")
+        print("1. Please check if you're logged in.")
+        print("2. If not, log in manually in the browser window.")
+        print("3. When ready to exit and save the current state, hit Enter below.\n")
+        
+        # Await the blocking call to wait for the user to press Enter.
+        await wait_for_enter()
+        
+        print("\n=== Saving State ===")
+        await context.storage_state(path=auth_state_path)
+        print(f"✓ Authentication state saved to: {os.path.abspath(auth_state_path)}")
+        
+        print("\n=== Cleaning Up ===")
+        print("• Closing browser...")
+        await browser.close()
+        print("✓ Done!")
+
+if __name__ == "__main__":
+    asyncio.run(save_storage_state()) 

test_local.sh

@@ -54,8 +54,8 @@ if [ $? -eq 0 ]; then
     kill $GRADIO_PID
     
     if [ $E2E_STATUS -eq 0 ]; then
-        echo "All tests passed! Starting Gradio app for development..."
-        python app.py
+        echo "All tests passed!"
+        exit 0
     else
         echo "E2E tests failed! Please fix the issues before running the app."
         exit 1