daily-journal-prompt/simple_generate.py

#!/usr/bin/env python3
"""
Simple Daily Journal Prompt Generator
A lightweight version without rich dependency.
"""

import os
import json
import sys
import argparse
import configparser
from datetime import datetime
from typing import List, Dict, Any

from openai import OpenAI
from dotenv import load_dotenv


class SimplePromptGenerator:
    """Simple version without rich dependency."""

    def __init__(self, config_path: str = ".env"):
        """Initialize the generator with configuration."""
        self.config_path = config_path
        self.client = None
        self.historic_prompts = []
        self.pool_prompts = []
        self.prompt_template = ""
        self.settings = {}

        # Load configuration
        self._load_config()
        self._load_settings()

        # Load data files
        self._load_prompt_template()
        self._load_historic_prompts()
        self._load_pool_prompts()

    def _load_config(self):
        """Load configuration from environment file."""
        load_dotenv(self.config_path)

        # Get API key
        self.api_key = os.getenv("DEEPSEEK_API_KEY") or os.getenv("OPENAI_API_KEY")
        if not self.api_key:
            print("Error: No API key found in .env file")
            print("Please add DEEPSEEK_API_KEY or OPENAI_API_KEY to your .env file")
            sys.exit(1)

        # Get API base URL (default to DeepSeek)
        self.base_url = os.getenv("API_BASE_URL", "https://api.deepseek.com")

        # Get model (default to deepseek-chat)
        self.model = os.getenv("MODEL", "deepseek-chat")

        # Initialize OpenAI client
        self.client = OpenAI(
            api_key=self.api_key,
            base_url=self.base_url
        )

    def _load_settings(self):
        """Load settings from settings.cfg configuration file."""
        config = configparser.ConfigParser()

        # Set default values
        self.settings = {
            'min_length': 500,
            'max_length': 1000,
            'num_prompts': 6
        }

        try:
            config.read('settings.cfg')

            if 'prompts' in config:
                prompts_section = config['prompts']

                # Load min_length
                if 'min_length' in prompts_section:
                    self.settings['min_length'] = int(prompts_section['min_length'])

                # Load max_length
                if 'max_length' in prompts_section:
                    self.settings['max_length'] = int(prompts_section['max_length'])

                # Load num_prompts
                if 'num_prompts' in prompts_section:
                    self.settings['num_prompts'] = int(prompts_section['num_prompts'])

        except FileNotFoundError:
            print("Warning: settings.cfg not found, using default values")
        except ValueError as e:
            print(f"Warning: Invalid value in settings.cfg: {e}, using default values")
        except Exception as e:
            print(f"Warning: Error reading settings.cfg: {e}, using default values")

    def _load_prompt_template(self):
        """Load the prompt template from ds_prompt.txt and update with config values."""
        try:
            with open("ds_prompt.txt", "r") as f:
                template = f.read()

            # Replace hardcoded values with config values
            template = template.replace(
                "between 500 and 1000 characters",
                f"between {self.settings['min_length']} and {self.settings['max_length']} characters"
            )

            # Replace the number of prompts (6) with config value
            template = template.replace(
                "Please generate 6 writing prompts",
                f"Please generate {self.settings['num_prompts']} writing prompts"
            )

            self.prompt_template = template
        except FileNotFoundError:
            print("Error: ds_prompt.txt not found")
            sys.exit(1)

    def _load_historic_prompts(self):
        """Load historic prompts from JSON file."""
        try:
            with open("historic_prompts.json", "r") as f:
                self.historic_prompts = json.load(f)
        except (FileNotFoundError, json.JSONDecodeError):
            print("Warning: Starting with empty prompt history")
            self.historic_prompts = []

    def _save_historic_prompts(self):
        """Save historic prompts to JSON file (keeping only last 60)."""
        # Keep only the last 60 prompts
        if len(self.historic_prompts) > 60:
            self.historic_prompts = self.historic_prompts[-60:]

        with open("historic_prompts.json", "w") as f:
            json.dump(self.historic_prompts, f, indent=2)

    def _prepare_prompt(self) -> str:
        """Prepare the full prompt with historic context."""
        if self.historic_prompts:
            historic_context = json.dumps(self.historic_prompts, indent=2)
            full_prompt = f"{self.prompt_template}\n\nPrevious prompts:\n{historic_context}"
        else:
            full_prompt = self.prompt_template

        return full_prompt

    def _parse_ai_response(self, response_content: str) -> List[str]:
        """Parse the AI response to extract new prompts."""
        try:
            # Try to parse as JSON
            data = json.loads(response_content)

            # Check if data is a list (new format)
            if isinstance(data, list):
                # Return the list of prompt strings directly
                # Ensure we have the correct number of prompts
                if len(data) >= self.settings['num_prompts']:
                    return data[:self.settings['num_prompts']]
                else:
                    print(f"Warning: AI returned {len(data)} prompts, expected {self.settings['num_prompts']}")
                    return data
            elif isinstance(data, dict):
                # Fallback for old format: dictionary with newprompt0, newprompt1, etc.
                print("Warning: AI returned dictionary format, expected list format")
                new_prompts = []
                for i in range(self.settings['num_prompts']):
                    key = f"newprompt{i}"
                    if key in data:
                        new_prompts.append(data[key])
                return new_prompts
            else:
                print(f"Warning: AI returned unexpected data type: {type(data)}")
                return []

        except json.JSONDecodeError:
            # If not valid JSON, try to extract prompts from text
            print("Warning: AI response is not valid JSON, attempting to extract prompts...")

            # Look for patterns in the text
            lines = response_content.strip().split('\n')
            new_prompts = []

            for i, line in enumerate(lines[:self.settings['num_prompts']]):
                line = line.strip()
                if line and len(line) > 50:
                    new_prompts.append(line)

            return new_prompts

    def generate_prompts(self) -> List[str]:
        """Generate new journal prompts using AI."""
        print("\nGenerating new journal prompts...")

        # Prepare the prompt
        full_prompt = self._prepare_prompt()

        try:
            # Call the AI API
            print("Calling AI API...")
            response = self.client.chat.completions.create(
                model=self.model,
                messages=[
                    {"role": "system", "content": "You are a creative writing assistant that generates journal prompts. Always respond with valid JSON."},
                    {"role": "user", "content": full_prompt}
                ],
                temperature=0.7,
                max_tokens=2000
            )

            response_content = response.choices[0].message.content

        except Exception as e:
            print(f"Error calling AI API: {e}")
            return []

        # Parse the response
        new_prompts = self._parse_ai_response(response_content)

        if not new_prompts:
            print("Error: Could not parse any prompts from AI response")
            return []

        # Note: Prompts are NOT added to historic_prompts here
        # They will be added only when the user chooses one

        return new_prompts

    def display_prompts(self, prompts: List[Dict[str, str]]):
        """Display generated prompts in a simple format."""
        print("\n" + "="*60)
        print("✨ NEW JOURNAL PROMPTS GENERATED ✨")
        print("="*60 + "\n")

        for i, prompt_dict in enumerate(prompts, 1):
            # Extract prompt text
            prompt_key = list(prompt_dict.keys())[0]
            prompt_text = prompt_dict[prompt_key]

            print(f"Prompt #{i}:")
            print("-" * 40)
            print(prompt_text)
            print("-" * 40 + "\n")

    def show_history_stats(self):
        """Show statistics about prompt history."""
        total_prompts = len(self.historic_prompts)

        print("\nPrompt History Statistics:")
        print("-" * 30)
        print(f"Total prompts in history: {total_prompts}")
        print(f"History capacity: 60 prompts")
        print(f"Available slots: {max(0, 60 - total_prompts)}")

    def save_prompt_to_file(self, prompt_dict: Dict[str, str], filename: str = None):
        """Save a prompt to a text file."""
        prompt_key = list(prompt_dict.keys())[0]
        prompt_text = prompt_dict[prompt_key]

        if not filename:
            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
            filename = f"journal_prompt_{timestamp}.txt"

        with open(filename, "w") as f:
            f.write(f"Journal Prompt - {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n")
            f.write("="*50 + "\n\n")
            f.write(prompt_text)
            f.write("\n\n" + "="*50 + "\n")
            f.write("Happy writing! ✍️\n")

        print(f"Prompt saved to {filename}")


def main():
    """Main entry point for the simple script."""
    parser = argparse.ArgumentParser(description="Generate journal prompts using AI (simple version)")
    parser.add_argument(
        "--stats", "-s",
        action="store_true",
        help="Show history statistics"
    )
    parser.add_argument(
        "--save", "-S",
        type=int,
        help="Save a specific prompt number to file"
    )
    parser.add_argument(
        "--config", "-c",
        default=".env",
        help="Path to configuration file (default: .env)"
    )

    args = parser.parse_args()

    # Initialize generator
    generator = SimplePromptGenerator(config_path=args.config)

    if args.stats:
        generator.show_history_stats()
    else:
        # Generate prompts
        new_prompts = generator.generate_prompts()
        if new_prompts:
            generator.display_prompts(new_prompts)

            # Save specific prompt if requested
            if args.save:
                prompt_num = args.save
                if 1 <= prompt_num <= len(new_prompts):
                    generator.save_prompt_to_file(new_prompts[prompt_num - 1])
                else:
                    print(f"Error: Prompt number must be between 1 and {len(new_prompts)}")


if __name__ == "__main__":
    main()