1#!/usr/bin/env python3
2"""
3Simple NotebookLM Question Interface
4Based on MCP server implementation - simplified without sessions
5 
6Implements hybrid auth approach:
7- Persistent browser profile (user_data_dir) for fingerprint consistency
8- Manual cookie injection from state.json for session cookies (Playwright bug workaround)
9See: https://github.com/microsoft/playwright/issues/36139
10"""
11 
12import argparse
13import sys
14import time
15import re
16from pathlib import Path
17 
18from patchright.sync_api import sync_playwright
19 
20# Add parent directory to path
21sys.path.insert(0, str(Path(__file__).parent))
22 
23from auth_manager import AuthManager
24from notebook_manager import NotebookLibrary
25from config import QUERY_INPUT_SELECTORS, RESPONSE_SELECTORS
26from browser_utils import BrowserFactory, StealthUtils
27 
28 
29# Follow-up reminder (adapted from MCP server for stateless operation)
30# Since we don't have persistent sessions, we encourage comprehensive questions
31FOLLOW_UP_REMINDER = (
32    "\n\nEXTREMELY IMPORTANT: Is that ALL you need to know? "
33    "You can always ask another question! Think about it carefully: "
34    "before you reply to the user, review their original request and this answer. "
35    "If anything is still unclear or missing, ask me another comprehensive question "
36    "that includes all necessary context (since each question opens a new browser session)."
37)
38 
39 
40def ask_notebooklm(question: str, notebook_url: str, headless: bool = True) -> str:
41    """
42    Ask a question to NotebookLM
43 
44    Args:
45        question: Question to ask
46        notebook_url: NotebookLM notebook URL
47        headless: Run browser in headless mode
48 
49    Returns:
50        Answer text from NotebookLM
51    """
52    auth = AuthManager()
53 
54    if not auth.is_authenticated():
55        print("⚠️ Not authenticated. Run: python auth_manager.py setup")
56        return None
57 
58    print(f"💬 Asking: {question}")
59    print(f"📚 Notebook: {notebook_url}")
60 
61    playwright = None
62    context = None
63 
64    try:
65        # Start playwright
66        playwright = sync_playwright().start()
67 
68        # Launch persistent browser context using factory
69        context = BrowserFactory.launch_persistent_context(
70            playwright,
71            headless=headless
72        )
73 
74        # Navigate to notebook
75        page = context.new_page()
76        print("  🌐 Opening notebook...")
77        page.goto(notebook_url, wait_until="domcontentloaded")
78 
79        # Wait for NotebookLM
80        page.wait_for_url(re.compile(r"^https://notebooklm\.google\.com/"), timeout=10000)
81 
82        # Wait for query input (MCP approach)
83        print("  ⏳ Waiting for query input...")
84        query_element = None
85 
86        for selector in QUERY_INPUT_SELECTORS:
87            try:
88                query_element = page.wait_for_selector(
89                    selector,
90                    timeout=10000,
91                    state="visible"  # Only check visibility, not disabled!
92                )
93                if query_element:
94                    print(f"  ✓ Found input: {selector}")
95                    break
96            except:
97                continue
98 
99        if not query_element:
100            print("  ❌ Could not find query input")
101            return None
102 
103        # Type question (human-like, fast)
104        print("  ⏳ Typing question...")
105        
106        # Use primary selector for typing
107        input_selector = QUERY_INPUT_SELECTORS[0]
108        StealthUtils.human_type(page, input_selector, question)
109 
110        # Submit
111        print("  📤 Submitting...")
112        page.keyboard.press("Enter")
113 
114        # Small pause
115        StealthUtils.random_delay(500, 1500)
116 
117        # Wait for response (MCP approach: poll for stable text)
118        print("  ⏳ Waiting for answer...")
119 
120        answer = None
121        stable_count = 0
122        last_text = None
123        deadline = time.time() + 120  # 2 minutes timeout
124 
125        while time.time() < deadline:
126            # Check if NotebookLM is still thinking (most reliable indicator)
127            try:
128                thinking_element = page.query_selector('div.thinking-message')
129                if thinking_element and thinking_element.is_visible():
130                    time.sleep(1)
131                    continue
132            except:
133                pass
134 
135            # Try to find response with MCP selectors
136            for selector in RESPONSE_SELECTORS:
137                try:
138                    elements = page.query_selector_all(selector)
139                    if elements:
140                        # Get last (newest) response
141                        latest = elements[-1]
142                        text = latest.inner_text().strip()
143 
144                        if text:
145                            if text == last_text:
146                                stable_count += 1
147                                if stable_count >= 3:  # Stable for 3 polls
148                                    answer = text
149                                    break
150                            else:
151                                stable_count = 0
152                                last_text = text
153                except:
154                    continue
155 
156            if answer:
157                break
158 
159            time.sleep(1)
160 
161        if not answer:
162            print("  ❌ Timeout waiting for answer")
163            return None
164 
165        print("  ✅ Got answer!")
166        # Add follow-up reminder to encourage Claude to ask more questions
167        return answer + FOLLOW_UP_REMINDER
168 
169    except Exception as e:
170        print(f"  ❌ Error: {e}")
171        import traceback
172        traceback.print_exc()
173        return None
174 
175    finally:
176        # Always clean up
177        if context:
178            try:
179                context.close()
180            except:
181                pass
182 
183        if playwright:
184            try:
185                playwright.stop()
186            except:
187                pass
188 
189 
190def main():
191    parser = argparse.ArgumentParser(description='Ask NotebookLM a question')
192 
193    parser.add_argument('--question', required=True, help='Question to ask')
194    parser.add_argument('--notebook-url', help='NotebookLM notebook URL')
195    parser.add_argument('--notebook-id', help='Notebook ID from library')
196    parser.add_argument('--show-browser', action='store_true', help='Show browser')
197 
198    args = parser.parse_args()
199 
200    # Resolve notebook URL
201    notebook_url = args.notebook_url
202 
203    if not notebook_url and args.notebook_id:
204        library = NotebookLibrary()
205        notebook = library.get_notebook(args.notebook_id)
206        if notebook:
207            notebook_url = notebook['url']
208        else:
209            print(f"❌ Notebook '{args.notebook_id}' not found")
210            return 1
211 
212    if not notebook_url:
213        # Check for active notebook first
214        library = NotebookLibrary()
215        active = library.get_active_notebook()
216        if active:
217            notebook_url = active['url']
218            print(f"📚 Using active notebook: {active['name']}")
219        else:
220            # Show available notebooks
221            notebooks = library.list_notebooks()
222            if notebooks:
223                print("\n📚 Available notebooks:")
224                for nb in notebooks:
225                    mark = " [ACTIVE]" if nb.get('id') == library.active_notebook_id else ""
226                    print(f"  {nb['id']}: {nb['name']}{mark}")
227                print("\nSpecify with --notebook-id or set active:")
228                print("python scripts/run.py notebook_manager.py activate --id ID")
229            else:
230                print("❌ No notebooks in library. Add one first:")
231                print("python scripts/run.py notebook_manager.py add --url URL --name NAME --description DESC --topics TOPICS")
232            return 1
233 
234    # Ask the question
235    answer = ask_notebooklm(
236        question=args.question,
237        notebook_url=notebook_url,
238        headless=not args.show_browser
239    )
240 
241    if answer:
242        print("\n" + "=" * 60)
243        print(f"Question: {args.question}")
244        print("=" * 60)
245        print()
246        print(answer)
247        print()
248        print("=" * 60)
249        return 0
250    else:
251        print("\n❌ Failed to get answer")
252        return 1
253 
254 
255if __name__ == "__main__":
256    sys.exit(main())
257