managarten/apps-archived/maerchenzauber/apps/backend/src/app.controller.ts

import { Controller, Get, Post, Body, Res } from '@nestjs/common';
import { Response } from 'express';
import { AppService } from './app.service';
import { ImageSupabaseService } from './core/services/image-supabase.service';

@Controller()
export class AppController {
	constructor(
		private readonly appService: AppService,
		private readonly imageService: ImageSupabaseService
	) {}

	@Get()
	getHello(): string {
		return this.appService.getHello();
	}

	@Post('test-imagen')
	async testImagen(@Body('prompt') prompt: string, @Res() res: Response) {
		try {
			const result = await this.imageService.generateImage(
				prompt || 'a dog reading a newspaper',
				'test'
			);

			if (result.error) {
				console.error('Image generation error:', result.error);
				return res.status(500).json({ error: result.error.message || 'Failed to generate image' });
			}

			if (!result.data) {
				return res.status(500).json({ error: 'No image data received' });
			}

			return res.status(200).json({ imageUrl: result.data });
		} catch (error) {
			console.error('Unexpected error:', error);
			return res.status(500).json({
				error: error instanceof Error ? error.message : 'Failed to generate image',
			});
		}
	}

	@Post('test-seedream')
	async testSeedream(
		@Body('prompt') prompt: string,
		@Body('characterImageUrl') characterImageUrl: string | undefined,
		@Body('type') type: 'character' | 'story' = 'story',
		@Res() res: Response
	) {
		try {
			console.log('Testing Seedream-4 with:');
			console.log('  - Prompt:', prompt);
			console.log('  - Type:', type);
			console.log('  - Character Image:', characterImageUrl || 'none');

			// Create signed URL if character image provided
			let signedUrl: string | undefined;
			if (characterImageUrl) {
				const signedResult =
					await this.imageService.createSignedUrlForCharacterImage(characterImageUrl);
				if (signedResult.error) {
					console.error('Failed to create signed URL:', signedResult.error);
				} else {
					signedUrl = signedResult.data!;
					console.log('  - Signed URL created successfully');
				}
			}

			// Generate image with appropriate path (determines aspect ratio)
			const path = type === 'character' ? 'test/characters' : 'test/stories';
			const result = await this.imageService.generateImage(
				prompt || 'a friendly dragon in a magical forest',
				path,
				undefined,
				undefined,
				signedUrl
			);

			if (result.error) {
				console.error('Image generation error:', result.error);
				return res.status(500).json({ error: result.error.message || 'Failed to generate image' });
			}

			if (!result.data) {
				return res.status(500).json({ error: 'No image data received' });
			}

			return res.status(200).json({
				imageUrl: result.data,
				aspectRatio: type === 'character' ? '1:1' : '3:4',
				usedCharacterImage: !!signedUrl,
			});
		} catch (error) {
			console.error('Unexpected error:', error);
			return res.status(500).json({
				error: error instanceof Error ? error.message : 'Failed to generate image',
			});
		}
	}
}