Flutter PDF处理插件pdf_gemini的使用

Flutter PDF处理插件pdf_gemini的使用

该插件提供了与PDF文件交互、使用Google Gemini API进行聊天以及管理上传文件的简单方法。

特性

• 将PDF文件上传到Gemini
• 使用Google Gemini API对PDF文件进行提示并获取生成响应
• 轻量级且易于集成到任何Flutter应用中

安装

在命令行中运行以下命令：

flutter pub add pdf_gemini

使用

步骤1: 导入包

在Dart文件中导入包：

import 'package:pdf_gemini/pdf_gemini.dart';

步骤2: 提供API密钥

当使用API服务时，实例化服务类并传递您的API密钥。以下是示例代码：

void main() async {
  const geminiApiKey = 'your_api_key_here';
  final genaiClient = GenaiClient(geminiApiKey: geminiApiKey);

  try {
    final testFile = File('');
    final data = await genaiClient.promptDocument(
      'file_name', 
      'pdf', 
      testFile.readAsBytesSync(), 
      'Your prompt',
    );
    print(data.text);
  } catch (e) {
    print('Error: $e');
  }
}

示例说明：

• 将'your_api_key_here'替换为实际的API密钥。
• promptDocument() 方法接受文档和提示输入，并返回生成的响应（文本：text, 引用：sources）。
• 如果请求失败，请确保捕获异常。

测试

为了确保包按预期工作，您可以编写测试以模拟API响应。以下是简单的测试示例：

import 'dart:io';

import 'package:flutter_test/flutter_test.dart';
import 'package:pdf_gemini/pdf_gemini.dart';

void main() {
  group('PDFChatClient', () {
    const geminiApiKey = '';
    final genaiClient = GenaiClient(geminiApiKey: geminiApiKey);

    test('Prompt PDF Test', () async {
      // 创建临时测试文件
      final testFile = File('your_file_path').readAsBytesSync();

      try {
        await genaiClient.promptDocument('file_name', 'pdf', testFile, 'your_prompt');
      } catch (e) {
        fail('Failed: $e');
      }
    });
  });
}

示例代码

以下是完整的示例代码：

// ignore_for_file: avoid_print

import 'dart:io';
import 'package:pdf_gemini/pdf_gemini.dart';
import 'package:pdf_gemini/src/genai_generated_response_model.dart'; // 确保此包包含在pubspec.yaml中

void main() async {
  // 替换为实际的Gemini API密钥
  const String geminiApiKey = 'YOUR_GEMINI_API_KEY';

  // 创建GenaiClient实例
  final genaiClient = GenaiClient(geminiApiKey: geminiApiKey);

  // 示例文件数据（替换为实际文件数据）
  File file = File('' /* 您的文件路径 */);
  String fileName = 'example.pdf';
  String fileType = 'pdf';

  try {
    // 上传文件并获取GenAI文件
    GenaiFile uploadedFile = await genaiClient.genaiFileManager.uploadFile(
      fileName,
      fileType,
      file.readAsBytesSync(),
    );

    print('上传的文件: ${uploadedFile.displayName}');

    // 使用上传的文件生成内容
    String prompt = "能否在这首诗中添加几行？";
    GenaiGeneratedResponseModel response = await genaiClient.promptDocument(
      uploadedFile.displayName,
      uploadedFile.mimeType,
      file.readAsBytesSync(),
      prompt,
    );

    print('生成的文本: ${response.text}');

    if (response.citationUri != null) {
      print('引用URI: ${response.citationUri}');
    }

    // 获取所有已上传的文件
    List<GenaiFile> uploadedFiles = await genaiClient.genaiFileManager.getUploadedFiles();

    print('已上传的文件:');
    for (var file in uploadedFiles) {
      print('- ${file.displayName} (${file.sizeBytes} 字节)');
    }
  } catch (e) {
    print('错误: $e');
  }
}