feat: Add .env.example configuration file and integrate Redis service into Docker Compose for improved task management. Update desktop application to support document indexing and extraction workflows, enhancing user experience with a new extraction wizard and stepper components. Revise README and documentation for clarity on setup and usage.

2026-07-05 22:39:07 -05:00
parent 9aca926988
commit d95f8f6ad8
70 changed files with 9807 additions and 499 deletions
@@ -0,0 +1,74 @@
+package main
+
+import (
+	"context"
+	"log"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/pys/pdf-count/internal/api"
+	"github.com/pys/pdf-count/internal/config"
+	"github.com/pys/pdf-count/internal/model"
+	"github.com/pys/pdf-count/internal/ocr"
+	"github.com/pys/pdf-count/internal/queue"
+	"github.com/pys/pdf-count/internal/storage"
+	"github.com/pys/pdf-count/internal/store"
+)
+
+func main() {
+	cfg := config.Load()
+	ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
+	defer cancel()
+
+	predictor, err := model.NewPredictor(cfg.ModelPath, cfg.InferencePath)
+	if err != nil {
+		log.Fatalf("model: %v", err)
+	}
+	defer predictor.Close()
+
+	poolSize := cfg.OCRPoolSize
+	if poolSize <= 0 {
+		poolSize = 4
+	}
+	ocrEngine, err := ocr.NewEngine(cfg.OCRLang, poolSize)
+	if err != nil {
+		log.Fatalf("ocr: %v", err)
+	}
+	defer ocrEngine.Close()
+
+	st, err := store.New(cfg.RedisURL, cfg.JobResultTTL)
+	if err != nil {
+		log.Fatalf("redis: %v", err)
+	}
+	defer st.Close()
+
+	stg, err := storage.New(cfg.MinIOEndpoint, cfg.MinIOAccessKey, cfg.MinIOSecretKey, cfg.MinIOBucket, cfg.MinIOUseSSL)
+	if err != nil {
+		log.Fatalf("minio: %v", err)
+	}
+	initCtx, initCancel := context.WithTimeout(ctx, 10*time.Second)
+	if err := stg.EnsureBucket(initCtx); err != nil {
+		initCancel()
+		log.Fatalf("minio bucket: %v", err)
+	}
+	initCancel()
+
+	pub, err := queue.NewPublisher(cfg.NATSURL, cfg.StreamName, cfg.SubjectPrefix)
+	if err != nil {
+		log.Fatalf("nats: %v", err)
+	}
+
+	openAPIPath := os.Getenv("OPENAPI_PATH")
+	if openAPIPath == "" {
+		openAPIPath = "api/openapi.json"
+	}
+
+	srv := api.NewServer(cfg, predictor, ocrEngine, st, stg, pub, api.LoadOpenAPI(openAPIPath))
+	addr := ":" + cfg.Port
+	log.Printf("api listening on %s", addr)
+	if err := api.ListenAndServe(ctx, addr, srv.Router()); err != nil {
+		log.Fatalf("api: %v", err)
+	}
+}