storia 0.4.0__tar.gz

storia-0.4.0/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.3
+Name: storia
+Version: 0.4.0
+Summary: A simple cli tool to help languages by using power of ai stories
+Author: Pratham Das
+Author-email: Pratham Das <prathamdas003@gmail.com>
+Requires-Dist: groq>=1.1.2
+Requires-Python: >=3.13
+Description-Content-Type: text/markdown
+

storia-0.4.0/pyproject.toml

@@ -0,0 +1,27 @@
+[project]
+name = "storia"
+version = "0.4.0"
+description = "A simple cli tool to help languages by using power of ai stories"
+readme = "README.md"
+authors = [
+    { name = "Pratham Das", email = "prathamdas003@gmail.com" }
+]
+requires-python = ">=3.13"
+dependencies = [
+    "groq>=1.1.2",
+]
+
+[build-system]
+requires = ["uv_build>=0.11.3,<0.12.0"]
+build-backend = "uv_build"
+
+[project.scripts]
+storia= "storia.main:main"
+
+[tool.setuptools.package-data]
+"storia.laguages" = ["*.json"]
+
+
+[pypi]
+username= "__token__"
+password="pypi-AgEIcHlwaS5vcmcCJGE3YjFkMjMwLTdlZWEtNGFjMy1iMTFhLTg4OWM2MTQ1MWI1ZQACD1sxLFsic3F1aWxpbyJdXQACLFsyLFsiNmY0NDZmYjItNWU0Yy00MjFhLWFjMGItZDE3NzcxNmJmNzg1Il1dAAAGICLsuNIGwUoYeZXv_qkHyaxSEFb9BQQu-jxpDAlHG0lj"

storia-0.4.0/src/storia/__init__.py

@@ -0,0 +1,12 @@
+from .constant import Mode, DefaultConfig
+from .main import main
+from .helper import config_to_dict, write_config, get_random_topics
+
+__all__ = [
+    "main",
+    "Mode",
+    "DefaultConfig",
+    "config_to_dict",
+    "write_config",
+    "get_random_topics",
+]

storia-0.4.0/src/storia/ai.py

@@ -0,0 +1,79 @@
+from groq import AsyncGroq
+from .constant import Stories
+from .api_key_manager import get_api_key_from_config
+from .prompt import prompt_text
+import sys
+
+
+def get_client() -> AsyncGroq:
+    """Get Groq client with API key from config."""
+    api_key = get_api_key_from_config()
+    return AsyncGroq(api_key=api_key)
+
+
+async def generate_story(topics: list[tuple[str, str]]) -> str:
+    client = get_client()
+    models = ["openai/gpt-oss-20b", "llama-3.3-70b-versatile"]
+
+    last_error = None
+    for model in models:
+        try:
+            words = [word for word, _ in topics]
+            response = await client.chat.completions.create(
+                model=model,
+                messages=[
+                    {"role": "system", "content": prompt_text},
+                    {
+                        "role": "user",
+                        "content": f"Target language: Italian\nWord list: {words}",
+                    },
+                ],
+                temperature=1,
+                max_tokens=4000,
+                top_p=0.95,
+            )
+            story = response.choices[0].message.content
+            if story:
+                return story
+        except Exception as e:
+            last_error = str(e)
+            print(f"Model {model} failed: {e}", file=sys.stderr)
+            continue
+
+    print("Error: All AI models failed to generate a story.", file=sys.stderr)
+    print(f"Last error: {last_error}", file=sys.stderr)
+    print("Please check your API key and try again.", file=sys.stderr)
+    sys.exit(1)
+
+
+def write_story(story: str, date: str) -> None:
+    parts = story.split("\n\n", 1)
+
+    if len(parts) != 2:
+        print("Error: AI returned an unexpected response format.", file=sys.stderr)
+        print(
+            "The story must have two parts separated by a blank line.", file=sys.stderr
+        )
+        print("Please try again.", file=sys.stderr)
+        sys.exit(1)
+
+    target_language_story = parts[0]
+    english_retelling = parts[1]
+    output = (
+        f"--------\n"
+        f"[{date}]\n\n"
+        f"{target_language_story}\n\n"
+        f"{english_retelling}\n"
+        f"--------\n\n"
+    )
+    print(output)
+    try:
+        with Stories.open("a", encoding="utf-8") as f:
+            f.write(output)
+    except FileNotFoundError:
+        print("Error: Could not save story to file.", file=sys.stderr)
+        print(f"Check that the directory exists: {Stories.parent}", file=sys.stderr)
+        sys.exit(1)
+    except Exception as e:
+        print(f"Error: Failed to save story: {e}", file=sys.stderr)
+        sys.exit(1)

storia-0.4.0/src/storia/api_key_manager.py

@@ -0,0 +1,62 @@
+from .constant import config
+import json
+import sys
+
+
+def get_api_key_from_config() -> str:
+    """Read API key from config, raise if missing."""
+    if not config.exists():
+        print("Error: Config file not found.", file=sys.stderr)
+        print("Please run 'storia' to initialize your configuration.", file=sys.stderr)
+        sys.exit(1)
+
+    try:
+        data = json.loads(config.read_text())
+    except json.JSONDecodeError:
+        print("Error: Config file is corrupted.", file=sys.stderr)
+        print("Try running 'storia --reset' to fix this.", file=sys.stderr)
+        sys.exit(1)
+
+    key = data.get("api_key")
+    if not key:
+        print("Error: API key not configured.", file=sys.stderr)
+        print("Please run 'storia' to add your Groq API key.", file=sys.stderr)
+        sys.exit(1)
+    return key
+
+
+def prompt_for_api_key() -> str:
+    """Prompt user for API key with validation."""
+    while True:
+        key = input("Enter your Groq API key: ").strip()
+        if not key:
+            print("API key cannot be empty. Please try again.")
+            continue
+        if len(key) < 10:
+            print("API key seems too short. Please check and try again.")
+            continue
+        return key
+
+
+def get_or_prompt_api_key() -> str:
+    """Get existing API key from config or prompt user if missing."""
+    if config.exists():
+        try:
+            data = json.loads(config.read_text())
+            if data.get("api_key"):
+                return data["api_key"]
+        except (json.JSONDecodeError, FileNotFoundError):
+            pass
+
+    print("API key not found. Please enter your Groq API key.")
+    api_key = prompt_for_api_key()
+
+    if config.exists():
+        try:
+            data = json.loads(config.read_text())
+            data["api_key"] = api_key
+            config.write_text(json.dumps(data, indent=2))
+        except (json.JSONDecodeError, FileNotFoundError):
+            pass
+
+    return api_key

storia-0.4.0/src/storia/constant.py

@@ -0,0 +1,24 @@
+from pathlib import Path
+from importlib.resources import files
+from dataclasses import dataclass
+from enum import Enum
+
+HOME = Path.home()
+DATA_DIR = HOME / ".storia"
+config = DATA_DIR / "config.json"
+Stories = DATA_DIR / "stories.txt"
+content_path = files("storia.languages").joinpath("italian.json")
+
+
+class Mode(Enum):
+    RANDOM = 0
+    TOPIC_BASED = 1
+
+
+@dataclass
+class DefaultConfig:
+    today: str
+    content: list
+    length: int
+    mode: Mode
+    api_key: str | None = None

storia-0.4.0/src/storia/helper.py

@@ -0,0 +1,38 @@
+import json
+import random
+from .constant import config, DATA_DIR, DefaultConfig
+
+
+def config_to_dict(cfg: DefaultConfig) -> dict:
+    return {
+        "today": cfg.today,
+        "content": cfg.content,
+        "length": cfg.length,
+        "mode": cfg.mode.value,
+        "api_key": cfg.api_key,
+    }
+
+
+def write_config(cfg: DefaultConfig, preserve_api_key: bool = True) -> None:
+    if not DATA_DIR.exists():
+        DATA_DIR.mkdir(parents=True, exist_ok=True)
+
+    if preserve_api_key and config.exists():
+        try:
+            existing = json.loads(config.read_text())
+        except (json.JSONDecodeError, FileNotFoundError):
+            existing = {}
+
+        # Update with new values, but only include api_key if it's not None
+        new_data = config_to_dict(cfg)
+        if cfg.api_key is None:
+            new_data.pop("api_key", None)
+        existing.update(new_data)
+    else:
+        existing = config_to_dict(cfg)
+
+    config.write_text(json.dumps(existing, indent=2))
+
+
+def get_random_topics(length: int, n: int = 10) -> list[int]:
+    return [random.randint(0, length - 1) for _ in range(n)]