aistudio-takeout 0.1.0__py3-none-any.whl

aistudio_takeout/__init__.py

@@ -0,0 +1,74 @@
+#!/usr/bin/env python3
+import sys
+import json
+import re
+from pathlib import Path
+from datetime import datetime
+
+
+def main():
+    if len(sys.argv) < 2 or sys.argv[1] in ["-h", "--help"]:
+        print(
+            "aistudio-takeout - Convert Google AI Studio chats to Markdown\n\n"
+            "Usage: aistudio-takeout <dir> [outdir]\n\n"
+            "Output:\n"
+            "  <outdir>/*.chat.md: Markdown files"
+        )
+        sys.exit(0)
+
+    src = Path(sys.argv[1]).resolve()
+    out = Path(sys.argv[2] if len(sys.argv) > 2 else "aistudio").resolve()
+
+    if not src.exists():
+        sys.exit(f"Error: {src} not found")
+    if not src.is_dir():
+        sys.exit(f"Error: {src} is not a directory")
+
+    print(f"Converting {src.name}...")
+    out.mkdir(exist_ok=True)
+
+    files = [f for f in src.iterdir() if f.is_file() and not f.name.startswith(".")]
+    n = 0
+
+    for f in sorted(files):
+        try:
+            data = json.loads(f.read_text(encoding="utf-8"))
+        except (json.JSONDecodeError, UnicodeDecodeError):
+            continue
+
+        chunks = data.get("chunkedPrompt", {}).get("chunks", [])
+        if not chunks:
+            continue
+
+        settings = data.get("runSettings", {})
+        model = settings.get("model", "unknown").replace("models/", "")
+        mtime = datetime.fromtimestamp(f.stat().st_mtime)
+        created = mtime.strftime("%Y-%m-%dT%H:%M:%S.000Z")
+        date_prefix = mtime.strftime("%Y-%m-%d")
+
+        title = f.stem
+        md = [
+            f"---\nprovider: aistudio\ncreated: {created}\nmodel: {model}\n---\n\n# {title}\n"
+        ]
+
+        for chunk in chunks:
+            role = chunk.get("role", "unknown")
+            text = chunk.get("text", "")
+            if not text and "parts" in chunk:
+                text = chunk["parts"][0].get("text", "") if chunk["parts"] else ""
+            md.append(f"\n## {'User' if role == 'user' else 'Model'}\n\n{text}\n")
+
+        safe = (
+            re.sub(r"[^a-zA-Z0-9\u4e00-\u9fff]+", "-", title).strip("-") or "untitled"
+        )
+        (out / f"{date_prefix}-{safe}.chat.md").write_text(
+            "".join(md), encoding="utf-8"
+        )
+        n += 1
+        print(f"✓ {date_prefix}-{safe}.chat.md")
+
+    print(f"\nDone! {n} chats → {out}/")
+
+
+if __name__ == "__main__":
+    main()

aistudio_takeout-0.1.0.dist-info/METADATA

@@ -0,0 +1,40 @@
+Metadata-Version: 2.3
+Name: aistudio-takeout
+Version: 0.1.0
+Summary: Convert Google AI Studio chats to Markdown
+Requires-Python: >=3.8
+Project-URL: Repository, https://github.com/mefengl/aistudio-takeout
+Description-Content-Type: text/markdown
+
+# aistudio-takeout
+
+Convert Google AI Studio chats to Markdown.
+
+## Install
+
+```bash
+pip install aistudio-takeout
+```
+
+## Usage
+
+```bash
+aistudio-takeout "Google AI Studio"           # Creates aistudio/*.chat.md
+aistudio-takeout "Google AI Studio" output    # Creates output/*.chat.md
+```
+
+Output:
+```
+aistudio/
+├── 2025-01-15-chat-title.chat.md
+├── 2025-01-20-another-chat.chat.md
+└── ...
+```
+
+## Requirements
+
+- Python 3.8+
+
+## License
+
+MIT

aistudio_takeout-0.1.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+aistudio_takeout/__init__.py,sha256=TdcX52sGE2CvEpwMmECi-H9_n46mTyvpRi3jjfdhKu4,2301
+aistudio_takeout-0.1.0.dist-info/WHEEL,sha256=fAguSjoiATBe7TNBkJwOjyL1Tt4wwiaQGtNtjRPNMQA,80
+aistudio_takeout-0.1.0.dist-info/entry_points.txt,sha256=D6W8qRXs4pS6Exx-EiuVFiHqALaFyn9rnmrG3RcKOx8,60
+aistudio_takeout-0.1.0.dist-info/METADATA,sha256=6Tesyda2mVo3Y03N9EyJAJuE_pt62kpounynTVCPygo,706
+aistudio_takeout-0.1.0.dist-info/RECORD,,

aistudio_takeout-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: uv 0.9.28
+Root-Is-Purelib: true
+Tag: py3-none-any

aistudio_takeout-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+aistudio-takeout = aistudio_takeout:main
+