aichat2md 1.0.1__py3-none-any.whl → 1.1.0__py3-none-any.whl

aichat2md/__init__.py

@@ -1,5 +1,5 @@
 """aichat2md - Convert AI chat conversations to structured Markdown."""
 
-__version__ = "1.0.1"
+__version__ = "1.1.0"
 __author__ = "PlaceNameDay"
 __description__ = "Convert AI chat conversations to structured Markdown"

aichat2md/extractors/playwright_extractor.py

@@ -1,14 +1,51 @@
-"""Extract content from ChatGPT share URLs using Playwright."""
+"""Extract content from AI chat share URLs using Playwright."""
 
 from playwright.sync_api import sync_playwright, TimeoutError as PlaywrightTimeoutError
 
 
+def _detect_platform(url: str) -> str:
+    """
+    Detect platform from URL.
+
+    Args:
+        url: Share URL
+
+    Returns:
+        Platform name: 'doubao', 'gemini', or 'default'
+    """
+    url_lower = url.lower()
+    if 'doubao.com' in url_lower:
+        return 'doubao'
+    elif 'gemini.google.com' in url_lower or 'g.co' in url_lower:
+        return 'gemini'
+    else:
+        return 'default'
+
+
+def _get_wait_time(platform: str) -> int:
+    """
+    Get wait time in milliseconds for platform.
+
+    Args:
+        platform: Platform name from _detect_platform
+
+    Returns:
+        Wait time in milliseconds
+    """
+    wait_times = {
+        'doubao': 3000,
+        'gemini': 5000,
+        'default': 2000
+    }
+    return wait_times.get(platform, 2000)
+
+
 def extract_from_url(url: str, timeout: int = 30000) -> str:
     """
-    Extract text content from ChatGPT share URL.
+    Extract text content from AI chat share URL.
 
     Args:
-        url: ChatGPT share URL (e.g., https://chatgpt.com/share/...)
+        url: Share URL (ChatGPT, Gemini, Doubao, etc.)
         timeout: Page load timeout in milliseconds
 
     Returns:
@@ -21,17 +58,30 @@ def extract_from_url(url: str, timeout: int = 30000) -> str:
     if not url.startswith('http'):
         raise ValueError(f"Invalid URL: {url}")
 
+    # Detect platform and get corresponding wait time
+    platform = _detect_platform(url)
+    wait_time = _get_wait_time(platform)
+
     try:
         with sync_playwright() as p:
             browser = p.chromium.launch(headless=True)
             page = browser.new_page()
 
-            # Navigate and wait for network idle
-            page.goto(url, wait_until='networkidle', timeout=timeout)
+            # Navigate with appropriate wait strategy
+            # Use 'load' for Gemini/Doubao (networkidle may timeout due to ongoing requests)
+            wait_strategy = 'load' if platform in ['gemini', 'doubao'] else 'networkidle'
+            page.goto(url, wait_until=wait_strategy, timeout=timeout)
+
+            # Wait for content to load
+            # Try to wait for main selector (works for ChatGPT)
+            try:
+                page.wait_for_selector('main', timeout=10000)
+            except PlaywrightTimeoutError:
+                # Some platforms may not have 'main' element, continue anyway
+                pass
 
-            # Wait for conversation content to load
-            # ChatGPT share pages typically have conversation in main content area
-            page.wait_for_selector('main', timeout=10000)
+            # Additional wait for dynamic content based on platform
+            page.wait_for_timeout(wait_time)
 
             # Extract plain text from body
             content = page.inner_text('body')

{aichat2md-1.0.1.dist-info → aichat2md-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aichat2md
-Version: 1.0.1
+Version: 1.1.0
 Summary: Convert AI chat conversations to structured Markdown
 Author: PlaceNameDay
 License: MIT
@@ -32,7 +32,7 @@ Convert AI chat conversations to structured Markdown documents.
 
 ## Features
 
-- 🌐 **Extract from URLs** - ChatGPT share links (with JS rendering via Playwright)
+- 🌐 **Extract from URLs** - ChatGPT, Gemini, Doubao share links (with JS rendering via Playwright)
 - 📄 **Extract from webarchive** - Safari .webarchive files (offline mode)
 - 🤖 **Multiple AI backends** - DeepSeek, OpenAI, Groq, or any OpenAI-compatible API
 - 🌍 **Bilingual support** - English/Chinese prompts
@@ -55,6 +55,30 @@ aichat2md https://chatgpt.com/share/xxx
 aichat2md ~/Downloads/chat.webarchive
 ```
 
+## Supported Platforms
+
+- **ChatGPT** - chatgpt.com share links
+- **Gemini** - gemini.google.com or g.co share links
+- **Doubao (豆包)** - doubao.com share links
+- **Webarchive** - Safari exported .webarchive files (any platform)
+
+### Usage Examples
+
+```bash
+# ChatGPT
+aichat2md https://chatgpt.com/share/xxx
+
+# Gemini (supports both long and short URLs)
+aichat2md https://gemini.google.com/share/xxx
+aichat2md https://g.co/gemini/share/xxx
+
+# Doubao
+aichat2md https://www.doubao.com/thread/xxx
+
+# Webarchive file
+aichat2md ~/Downloads/conversation.webarchive
+```
+
 ## Supported AI Backends
 
 - **DeepSeek** (default) - Cost-effective, Chinese service

{aichat2md-1.0.1.dist-info → aichat2md-1.1.0.dist-info}/RECORD

@@ -1,16 +1,16 @@
-aichat2md/__init__.py,sha256=DUbRGVZhGAyn7omeAomaWvlMQOLxhvgJeLhD5LIrNNY,196
+aichat2md/__init__.py,sha256=sxLgNjR13nToItNNkTgYXZAgeKNi2kfzjJGov_P_NBk,196
 aichat2md/cli.py,sha256=bu_lnD85xLB-xKA04iMwj4WgKC0IkkJsHMnsJSA65H4,6905
 aichat2md/config.py,sha256=VO4fA_ByRKVRPa61W3VwIBjMPDsMt3iagFP2NkBSU7U,4351
 aichat2md/structurizer.py,sha256=0v1Hjo9KYcurBEaKJNt4MaqfVHzgEfHmH-KbIPO1Zcg,4213
 aichat2md/extractors/__init__.py,sha256=HzIWd2aZBACnWs2N2pPjIa7vjM-azPz-bqEviN0QgTs,217
-aichat2md/extractors/playwright_extractor.py,sha256=eB3VLogTnv6uYm3DAfT_8t6CmIsyt3SIBo0Slgd7Rc4,1752
+aichat2md/extractors/playwright_extractor.py,sha256=mUfeo8A_NzHgtzTuLvwmBZXwPgvw7H7cSDUup4MS7q4,3066
 aichat2md/extractors/webarchive_extractor.py,sha256=eIZIVzLlBgO41Yzz8EKmjA8Diq3btlQO8S5mljDQWfs,2842
 aichat2md/prompts/__init__.py,sha256=cPdhDyL1QeVhl5gVFYb50zYMi24iGmxz6R_rrVy1-yk,48
 aichat2md/prompts/system_prompt_en.txt,sha256=luB5o84AQOqCkBq0lM3KsrK_yyCO9yaYu8iqgg3lXoY,1488
 aichat2md/prompts/system_prompt_zh.txt,sha256=UxiVgf2kUFp-iXA15nPDKBG7xdQAnhy3q9g5ki7bEPU,1344
-aichat2md-1.0.1.dist-info/licenses/LICENSE,sha256=g3TWU1mkL2Cn4XEm7hRrNHQySEheXc1VVy7cyQoXOyA,1069
-aichat2md-1.0.1.dist-info/METADATA,sha256=PeiizU00Fmlrf_9YAuB_gJ38Vu1ZdhYfvta_7SRbHgY,6290
-aichat2md-1.0.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-aichat2md-1.0.1.dist-info/entry_points.txt,sha256=N_gW2xKLteEm0vGAnhMcJQ6y8uRpOdlG4f477os5VLE,49
-aichat2md-1.0.1.dist-info/top_level.txt,sha256=o9-3lW1WoPj9xi0KCcPJLVRBmkO8lbuNqKq9tk0qnNA,10
-aichat2md-1.0.1.dist-info/RECORD,,
+aichat2md-1.1.0.dist-info/licenses/LICENSE,sha256=g3TWU1mkL2Cn4XEm7hRrNHQySEheXc1VVy7cyQoXOyA,1069
+aichat2md-1.1.0.dist-info/METADATA,sha256=OxLygIoC-ScwCBedgHambY6kwddcBJaYDktXDqSxPfE,6873
+aichat2md-1.1.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+aichat2md-1.1.0.dist-info/entry_points.txt,sha256=N_gW2xKLteEm0vGAnhMcJQ6y8uRpOdlG4f477os5VLE,49
+aichat2md-1.1.0.dist-info/top_level.txt,sha256=o9-3lW1WoPj9xi0KCcPJLVRBmkO8lbuNqKq9tk0qnNA,10
+aichat2md-1.1.0.dist-info/RECORD,,