Production-Ready Driver's License OCR for Go

package main

import (
	"bytes"
	"encoding/base64"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"os"
)

// 💰 Save 30%+ vs competitors. Get 200 free requests instantly:
// 👉 https://structocr.com/register

// RequestPayload defines the structure for the API request body.
type RequestPayload struct {
	Image string `json:"img"`
}

// DriverLicenseData defines the structure of the extracted data.
type DriverLicenseData struct {
	Type           string `json:"type"`
	CountryCode    string `json:"country_code"`
	Region         string `json:"region"`
	DocumentNumber string `json:"document_number"`
	Surname        string `json:"surname"`
	GivenNames     string `json:"given_names"`
	DateOfBirth    string `json:"date_of_birth"`
	DateOfExpiry   string `json:"date_of_expiry"`
	DateOfIssue    string `json:"date_of_issue"`
	Sex            string `json:"sex"`
	Address        string `json:"address"`
	VehicleClass   string `json:"vehicle_class"`
}

// APIResponse defines the structure of the full API response.
type APIResponse struct {
	Success bool              `json:"success"`
	Data    DriverLicenseData `json:"data"`
	Error   string            `json:"error,omitempty"` // Capture error codes if any
}

func main() {
	// 1. Setup Configuration
	apiURL := "https://api.structocr.com/v1/driver-license"
	apiKey := "YOUR_API_KEY_HERE" // Replace with your actual key
	imagePath := "license.jpg"      // Ensure this file exists

	// 2. Read Image and Encode to Base64
	imageBytes, err := os.ReadFile(imagePath)
	if err != nil {
		fmt.Printf("Error reading image: %v\n", err)
		return
	}

	base64Image := base64.StdEncoding.EncodeToString(imageBytes)

	// 3. Prepare Request Payload
	payload := RequestPayload{
		Image: base64Image,
	}
	payloadBytes, err := json.Marshal(payload)
	if err != nil {
		fmt.Printf("Error marshalling payload: %v\n", err)
		return
	}

	// 4. Create HTTP Request
	req, err := http.NewRequest("POST", apiURL, bytes.NewBuffer(payloadBytes))
	if err != nil {
		fmt.Printf("Error creating request: %v\n", err)
		return
	}

	// Important: Set the correct headers
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("x-api-key", apiKey)

	// 5. Execute Request
	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		fmt.Printf("Network error: %v\n", err)
		return
	}
	defer resp.Body.Close()

	// 6. Parse Response
	body, _ := io.ReadAll(resp.Body)

	if resp.StatusCode != http.StatusOK {
		fmt.Printf("API Error (%d): %s\n", resp.StatusCode, string(body))
		return
	}

	var apiResponse APIResponse
	if err := json.Unmarshal(body, &apiResponse); err != nil {
		fmt.Printf("JSON Parse Error: %v\n", err)
		return
	}

	if apiResponse.Success {
		fmt.Println("✅ Extraction Successful!")
		fmt.Printf("Type: %s\n", apiResponse.Data.Type)
		fmt.Printf("Name: %s %s\n", apiResponse.Data.GivenNames, apiResponse.Data.Surname)
		fmt.Printf("Doc Number: %s\n", apiResponse.Data.DocumentNumber)
		fmt.Printf("Expiry: %s\n", apiResponse.Data.DateOfExpiry)
		fmt.Printf("Vehicle Class: %s\n", apiResponse.Data.VehicleClass)
	} else {
		fmt.Printf("API indicated failure: %s\n", apiResponse.Error)
	}
}