import {
  HttpRequest,
  HttpMethod,
  httpClient,
} from '@activepieces/pieces-common';
import { Property, createAction } from '@activepieces/pieces-framework';
import { openaiAuth } from '../auth';
import FormData from 'form-data';
import mime from 'mime-types';
import { Languages, baseUrl } from '../common/common';

export const transcribeAction = createAction({
  audience: 'human',
  name: 'transcribe',
  displayName: 'Transcribe Audio',
  description: 'Transcribe audio to text using whisper-1 model',
  auth: openaiAuth,
  props: {
    audio: Property.File({
      displayName: 'Audio',
      required: true,
      description: 'Audio file to transcribe',
    }),
    language: Property.StaticDropdown({
      displayName: 'Language of the Audio',
      description: 'Language of the audio file the default is en (English).',
      required: false,
      options: {
        options: Languages,
      },
      defaultValue: 'en',
    }),
  },
  run: async (context) => {
    const fileData = context.propsValue.audio;
    const mimeType = mime.lookup(fileData.extension ? fileData.extension : '');
    let language = context.propsValue.language;
    // if language is not in languages list, default to english
    if (!Languages.some((l) => l.value === language)) {
      language = 'en';
    }

    const form = new FormData();
    form.append('file', fileData.data, {
      filename: fileData.filename,
      contentType: mimeType as string,
    });
    form.append('model', 'whisper-1');
    form.append('language', language);

    const headers = {
      Authorization: `Bearer ${context.auth.secret_text}`,
    };

    const request: HttpRequest = {
      method: HttpMethod.POST,
      url: `${baseUrl}/audio/transcriptions`,
      body: form,
      headers: {
        ...form.getHeaders(),
        ...headers,
      },
    };
    const response = await httpClient.sendRequest(request);
    return response.body;
  },
});
