record_real_demo_final.py

#!/usr/bin/env python3
"""
REAL DEMO: Add API Key + Upload Document + Chat with AI
"""

import asyncio
import os
import shutil
from pathlib import Path
from playwright.async_api import async_playwright

OUTPUT_DIR = Path("docs/images")
GIF_PATH = OUTPUT_DIR / "demo.gif"
FRONTEND_URL = "http://localhost:3002"
SAMPLE_DOC = Path("sample_docs/product_guide.txt").absolute()



async def main():
    OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
    frames_dir = OUTPUT_DIR / "frames"
    if frames_dir.exists():
        shutil.rmtree(frames_dir)
    frames_dir.mkdir()
    
    print("=" * 70)
    print("REAL DEMO: API Key + Document Upload + AI Chat")
    print("=" * 70)
    
    async with async_playwright() as p:
        browser = await p.chromium.launch(headless=False, slow_mo=30)
        ctx = await browser.new_context(viewport={"width": 1280, "height": 720})
        page = await ctx.new_page()
        
        fc = [0]
        async def snap(d="", n=1):
            for i in range(n):
                fc[0] += 1
                await page.screenshot(path=str(frames_dir / f"f_{fc[0]:03d}.png"))
                if i == 0 and d: print(f"  [{fc[0]:2d}] {d}")
                await asyncio.sleep(0.08)
        
        # ===== 1. LANDING =====
        print("\n[1] Landing")
        await page.goto(FRONTEND_URL, timeout=30000)
        await asyncio.sleep(1.5)
        await snap("Info Naut", 3)
        
        # ===== 2. DASHBOARD =====
        print("[2] Dashboard")
        await page.goto(f"{FRONTEND_URL}/dashboard", timeout=30000)
        await asyncio.sleep(1.5)
        await snap("Dashboard", 3)
        
        # ===== 3. API KEYS - ADD THE API KEY =====
        print("[3] API Keys - Adding Key")
        await page.goto(f"{FRONTEND_URL}/dashboard/api-keys", timeout=30000)
        await asyncio.sleep(1.5)
        await snap("API Keys Page", 3)
        
        # Click Create API Key button
        create_btn = page.locator('button:has-text("Create API Key"), button:has-text("Create")')
        if await create_btn.count() > 0:
            await create_btn.first.click()
            await asyncio.sleep(1.5)
            await snap("Create API Key Dialog", 3)
            
            # Find input in dialog - look for visible input
            dialog_input = page.locator('[role="dialog"] input, .dialog input, div[data-state="open"] input')
            if await dialog_input.count() > 0:
                await dialog_input.first.fill("Production API Key")
                await asyncio.sleep(0.5)
                await snap("Key Name Entered", 2)
                
                # Find submit button in dialog
                dialog_submit = page.locator('[role="dialog"] button:has-text("Create"), div[data-state="open"] button:has-text("Create")')
                if await dialog_submit.count() > 0:
                    await dialog_submit.last.click()
                    await asyncio.sleep(2)
                    await snap("API Key Created!", 3)
            else:
                # Close dialog if no input found
                close_btn = page.locator('[aria-label="Close"], button:has-text("Cancel")')
                if await close_btn.count() > 0:
                    await close_btn.first.click()
        
        await asyncio.sleep(1)
        await snap("API Keys", 3)
        
        # ===== 4. CREATE KNOWLEDGE BASE =====
        print("[4] Create Knowledge Base")
        await page.goto(f"{FRONTEND_URL}/dashboard/knowledge/new", timeout=30000)
        await asyncio.sleep(1.5)
        await snap("New KB Form", 2)
        
        # Fill name
        name_input = page.locator('input').first
        if await name_input.count() > 0:
            await name_input.fill("Product Documentation")
            await asyncio.sleep(0.3)
            await snap("KB Name", 2)
        
        # Fill description
        textarea = page.locator('textarea').first
        if await textarea.count() > 0:
            await textarea.fill("Complete product guides and documentation")
            await asyncio.sleep(0.3)
            await snap("KB Description", 2)
        
        # Create
        await page.click('button:has-text("Create")')
        await asyncio.sleep(2)
        await snap("KB Created!", 3)
        
        # ===== 5. UPLOAD DOCUMENT =====
        print("[5] Upload Document")
        await page.goto(f"{FRONTEND_URL}/dashboard/knowledge", timeout=30000)
        await asyncio.sleep(1.5)
        await snap("KB List", 2)
        
        # Click on KB
        kb_link = page.locator('a[href*="/knowledge/"]').first
        if await kb_link.count() > 0:
            await kb_link.click()
            await asyncio.sleep(2)
            await snap("Inside Knowledge Base", 2)
        
        # Upload document
        file_input = page.locator('input[type="file"]')
        if await file_input.count() > 0:
            print("    Uploading product_guide.txt...")
            await file_input.set_input_files(str(SAMPLE_DOC))
            await asyncio.sleep(1)
            await snap("Document Selected", 2)
            
            # Wait for processing
            print("    Processing document...")
            await asyncio.sleep(5)
            await snap("Document Uploaded & Processed!", 4)
        else:
            # Try upload button first
            upload_btn = page.locator('button:has-text("Upload"), button:has-text("Add")')
            if await upload_btn.count() > 0:
                await upload_btn.first.click()
                await asyncio.sleep(1)
                await snap("Upload Dialog", 2)
                
                file_input = page.locator('input[type="file"]')
                if await file_input.count() > 0:
                    await file_input.set_input_files(str(SAMPLE_DOC))
                    print("    Uploading...")
                    await asyncio.sleep(5)
                    await snap("Document Uploaded!", 3)
        
        await asyncio.sleep(2)
        await snap("KB with Document", 3)
        
        # ===== 6. CREATE CHAT =====
        print("[6] Create Chat")
        await page.goto(f"{FRONTEND_URL}/dashboard/chat/new", timeout=30000)
        await asyncio.sleep(2)
        await snap("New Chat", 2)
        
        # Select KB with radio
        radio = page.locator('input[type="radio"]').first
        if await radio.count() > 0:
            await radio.click()
            await asyncio.sleep(0.5)
            await snap("KB Selected for Chat", 2)
        
        # Fill title
        title_input = page.locator('input[placeholder*="title" i]')
        if await title_input.count() > 0:
            await title_input.first.fill("Product Questions")
        
        # Start chat
        start_btn = page.locator('button:has-text("Start"), button[type="submit"]')
        if await start_btn.count() > 0:
            await start_btn.first.click()
            print("    Starting chat...")
            await asyncio.sleep(3)
            await snap("Chat Started!", 3)
        
        # ===== 7. CHAT WITH AI =====
        print("[7] Chat with AI (Real OpenAI)")
        await asyncio.sleep(2)
        await snap("Chat Interface", 3)
        
        # Find message input
        msg_input = page.locator('input[placeholder*="message" i]')
        if await msg_input.count() > 0:
            # FIRST QUESTION
            q1 = "What are the key features of Info Naut?"
            print(f"    Q1: {q1}")
            await msg_input.fill(q1)
            await asyncio.sleep(0.5)
            await snap("First Question", 3)
            
            # Send
            await page.locator('button[type="submit"]').click()
            await snap("Sending...", 2)
            
            print("    Waiting for AI response...")
            await asyncio.sleep(15)
            await snap("AI Response 1!", 6)
            
            await page.evaluate("window.scrollTo(0, 9999)")
            await asyncio.sleep(1)
            await snap("Full AI Response", 4)
            
            # SECOND QUESTION
            print("    Q2: Follow-up question...")
            await asyncio.sleep(2)
            msg_input2 = page.locator('input[placeholder*="message" i]')
            if await msg_input2.count() > 0:
                q2 = "How does the vector search work?"
                await msg_input2.fill(q2)
                await asyncio.sleep(0.5)
                await snap("Second Question", 2)
                
                await page.locator('button[type="submit"]').click()
                print("    Waiting for 2nd response...")
                await asyncio.sleep(12)
                await snap("AI Response 2!", 5)
                
                await page.evaluate("window.scrollTo(0, 9999)")
                await asyncio.sleep(1)
                await snap("Complete Chat", 4)
        
        # ===== 8. FINAL DASHBOARD =====
        print("[8] Final")
        await page.goto(f"{FRONTEND_URL}/dashboard", timeout=30000)
        await asyncio.sleep(1)
        await snap("Complete!", 3)
        
        await browser.close()
        
        print(f"\nTotal Frames: {fc[0]}")
        
        # CREATE GIF
        print("\nCreating GIF...")
        from PIL import Image
        files = sorted(frames_dir.glob("f_*.png"))
        frames = []
        for f in files:
            img = Image.open(f).resize((800, 450), Image.LANCZOS)
            img = img.convert('P', palette=Image.ADAPTIVE, colors=256)
            frames.append(img)
        
        frames[0].save(GIF_PATH, save_all=True, append_images=frames[1:],
                      duration=280, loop=0, optimize=True)
        
        sz = os.path.getsize(GIF_PATH) / (1024*1024)
        print(f"GIF: {sz:.2f} MB ({len(frames)} frames)")
        
        if sz > 5:
            opt = [frames[i] for i in range(0, len(frames), 2)]
            opt[0].save(GIF_PATH, save_all=True, append_images=opt[1:],
                       duration=560, loop=0, optimize=True)
            print(f"Optimized: {os.path.getsize(GIF_PATH)/(1024*1024):.2f} MB")
        
        shutil.rmtree(frames_dir)
        
        print("\n" + "="*70)
        print("SUCCESS! Demo with API Key + Document + Real AI Chat!")
        print("="*70)


if __name__ == "__main__":
    asyncio.run(main())