diff --git a/README.md b/README.md index f915952..4276357 100644 --- a/README.md +++ b/README.md @@ -151,6 +151,8 @@ If you prefer to run the application manually: | `LOG_LEVEL` | The log level for the application (`info`, `debug`, `warn`, `error`). Default is `info`. | No | | `LISTEN_INTERFACE` | The interface paperless-gpt listens to. Default is `:8080` | No | | `WEBUI_PATH` | The path to load static content from. Default is `./web-app/dist` | No | +| `AUTO_GENERATE_TITLE` | Enable/disable title generation when automatically applying suggestions with `paperless-gpt-auto`. Default is `true` | No | +| `AUTO_GENERATE_TAGS` | Enable/disable tag generation when automatically applying suggestions with `paperless-gpt-auto`. Default is `true` | No | **Note:** When using Ollama, ensure that the Ollama server is running and accessible from the paperless-gpt container. diff --git a/main.go b/main.go index 0efad54..91c86a7 100644 --- a/main.go +++ b/main.go @@ -39,6 +39,8 @@ var ( logLevel = strings.ToLower(os.Getenv("LOG_LEVEL")) listenInterface = os.Getenv("LISTEN_INTERFACE") webuiPath = os.Getenv("WEBUI_PATH") + autoGenerateTitle = os.Getenv("AUTO_GENERATE_TITLE") + autoGenerateTags = os.Getenv("AUTO_GENERATE_TAGS") // Templates titleTemplate *template.Template @@ -287,8 +289,8 @@ func (app *App) processAutoTagDocuments() (int, error) { suggestionRequest := GenerateSuggestionsRequest{ Documents: documents, - GenerateTitles: true, - GenerateTags: true, + GenerateTitles: strings.ToLower(autoGenerateTitle) != "false", + GenerateTags: strings.ToLower(autoGenerateTags) != "false", } suggestions, err := app.generateDocumentSuggestions(ctx, suggestionRequest)