anki_generator 1.1.0 → 1.4.0

data/lib/anki_generator/server.rb

@@ -0,0 +1,98 @@
+# frozen_string_literal: true
+
+require 'json'
+require 'sinatra/base'
+require_relative 'apkg_exporter'
+require_relative 'client_factory'
+require_relative 'errors'
+require_relative 'importers/markdown'
+require_relative 'importers/yaml'
+
+module AnkiGenerator
+  # Localhost preview-and-export UI for the gem, as a modular Sinatra app.
+  # Thin controllers only: parsing delegates to Importers, export to
+  # ApkgExporter, generation to ClientFactory. The page lives in
+  # server/public/index.html (plain static HTML + vanilla JS).
+  class Server < Sinatra::Base
+    set :root, File.dirname(__FILE__)
+    set :default_provider, nil
+    set :protection, except: :json_csrf
+
+    get '/' do
+      send_file File.expand_path('server/public/index.html', __dir__)
+    end
+
+    post '/parse' do
+      payload = json_payload!
+      format = payload['format'].to_s
+
+      cards = case format
+              when 'markdown'
+                Importers::Markdown.parse(payload['content'].to_s)
+              when 'yaml'
+                Importers::Yaml.parse(payload['content'].to_s)
+              else
+                raise ValidationError, "Unknown format '#{format}' (markdown or yaml)"
+              end
+
+      json_response(cards:)
+    rescue AnkiGenerator::Error => e
+      error_json(e.message)
+    end
+
+    post '/generate' do
+      payload = json_payload!
+      client = ClientFactory.build(
+        provider: payload['provider'] || settings.default_provider,
+        model: empty_to_nil(payload['model']),
+        api_key: empty_to_nil(payload['api_key'])
+      )
+
+      cards = client.generate_multiple_flashcards(
+        topics: [payload['prompt'].to_s],
+        difficulty: payload['difficulty'].to_s.empty? ? 'medium' : payload['difficulty'],
+        count: (payload['count'] || 10).to_i
+      )
+
+      json_response(cards:)
+    rescue AnkiGenerator::Error => e
+      error_json(e.message)
+    end
+
+    post '/export' do
+      payload = json_payload!
+      result = ApkgExporter.export(deck_name: payload['deck_name'].to_s, cards: payload['cards'])
+
+      content_type result[:content_type]
+      headers 'Content-Disposition' => %(attachment; filename="#{result[:filename]}")
+      result[:bytes]
+    rescue AnkiGenerator::Error => e
+      error_json(e.message)
+    end
+
+    private
+
+    # Parses the request body as JSON once. Malformed payloads halt with 400.
+    def json_payload!
+      body = request.body.read
+      JSON.parse(body.empty? ? '{}' : body)
+    rescue JSON::ParserError => e
+      halt 400, { 'Content-Type' => 'application/json' }, { error: e.message }.to_json
+    end
+
+    def json_response(payload)
+      content_type :json
+      payload.to_json
+    end
+
+    def error_json(message, status = 422)
+      content_type :json
+      status status
+      { error: message }.to_json
+    end
+
+    def empty_to_nil(value)
+      value.nil? || value.to_s.empty? ? nil : value
+    end
+  end
+end

data/lib/anki_generator/ui.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module AnkiGenerator
+  # Presentation boundary for user-facing output. Commands and utilities emit
+  # through a UI instance instead of writing to $stdout directly, which keeps
+  # the library quiet when embedded and makes output assertions trivial in tests.
+  class UI
+    def initialize(io = $stdout)
+      @io = io
+    end
+
+    def info(message)
+      @io.puts(message)
+    end
+
+    def success(message)
+      @io.puts("✅ #{message}")
+    end
+
+    def warn(message)
+      @io.puts("⚠️  #{message}")
+    end
+
+    def error(message)
+      @io.puts("❌ #{message}")
+    end
+  end
+end

data/lib/anki_generator/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module AnkiGenerator
+  VERSION = '1.4.0'
+end

data/lib/anki_generator.rb

@@ -1,116 +1,20 @@
 # frozen_string_literal: true
 
-require 'yaml'
-require 'anki2'
-require_relative 'openrouter_client'
-
-# AnkiGenerator -> Class
-class AnkiGenerator
-  attr_accessor :deck_file, :cards, :name, :openrouter_client
-
-  def initialize(deck_file:, name:, api_key: nil, model: 'openai/gpt-3.5-turbo')
-    self.deck_file = deck_file
-    self.name = name
-    self.cards = []
-    self.openrouter_client = OpenRouterClient.new(api_key: api_key, model: model) if api_key || ENV['OPENROUTER_API_KEY']
-
-    fill_cards
-  end
-
-  def fill_cards
-    yaml_content = YAML.load_file(deck_file)
-    
-    # Handle both old format (array of cards) and new format (with ai_generation config)
-    if yaml_content.is_a?(Hash) && yaml_content['ai_generation']
-      process_ai_generation(yaml_content)
-    else
-      self.cards = yaml_content.is_a?(Array) ? yaml_content : []
-    end
-  end
-
-  def process_ai_generation(config, attachments: nil)
-    ai_config = config['ai_generation']
-    existing_cards = config['cards'] || []
-    
-    # Generate AI cards if specified and client is available
-    if ai_config['topics'] && openrouter_client
-      ai_cards = generate_ai_cards(
-        topics: ai_config['topics'],
-        context: ai_config['context'],
-        difficulty: ai_config['difficulty'] || 'medium',
-        count: ai_config['count'] || 5,
-        attachments: attachments
-      )
-      
-      self.cards = existing_cards + ai_cards
-    else
-      self.cards = existing_cards
-    end
-    
-    # Save the generated cards back to YAML for future reference
-    save_generated_cards_to_yaml(config) if ai_config['save_generated'] && openrouter_client
-  end
-
-  def generate_ai_cards(topics:, context: nil, difficulty: 'medium', count: 5, attachments: nil)
-    if topics.is_a?(Array) && topics.length > 1
-      openrouter_client.generate_multiple_flashcards(
-        topics: topics,
-        context: context,
-        difficulty: difficulty,
-        count: count,
-        attachments: attachments
-      )
-    else
-      topic = topics.is_a?(Array) ? topics.first : topics
-      [openrouter_client.generate_flashcard(
-        topic: topic,
-        context: context,
-        difficulty: difficulty,
-        attachments: attachments
-      )]
-    end
-  end
-
-  def save_generated_cards_to_yaml(original_config)
-    output_file = deck_file.gsub('.yaml', '_generated.yaml')
-    
-    updated_config = original_config.dup
-    updated_config['cards'] = cards
-    updated_config['ai_generation']['save_generated'] = false # Prevent recursive generation
-    
-    File.open(output_file, 'w') do |file|
-      file.write(updated_config.to_yaml)
-    end
-    
-    puts "Generated cards saved to: #{output_file}"
-  end
-
-  def generate_apkg(output_path:)
-    deck = Anki2.new({ name: name, output_path: output_path })
-
-    cards.each do |card|
-      deck.add_card(card['front'], card['back'])
-    end
-
-    deck.save
-  end
-
-  def add_card(front:, back:)
-    cards << { 'front' => front, 'back' => back }
-  end
-
-  def sync_with_existing_deck(existing_yaml_file)
-    return unless File.exist?(existing_yaml_file)
-    
-    existing_cards = YAML.load_file(existing_yaml_file)
-    existing_cards = existing_cards.is_a?(Array) ? existing_cards : existing_cards['cards'] || []
-    
-    # Simple deduplication based on front text
-    existing_fronts = existing_cards.map { |card| card['front'] }
-    new_cards = cards.reject { |card| existing_fronts.include?(card['front']) }
-    
-    self.cards = existing_cards + new_cards
-    
-    puts "Synced #{new_cards.length} new cards with existing deck"
-  end
-end
+require_relative 'anki_generator/version'
+require_relative 'anki_generator/errors'
+require_relative 'anki_generator/ui'
+require_relative 'anki_generator/card'
+require_relative 'anki_generator/apkg_schema'
+require_relative 'anki_generator/apkg_writer'
+require_relative 'anki_generator/file_processor'
+require_relative 'anki_generator/prompt_builder'
+require_relative 'anki_generator/card_schema'
+require_relative 'anki_generator/llm_client'
+require_relative 'anki_generator/client_factory'
+require_relative 'anki_generator/anki_connect_client'
+require_relative 'anki_generator/deck_builder'
+require_relative 'anki_generator/apkg_exporter'
+require_relative 'anki_generator/importers/csv'
+require_relative 'anki_generator/importers/markdown'
+require_relative 'anki_generator/importers/yaml'
+require_relative 'anki_generator/server'

data/prompt.txt

@@ -0,0 +1,5 @@
+front: What is the main contribution of the Pattern-defeating Quicksort (pdqsort) paper? back: It introduces a new solution to the Dutch national flag problem, providing a worst-case runtime of O(nk) for inputs with k distinct elements and a hybrid sorting algorithm combining quicksort, insertion sort, and heapsort.
+
+Following that pattern, structure
+
+Generate as many anki cards deeply focused in bioinformatics about the pdf sent. Make as many as you want. be exhaustive

metadata

@@ -1,45 +1,45 @@
 --- !ruby/object:Gem::Specification
 name: anki_generator
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.4.0
 platform: ruby
 authors:
 - Ceb
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-01-01 00:00:00.000000000 Z
+date: 2026-10-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: anki2
+  name: dotenv
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.1.2
+        version: '2.8'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.1.2
+        version: '2.8'
 - !ruby/object:Gem::Dependency
-  name: thor
+  name: json
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.2'
+        version: '2.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '1.2'
+        version: '2.0'
 - !ruby/object:Gem::Dependency
-  name: faraday
+  name: ruby_llm
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
@@ -53,80 +53,93 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '2.0'
 - !ruby/object:Gem::Dependency
-  name: json
+  name: rubyzip
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '2.0'
+        version: '2.3'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '2.0'
+        version: '2.3'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3.0'
 - !ruby/object:Gem::Dependency
-  name: dotenv
+  name: sinatra
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.8'
+        version: '4.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '2.8'
+        version: '4.0'
 - !ruby/object:Gem::Dependency
-  name: minitest
+  name: sqlite3
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '5.25'
-  type: :development
+        version: '1.6'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.6'
+    - - "<"
       - !ruby/object:Gem::Version
-        version: '5.25'
+        version: '3.0'
 - !ruby/object:Gem::Dependency
-  name: rake
+  name: thor
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '13.0'
-  type: :development
+        version: '1.2'
+  type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '13.0'
+        version: '1.2'
 - !ruby/object:Gem::Dependency
-  name: rubocop
+  name: webrick
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '1.0'
-  type: :development
+        version: '1.8'
+  type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '1.0'
-description: A powerful command-line tool that generates Anki flashcard decks (.apkg)
-  from YAML files, direct prompts, or file attachments. Features AI-powered content
-  generation using OpenRouter API with support for multiple models (GPT, Claude, Llama),
-  file and directory attachment processing for context-aware generation, prompt file
-  support, intelligent content filtering, and flexible deck management with sync capabilities.
+        version: '1.8'
+description: A command-line tool that generates Anki flashcard decks (.apkg) from
+  YAML files, direct prompts, or file attachments. Features AI-powered content generation
+  via the provider-agnostic ruby_llm gem (Gemini, OpenAI, Anthropic, OpenRouter, Ollama,
+  and more), file and directory attachment processing for context-aware generation,
+  prompt file support, intelligent content filtering, and flexible deck management
+  with sync capabilities.
 email:
 - ceeb.developer@gmail.com
 executables:
@@ -134,12 +147,55 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
+- ".gitignore"
+- ".rubocop.yml"
+- ".ruby-version"
+- ".tool-versions"
+- CHANGELOG.md
+- Gemfile
+- Makefile
 - README.md
+- Rakefile
+- anki_generator.gemspec
 - bin/anki_generator
-- lib/anki_cli.rb
+- docs/CI_SETUP.md
+- docs/architecture/current-v1.3.0.architecture.json
+- docs/architecture/current-v1.3.0.html
+- docs/architecture/current-v1.3.0.visual-check.json
+- docs/architecture/phase3-proposed.architecture.json
+- docs/architecture/phase3-proposed.html
+- docs/architecture/phase3-proposed.visual-check.json
+- docs/phase3-draft.md
+- examples/example_class.rb
+- examples/manual_cards.yaml
+- examples/study_prompt.txt
+- input/input.yaml.example
 - lib/anki_generator.rb
-- lib/file_processor.rb
-- lib/openrouter_client.rb
+- lib/anki_generator/anki_connect_client.rb
+- lib/anki_generator/apkg_schema.rb
+- lib/anki_generator/apkg_writer.rb
+- lib/anki_generator/card.rb
+- lib/anki_generator/cli.rb
+- lib/anki_generator/client_factory.rb
+- lib/anki_generator/commands/create_ai_template.rb
+- lib/anki_generator/commands/generate_deck.rb
+- lib/anki_generator/commands/generate_yaml.rb
+- lib/anki_generator/commands/import.rb
+- lib/anki_generator/commands/prompt_based.rb
+- lib/anki_generator/commands/prompt_to_deck.rb
+- lib/anki_generator/commands/push.rb
+- lib/anki_generator/commands/serve.rb
+- lib/anki_generator/commands/test_api.rb
+- lib/anki_generator/deck_builder.rb
+- lib/anki_generator/errors.rb
+- lib/anki_generator/file_processor.rb
+- lib/anki_generator/importers/csv.rb
+- lib/anki_generator/importers/markdown.rb
+- lib/anki_generator/prompt_builder.rb
+- lib/anki_generator/server.rb
+- lib/anki_generator/ui.rb
+- lib/anki_generator/version.rb
+- prompt.txt
 homepage: https://github.com/pinkfloydsito/anki_generator
 licenses:
 - MIT
@@ -149,6 +205,7 @@ metadata:
   changelog_uri: https://github.com/pinkfloydsito/anki_generator/blob/main/CHANGELOG.md
   bug_tracker_uri: https://github.com/pinkfloydsito/anki_generator/issues
   documentation_uri: https://github.com/pinkfloydsito/anki_generator/blob/main/README.md
+  rubygems_mfa_required: 'true'
 post_install_message:
 rdoc_options: []
 require_paths:
@@ -157,14 +214,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 2.7.0
+      version: 3.1.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.7
+rubygems_version: 3.5.22
 signing_key:
 specification_version: 4
 summary: AI-powered Anki flashcard generator with file attachment support