deepgram

Documentation

Index

• Variables

Variables

var Command = &cobra.Command{
	Use:   "deepgram <audio_file | audio_url>",
	Short: "Generate transcript of an audio file using Deepgram API.",
	Args:  cobra.ExactArgs(1),
	PreRunE: func(cmd *cobra.Command, args []string) error {
		useFile, _ := cmd.Flags().GetBool("from-file")
		useURL, _ := cmd.Flags().GetBool("from-url")

		if !(useFile || useURL) {
			return errors.New("one of --from-file or --from-url must be specified")
		}
		return nil
	},
	RunE: func(cmd *cobra.Command, args []string) error {
		apiKey := viper.GetString("deepgram_api_key")
		if apiKey == "" {
			return errors.New("Deepgram API key not found. Please run 'podscript configure' or set the DEEPGRAM_API_KEY environment variable.")
		}

		folder, _ := cmd.Flags().GetString("path")
		suffix, _ := cmd.Flags().GetString("suffix")
		if folder != "" {
			fi, err := os.Stat(folder)
			if err != nil || !fi.IsDir() {
				return fmt.Errorf("path not found: %s", folder)
			}
		}
		timestamp := time.Now().Format("2006-01-02-150405")
		var filenameSuffix string
		if suffix == "" {
			filenameSuffix = timestamp
		} else {
			filenameSuffix = fmt.Sprintf("%s_%s", timestamp, suffix)
		}
		client.InitWithDefault()

		ctx := context.Background()

		options := &interfaces.PreRecordedTranscriptionOptions{
			Model:       "nova-2",
			SmartFormat: true,
			Punctuate:   true,
			Diarize:     true,
			Utterances:  true,
		}

		c := client.New(apiKey, &interfaces.ClientOptions{})
		dg := prerecorded.New(c)

		useFile, _ := cmd.Flags().GetBool("from-file")
		useURL, _ := cmd.Flags().GetBool("from-url")

		if (useFile && useURL) || (!useFile && !useURL) {

			return errors.New("only one of --from-file or --from-url must be specified")
		}

		var (
			res *api.PreRecordedResponse
			err error
		)
		if useFile {
			var fi fs.FileInfo
			fi, err = os.Stat(args[0])
			if err != nil || fi.IsDir() {
				return fmt.Errorf("invalid file path or URL: %s", args[0])
			}
			res, err = dg.FromFile(ctx, args[0], options)
		} else {
			if !client.IsURL(args[0]) {
				return fmt.Errorf("could not parse URL %s", args[0])
			}
			res, err = dg.FromURL(ctx, args[0], options)
		}

		if err != nil {
			return err
		}

		data, err := json.Marshal(res)
		if err != nil {
			return fmt.Errorf("json.Marshal failed: %w", err)
		}

		jsonFilename := path.Join(folder, fmt.Sprintf("deepgram_api_response_%s.json", filenameSuffix))
		if err = os.WriteFile(jsonFilename, data, 0644); err != nil {
			return fmt.Errorf("failed to write JSON response: %w", err)
		}
		fmt.Printf("wrote raw JSON API response to %s\n", jsonFilename)

		transcriptFilename := path.Join(folder, fmt.Sprintf("deepgram_transcript_%s.txt", filenameSuffix))
		if err = os.WriteFile(transcriptFilename, []byte(res.Results.Channels[0].Alternatives[0].Paragraphs.Transcript), 0644); err != nil {
			return fmt.Errorf("failed to write transcript: %w", err)
		}
		fmt.Printf("wrote transcript to %s\n", transcriptFilename)
		return nil
	},
}