aichat2md 1.0.1__tar.gz → 1.1.0__tar.gz

{aichat2md-1.0.1 → aichat2md-1.1.0}/PKG-INFO

@@ -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 → aichat2md-1.1.0}/README.md

@@ -4,7 +4,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
@@ -27,6 +27,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 → aichat2md-1.1.0}/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-1.1.0/aichat2md/extractors/playwright_extractor.py

@@ -0,0 +1,108 @@
+"""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 AI chat share URL.
+
+    Args:
+        url: Share URL (ChatGPT, Gemini, Doubao, etc.)
+        timeout: Page load timeout in milliseconds
+
+    Returns:
+        Extracted plain text content
+
+    Raises:
+        PlaywrightTimeoutError: If page fails to load
+        ValueError: If URL is invalid
+    """
+    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 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
+
+            # Additional wait for dynamic content based on platform
+            page.wait_for_timeout(wait_time)
+
+            # Extract plain text from body
+            content = page.inner_text('body')
+
+            browser.close()
+
+            return content.strip()
+
+    except PlaywrightTimeoutError as e:
+        raise PlaywrightTimeoutError(
+            f"Failed to load page within {timeout}ms. "
+            "Check your network connection and URL validity."
+        ) from e
+
+
+if __name__ == "__main__":
+    # Manual test
+    import sys
+    if len(sys.argv) > 1:
+        url = sys.argv[1]
+        print(f"Extracting from: {url}")
+        content = extract_from_url(url)
+        print(f"Extracted {len(content)} characters")
+        print(content[:500])

{aichat2md-1.0.1 → aichat2md-1.1.0}/aichat2md.egg-info/PKG-INFO

@@ -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 → aichat2md-1.1.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aichat2md"
-version = "1.0.1"
+version = "1.1.0"
 description = "Convert AI chat conversations to structured Markdown"
 readme = "README.md"
 requires-python = ">=3.8"

aichat2md-1.0.1/aichat2md/extractors/playwright_extractor.py

@@ -1,58 +0,0 @@
-"""Extract content from ChatGPT share URLs using Playwright."""
-
-from playwright.sync_api import sync_playwright, TimeoutError as PlaywrightTimeoutError
-
-
-def extract_from_url(url: str, timeout: int = 30000) -> str:
-    """
-    Extract text content from ChatGPT share URL.
-
-    Args:
-        url: ChatGPT share URL (e.g., https://chatgpt.com/share/...)
-        timeout: Page load timeout in milliseconds
-
-    Returns:
-        Extracted plain text content
-
-    Raises:
-        PlaywrightTimeoutError: If page fails to load
-        ValueError: If URL is invalid
-    """
-    if not url.startswith('http'):
-        raise ValueError(f"Invalid URL: {url}")
-
-    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)
-
-            # Wait for conversation content to load
-            # ChatGPT share pages typically have conversation in main content area
-            page.wait_for_selector('main', timeout=10000)
-
-            # Extract plain text from body
-            content = page.inner_text('body')
-
-            browser.close()
-
-            return content.strip()
-
-    except PlaywrightTimeoutError as e:
-        raise PlaywrightTimeoutError(
-            f"Failed to load page within {timeout}ms. "
-            "Check your network connection and URL validity."
-        ) from e
-
-
-if __name__ == "__main__":
-    # Manual test
-    import sys
-    if len(sys.argv) > 1:
-        url = sys.argv[1]
-        print(f"Extracting from: {url}")
-        content = extract_from_url(url)
-        print(f"Extracted {len(content)} characters")
-        print(content[:500])