Upload 64 files

app/services/browser.py

@@ -177,42 +177,69 @@ class BrowserEngine(threading.Thread):
                 
                 raise
 
-            # Poll until the response stabilises
+            # --- Response Polling (Two-Phase) ---
             last_text = ""
             unchanged_count = 0
             start_polling = asyncio.get_event_loop().time()
+            found_first_char = False
+            
+            print("[PhantomAPI] ⏳ Phase 1: Waiting for first character...")
             
-            # Smart stabilization:
-            # - If text is empty, wait longer to give it time to start typing.
-            # - If text exists, 2 seconds of silence is enough.
             while True:
-                # Safety break: 90 seconds total
-                if asyncio.get_event_loop().time() - start_polling > 90:
+                # hard break: 120 seconds total
+                if asyncio.get_event_loop().time() - start_polling > 120:
                     print("[PhantomAPI] ⚠️ Hard timeout reached.")
                     break
 
+                # Target the actual content div if possible
                 elements = await page.query_selector_all(
                     '[data-message-author-role="assistant"]'
                 )
+                
                 if elements:
-                    current_text = await elements[-1].inner_text()
+                    # Look for markdown/prose inside the assistant bubble
+                    container = elements[-1]
+                    content_el = await container.query_selector(".markdown, .prose")
+                    target = content_el if content_el else container
                     
-                    if current_text == last_text:
-                        unchanged_count += 1
-                    else:
-                        if len(current_text) > len(last_text):
-                            print(f"[PhantomAPI] ⏳ Generating... ({len(current_text)} chars)")
-                        last_text = current_text
-                        unchanged_count = 0
+                    current_text = await target.inner_text()
+                    current_text = current_text.strip()
                     
-                    # Determine threshold
-                    threshold = 20 if len(last_text.strip()) == 0 else 4
+                    # Phase 1: Wait for text to appear
+                    if not found_first_char:
+                        if len(current_text) > 0:
+                            found_first_char = True
+                            print(f"[PhantomAPI] 📢 First character detected! Phase 2: Monitoring stream...")
+                        else:
+                            # If we've waited > 60s for the first char, something is wrong
+                            if asyncio.get_event_loop().time() - start_polling > 60:
+                                print("[PhantomAPI] ❌ Giving up: No text appeared after 60s.")
+                                break
                     
-                    if unchanged_count >= threshold:
-                        break
+                    # Phase 2: Monitor stability
+                    if found_first_char:
+                        if current_text == last_text:
+                            unchanged_count += 1
+                        else:
+                            if len(current_text) > len(last_text):
+                                print(f"[PhantomAPI] ⏳ Generating... ({len(current_text)} chars)")
+                            last_text = current_text
+                            unchanged_count = 0
+                        
+                        # Once text starts, 2 seconds (4 iterations) of silence means done
+                        if unchanged_count >= 4:
+                            break
                 
                 await asyncio.sleep(0.5)
 
+            if not found_first_char:
+                # Fail diagnostics
+                print("[PhantomAPI] ❌ Final check: Assistant bubble was empty.")
+                bubble_html = await page.evaluate(
+                    "() => document.querySelector('[data-message-author-role=\"assistant\"]')?.outerHTML"
+                )
+                print(f"[PhantomAPI] 📍 Bubble HTML: {bubble_html[:500]}")
+
             print(f"[PhantomAPI] ✨ Response complete ({len(last_text)} chars).")
             return last_text.strip()