cdpd 0.1.0__py3-none-any.whl

cdpd/__init__.py

@@ -0,0 +1,30 @@
+from .core import (
+    api_key,
+    model,
+    set,
+    set_api_key,
+    set_model,
+    get_model,
+    cdpd,
+    cd,
+    pd,
+)
+
+__all__ = [
+    "api_key",
+    "model",
+    "set",
+    "set_api_key",
+    "set_model",
+    "get_model",
+    "cdpd",
+    "cd",
+    "pd",
+]
+
+
+def load_ipython_extension(ipython):
+    """Hỗ trợ load extension qua %load_ext cdpd."""
+    ipython.register_magic_function(cdpd, magic_kind="line")
+    ipython.register_magic_function(cd, magic_kind="cell")
+    ipython.register_magic_function(pd, magic_kind="cell")

cdpd/core.py

@@ -0,0 +1,165 @@
+import os
+import re
+import httpx
+from openai import OpenAI
+from IPython import get_ipython
+from IPython.core.magic import register_cell_magic, register_line_magic
+
+DEFAULT_MODEL = "gemini-3.1-flash-lite"
+RULE_PEP8 = "Executable Python code only. Zero markdown, comments, docstrings, test cases, or explanation."
+SYSTEM_PROMPT_ANALYST = (
+    "You are an expert Data Analyst. Analyze the provided execution output and code directly "
+    "based on the user's request. Give concise, clear, and to-the-point insights formatted in clean Markdown. "
+    "Focus directly on answering the question accurately without unnecessary fluff."
+)
+
+api_key = None
+model = DEFAULT_MODEL
+_client = None
+
+
+def set(key: str, model_name: str = None):
+    """Cú pháp rút gọn để cấu hình nhanh API key (và model nếu muốn).
+    
+    Ví dụ:
+        cdpd.set("YOUR_API_KEY")
+        cdpd.set("YOUR_API_KEY", "gemini-3.1-flash-lite")
+    """
+    set_api_key(key)
+    if model_name:
+        set_model(model_name)
+
+
+def set_api_key(key: str):
+    """Thiết lập API key cho Gemini / OpenAI client."""
+    global api_key, _client
+    api_key = key
+    _client = None  # Reset client để khởi tạo lại với key mới khi dùng
+
+
+def set_model(model_name: str):
+    """Thiết lập model Gemini mặc định (mặc định là 'gemini-3.1-flash-lite')."""
+    global model
+    if not model_name or not isinstance(model_name, str):
+        raise ValueError("Tên model phải là chuỗi ký tự hợp lệ.")
+    model = model_name.strip()
+
+
+def get_model() -> str:
+    """Lấy tên model hiện tại đang được cấu hình."""
+    return model
+
+
+def _get_client() -> OpenAI:
+    """Lấy hoặc khởi tạo client với API key đã được cấu hình hoặc từ biến môi trường."""
+    global _client
+    if _client is not None:
+        return _client
+
+    key = api_key or os.environ.get("GEMINI_API_KEY") or os.environ.get("GOOGLE_API_KEY")
+    if not key:
+        raise ValueError(
+            "API key chưa được thiết lập! Bạn có thể dùng:\n"
+            "  - cdpd.set('YOUR_API_KEY')\n"
+            "  - cdpd.api_key = 'YOUR_API_KEY'\n"
+            "  - %cdpd YOUR_API_KEY\n"
+            "  - hoặc biến môi trường `GEMINI_API_KEY`."
+        )
+
+    _client = OpenAI(
+        api_key=key,
+        base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
+        http_client=httpx.Client(trust_env=False),
+    )
+    return _client
+
+
+@register_line_magic
+def cdpd(line):
+    """Line magic %cdpd để set nhanh API key trong notebook.
+    
+    Ví dụ:
+        %cdpd YOUR_API_KEY
+        %cdpd YOUR_API_KEY gemini-3.1-flash-lite
+    """
+    parts = line.strip().split()
+    if not parts:
+        print("Vui lòng truyền API key: %cdpd YOUR_API_KEY [MODEL]")
+        return
+    set(parts[0], parts[1] if len(parts) > 1 else None)
+    print("✅ Đã cấu hình cdpd thành công!")
+
+
+@register_cell_magic
+def cd(line, cell):
+    """IPython cell magic %%cd để định dạng và chuẩn hóa code theo chuẩn PEP 8 bằng Gemini."""
+    client = _get_client()
+    model_to_use = line.strip() if line.strip() else model
+
+    res = client.chat.completions.create(
+        model=model_to_use,
+        messages=[
+            {"role": "system", "content": RULE_PEP8},
+            {"role": "user", "content": cell},
+        ],
+        temperature=0.0,
+    )
+    raw = res.choices[0].message.content.strip()
+    clean = re.sub(r"^```(?:python)?\s*|\s*```$", "", raw, flags=re.DOTALL).strip()
+
+    ipython = get_ipython()
+    if ipython is not None:
+        ipython.set_next_input(clean, replace=True)
+    return clean
+
+
+@register_cell_magic
+def pd(line, cell):
+    """IPython cell magic %%pd để đọc output của cell trước và in nhận xét dưới dạng text để copy vào Markdown."""
+    user_prompt = f"{line}\n{cell}".strip()
+    if not user_prompt:
+        user_prompt = "Hãy nhận xét ngắn gọn, súc tích và nêu các điểm chính từ kết quả / số liệu ở trên."
+
+    ipython = get_ipython()
+    last_output = None
+    last_code = None
+
+    if ipython is not None:
+        # Lấy biến `_` (output gần nhất của cell trước)
+        last_output = ipython.user_ns.get("_", None)
+
+        # Lấy code của cell trước đó
+        in_history = ipython.user_ns.get("_ih", [])
+        if len(in_history) >= 2:
+            last_code = in_history[-2]
+
+    context_parts = []
+    if last_code:
+        context_parts.append(f"### Code thực thi ở cell trước:\n```python\n{str(last_code).strip()}\n```")
+
+    if last_output is not None:
+        output_str = str(last_output)
+        if len(output_str) > 10000:
+            output_str = output_str[:10000] + "\n... [output truncated]"
+        context_parts.append(f"### Output / Kết quả nhận được:\n```text\n{output_str}\n```")
+
+    context_content = "\n\n".join(context_parts)
+
+    final_user_message = (
+        f"{context_content}\n\n### Yêu cầu nhận xét / Câu hỏi:\n{user_prompt}"
+        if context_parts
+        else user_prompt
+    )
+
+    client = _get_client()
+    res = client.chat.completions.create(
+        model=model,
+        messages=[
+            {"role": "system", "content": SYSTEM_PROMPT_ANALYST},
+            {"role": "user", "content": final_user_message},
+        ],
+        temperature=0.2,
+    )
+    answer = res.choices[0].message.content.strip()
+    # In ra dạng text bình thường để dễ dàng copy vào cell Markdown
+    print(answer)

cdpd-0.1.0.dist-info/METADATA

@@ -0,0 +1,108 @@
+Metadata-Version: 2.4
+Name: cdpd
+Version: 0.1.0
+Summary: IPython cell magic tool: %%cd for code formatting and %%pd for AI output analysis and insights
+Author-email: Nguyen <skyland1620@gmail.com>
+License: MIT
+Keywords: ipython,jupyter,gemini,pep8,cdpd,data-analysis,ai
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Framework :: IPython
+Classifier: Framework :: Jupyter
+Classifier: Topic :: Software Development :: Quality Assurance
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: openai>=1.0.0
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: ipython>=7.0.0
+Dynamic: license-file
+
+# cdpd
+
+Thư viện IPython / Jupyter Cell Magic hỗ trợ lập trình và phân tích dữ liệu với Google Gemini:
+1. **`%%pd`**: Tự động đọc output/kết quả từ cell trước đó và in nhận xét ngắn gọn, đủ ý dưới dạng văn bản để bạn dễ dàng copy dán vào các cell Markdown báo cáo.
+2. **`%%cd`**: Tự động định dạng, tối ưu và chuẩn hóa code trong cell theo chuẩn PEP 8.
+
+## 🚀 Cài đặt
+
+```bash
+pip install cdpd
+```
+
+## ⚙️ Cấu hình (Cực kỳ ngắn gọn)
+
+Bạn có thể cấu hình nhanh bằng một trong các cách sau:
+
+### Cách 1: Dùng hàm `cdpd.set()` (Khuyến nghị)
+```python
+import cdpd
+
+# Chỉ cần set key:
+cdpd.set("YOUR_GEMINI_API_KEY")
+
+# Hoặc set cả key lẫn model cùng lúc:
+cdpd.set("YOUR_GEMINI_API_KEY", "gemini-3.1-flash-lite")
+```
+
+### Cách 2: Gán biến trực tiếp
+```python
+import cdpd
+
+cdpd.api_key = "YOUR_GEMINI_API_KEY"
+cdpd.model = "gemini-3.1-flash-lite"  # (tùy chọn)
+```
+
+### Cách 3: Dùng Line Magic `%cdpd` trực tiếp trong Notebook
+```python
+import cdpd
+
+%cdpd YOUR_GEMINI_API_KEY
+```
+
+### Cách 4: Qua biến môi trường
+```bash
+export GEMINI_API_KEY="YOUR_GEMINI_API_KEY"
+```
+
+---
+
+## 📖 Hướng dẫn sử dụng
+
+### 1. Phân tích & Nhận xét số liệu (`%%pd`)
+
+Sau khi chạy 1 cell sinh ra số liệu, biểu đồ hoặc DataFrame:
+
+```python
+# [Cell 1 - Code tính toán]
+df.describe()
+```
+
+Mở cell code mới ngay bên dưới để hỏi nhận xét:
+
+```python
+# [Cell 2]
+%%pd
+Nhận xét về độ lệch chuẩn và các giá trị bất thường của bảng số liệu trên.
+```
+
+> 💡 **Kết quả:** Gemini sẽ in nhận xét dạng text thuần. Bạn chỉ cần bôi đen, copy và dán vào cell Markdown để làm báo cáo/thuyết minh.
+
+---
+
+### 2. Chuẩn hóa & Format code (`%%cd`)
+
+Thêm `%%cd` ở dòng đầu của cell code cần format:
+
+```python
+%%cd
+def tinh_tong( a,b ):
+    x=a+b;return x
+```
+
+---
+
+## 📄 License
+
+MIT License. See [LICENSE](LICENSE) for details.

cdpd-0.1.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+cdpd/__init__.py,sha256=XoY4q8h3XokUwBmxRh8bD-LSuonJLUb_fiBYbMDZ_fs,541
+cdpd/core.py,sha256=fz-z5rno6lr-tvbZF8F-Bqrr3tFFdh6NxQNweMn1ffI,5544
+cdpd-0.1.0.dist-info/licenses/LICENSE,sha256=ESYyLizI0WWtxMeS7rGVcX3ivMezm-HOd5WdeOh-9oU,1056
+cdpd-0.1.0.dist-info/METADATA,sha256=Sl8XPKzWRqPCILpmCNmKcRwQ3QTNE3mXfEMF5fx93xY,2925
+cdpd-0.1.0.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+cdpd-0.1.0.dist-info/top_level.txt,sha256=YJHtlzD32ayW1rQwqwoqrs6BB1fCg9Lp_ouoE4S7X-U,5
+cdpd-0.1.0.dist-info/RECORD,,

cdpd-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (84.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

cdpd-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

cdpd-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+cdpd