initial commit

app/Http/Controllers/GoogleImageGenerationController.php

@@ -0,0 +1,83 @@
+<?php
+
+namespace App\Http\Controllers;
+
+use App\Services\Google\DTO\ImageGenerationRequest;
+use App\Services\Google\DTO\ImageGenerationResponse;
+use App\Services\Google\Error\APIError;
+use App\Services\Google\ImageAspectRatios;
+use App\Services\Google\ImageGenerationService;
+use App\Services\Image\ImageService;
+use Exception;
+use Google\Exception as GoogleException;
+use Illuminate\Http\JsonResponse;
+use Illuminate\Http\Request;
+use Illuminate\Validation\Rule;
+
+class GoogleImageGenerationController
+{
+    public function __construct(
+        private readonly ImageService $imageService,
+        private readonly ImageGenerationService $imageGenerationService,
+    ) {
+    }
+
+    public function generateImage(Request $request): JsonResponse|string
+    {
+        try {
+            $imageGenerationRequest = $this->buildImageGenerationRequest($request);
+            $imageGenerationResponse = $this->imageGenerationService->generateImage($imageGenerationRequest);
+
+            return new JsonResponse(
+                ['generations' => $this->handleGenerationResponse($imageGenerationResponse)]
+            );
+        } catch (GoogleException|Exception|APIError $e) {
+            return new JsonResponse([
+                'error' => $e->getMessage(),
+            ]);
+        }
+    }
+
+    private function buildImageGenerationRequest(Request $request): ImageGenerationRequest
+    {
+        // add nulls for optional fields
+        $request->merge([
+            'aspect_ratio' => $request->input('aspect_ratio'),
+            'negative_prompt' => $request->input('negative_prompt'),
+            'sample_count' => $request->input('sample_count'),
+        ]);
+
+        $values = $request->validate([
+            'prompt' => 'required|string',
+            'aspect_ratio' => [Rule::enum(ImageAspectRatios::class), 'nullable'],
+            'negative_prompt' => 'string|nullable',
+            'sample_count' => 'integer|nullable',
+        ]);
+
+
+        return new ImageGenerationRequest(
+            $values['prompt'],
+            $values['negative_prompt'],
+            $values['sample_count'],
+            $values['aspect_ratio'],
+        );
+    }
+
+    /**
+     * @throws Exception
+     */
+    private function handleGenerationResponse(ImageGenerationResponse $generationResponseData): array
+    {
+        if (isset($generationResponseData->error)) {
+            throw $generationResponseData->error;
+        }
+
+        $generations = [];
+
+        foreach ($generationResponseData->predictions as $prediction) {
+            $savedImage = $this->imageService->saveImage($prediction->bytesBase64Encoded, $prediction->mimeType);
+            $generations[] = $savedImage;
+        }
+        return $generations;
+    }
+}