mirror of
https://github.com/nextcloud/richdocuments.git
synced 2025-12-17 21:12:14 +01:00
109 lines
3.9 KiB
PHP
109 lines
3.9 KiB
PHP
<?php
|
|
/**
|
|
* SPDX-FileCopyrightText: 2025 Nextcloud GmbH and Nextcloud contributors
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
namespace OCA\Richdocuments\Service;
|
|
|
|
use League\CommonMark\GithubFlavoredMarkdownConverter;
|
|
use OCA\Richdocuments\AppInfo\Application;
|
|
use OCA\Richdocuments\TaskProcessing\TextToDocumentProvider;
|
|
use OCA\Richdocuments\TaskProcessing\TextToSpreadsheetProvider;
|
|
use OCP\IL10N;
|
|
use OCP\TaskProcessing\Exception\Exception;
|
|
use OCP\TaskProcessing\Exception\PreConditionNotMetException;
|
|
use OCP\TaskProcessing\Exception\UnauthorizedException;
|
|
use OCP\TaskProcessing\Exception\ValidationException;
|
|
use OCP\TaskProcessing\IManager;
|
|
use OCP\TaskProcessing\Task;
|
|
use OCP\TaskProcessing\TaskTypes\TextToText;
|
|
use RuntimeException;
|
|
|
|
class DocumentGenerationService {
|
|
public const TEXT_PROMPT = <<<EOF
|
|
Generate a markdown document with titles, paragraphs, bullet points and tables if needed.
|
|
Write the document in the same language as the description.
|
|
|
|
Here is the document description:
|
|
EOF;
|
|
|
|
public const SPREADSHEET_PROMPT = <<<EOF
|
|
Generate a CSV spreadsheet. Use the semicolon as a column separator.
|
|
Write the CSV content in the same language as the description.
|
|
|
|
Here is the CSV spreadsheet description:
|
|
EOF;
|
|
|
|
public function __construct(
|
|
private IManager $taskProcessingManager,
|
|
private RemoteService $remoteService,
|
|
private IL10N $l10n,
|
|
) {
|
|
}
|
|
|
|
public function generateTextDocument(?string $userId, string $description, string $targetFormat = TextToDocumentProvider::DEFAULT_TARGET_FORMAT, bool $includeWatermark = true) {
|
|
$prompt = self::TEXT_PROMPT;
|
|
$taskInput = $prompt . "\n\n" . $description;
|
|
$markdownContent = $this->runTextToTextTask($taskInput, $userId);
|
|
$converter = new GithubFlavoredMarkdownConverter();
|
|
$markdownContentWithAiNote = $includeWatermark ? $markdownContent . "\n\n" . $this->l10n->t('This document was generated using Artificial Intelligence') : $markdownContent;
|
|
$htmlContent = $converter->convert($markdownContentWithAiNote)->getContent();
|
|
$htmlStream = $this->stringToStream($htmlContent);
|
|
$docxContent = $this->remoteService->convertTo('document.html', $htmlStream, $targetFormat);
|
|
|
|
return $docxContent;
|
|
}
|
|
|
|
public function generateSpreadSheetDocument(?string $userId, string $description, string $targetFormat = TextToSpreadsheetProvider::DEFAULT_TARGET_FORMAT) {
|
|
$prompt = self::SPREADSHEET_PROMPT;
|
|
$taskInput = $prompt . "\n\n" . $description;
|
|
$csvContent = $this->runTextToTextTask($taskInput, $userId);
|
|
$csvStream = $this->stringToStream($csvContent);
|
|
|
|
// Passing these will ensure the CSV is correctly
|
|
// parsed into a spreadsheet
|
|
$conversionOptions = [
|
|
// Sets the input filter to use the following:
|
|
// 44 - , (comma) as the field separator
|
|
// 34 - " (double quote) as the text delimiter
|
|
// 76 - UTF-8 as the character set
|
|
// 1 - Start at line one of the input
|
|
'infilterOptions' => '44,34,76,1',
|
|
];
|
|
|
|
$xlsxContent = $this->remoteService->convertTo('document.csv', $csvStream, $targetFormat, $conversionOptions);
|
|
|
|
return $xlsxContent;
|
|
}
|
|
|
|
private function runTextToTextTask(string $input, ?string $userId): string {
|
|
$task = new Task(
|
|
TextToText::ID,
|
|
['input' => $input],
|
|
Application::APPNAME,
|
|
$userId,
|
|
);
|
|
try {
|
|
$task = $this->taskProcessingManager->runTask($task);
|
|
} catch (PreConditionNotMetException|UnauthorizedException|ValidationException|Exception $e) {
|
|
throw new RuntimeException($e->getMessage(), $e->getCode(), $e);
|
|
}
|
|
$taskOutput = $task->getOutput();
|
|
if ($taskOutput === null) {
|
|
throw new RuntimeException('Task with id ' . $task->getId() . ' does not have any output');
|
|
}
|
|
|
|
/** @var string $taskOutputString */
|
|
$taskOutputString = $taskOutput['output'];
|
|
|
|
return $taskOutputString;
|
|
}
|
|
|
|
private function stringToStream(string $text) {
|
|
$stream = fopen('php://memory', 'r+');
|
|
fwrite($stream, $text);
|
|
rewind($stream);
|
|
return $stream;
|
|
}
|
|
}
|