import SwiftUI @main struct GhibliGenApp: App { var body: some Scene { WindowGroup { ContentView() } } } import SwiftUI struct ContentView: View { @StateObject private var viewModel = ContentViewModel() var body: some View { VStack(spacing: 20) { Text("Ghibli Style Image Generator") .font(.title) if let image = viewModel.generatedImage { Image(uiImage: image) .resizable() .scaledToFit() .frame(maxWidth: 300, maxHeight: 300) .border(Color.gray) } else { Rectangle() .fill(Color.gray.opacity(0.1)) .frame(maxWidth: 300, maxHeight: 300) .border(Color.gray) .overlay { if viewModel.isLoading { ProgressView() } else { Text("Enter a prompt below") .foregroundColor(.secondary) } } } TextField("Enter image prompt...", text: $viewModel.prompt) .textFieldStyle(.roundedBorder) .padding(.horizontal) Button("Generate Image") { viewModel.generateImage() } .buttonStyle(.borderedProminent) .disabled(viewModel.isLoading || viewModel.prompt.isEmpty) Spacer() } .padding() .alert("Error", isPresented: $viewModel.showErrorAlert) { Button("OK") {} } message: { Text(viewModel.errorMessage) } } } struct ContentView_Previews: PreviewProvider { static var previews: some View { ContentView() } } import SwiftUI import Combine @MainActor class ContentViewModel: ObservableObject { @Published var prompt: String = "" @Published var generatedImage: UIImage? @Published var isLoading: Bool = false @Published var errorMessage: String = "" @Published var showErrorAlert: Bool = false private let geminiService = GeminiService() private var cancellables = Set() func generateImage() { guard !prompt.isEmpty else { return } isLoading = true generatedImage = nil // Clear previous image errorMessage = "" showErrorAlert = false // Append the style requirement let fullPrompt = "\(prompt), in the style of Ghibli animation" Task { do { let imageData = try await geminiService.generateImage(prompt: fullPrompt) self.generatedImage = UIImage(data: imageData) self.isLoading = false } catch { self.errorMessage = "Failed to generate image: \(error.localizedDescription)" self.showErrorAlert = true self.isLoading = false print("Error generating image: \(error)") // Log detailed error } } } } import Foundation import SwiftUI // For UIImage only if returning UIImage directly, otherwise just Foundation // --- Placeholder Structures for Gemini API Request/Response --- // IMPORTANT: Replace these with the actual structures required by the Gemini API struct GeminiImageRequest: Codable { let prompt: String // Add other parameters like size, quality, etc. as needed } struct GeminiImageResponse: Codable { // Assuming the API returns a URL to the image or base64 data let imageUrl: String? // Example: URL to the generated image let imageData: String? // Example: Base64 encoded image data // Add other relevant fields from the actual API response } // --- End Placeholder Structures --- enum GeminiError: Error { case invalidURL case requestFailed(Error) case invalidResponse case imageDataMissing case imageDecodingFailed case apiError(String) // For errors reported by the API itself } class GeminiService { // IMPORTANT: Replace with your actual Gemini API endpoint private let apiEndpoint = URL(string: "YOUR_GEMINI_IMAGE_GENERATION_API_ENDPOINT_HERE")! // IMPORTANT: Securely manage your API Key. // DO NOT hardcode it directly in the source code for production apps. // Consider using environment variables, a configuration file, or secure storage. private let apiKey = "YOUR_GEMINI_API_KEY_HERE" func generateImage(prompt: String) async throws -> Data { guard apiKey != "YOUR_GEMINI_API_KEY_HERE" else { print("ERROR: Please replace 'YOUR_GEMINI_API_KEY_HERE' with your actual Gemini API key in GeminiService.swift") throw GeminiError.apiError("API Key not set.") } guard apiEndpoint.absoluteString != "YOUR_GEMINI_IMAGE_GENERATION_API_ENDPOINT_HERE" else { print("ERROR: Please replace 'YOUR_GEMINI_IMAGE_GENERATION_API_ENDPOINT_HERE' with your actual Gemini API endpoint in GeminiService.swift") throw GeminiError.apiError("API Endpoint not set.") } var request = URLRequest(url: apiEndpoint) request.httpMethod = "POST" request.setValue("application/json", forHTTPHeaderField: "Content-Type") // Add Authorization header - Adjust based on Gemini API requirements request.setValue("Bearer \(apiKey)", forHTTPHeaderField: "Authorization") // Example: Bearer token // --- Create Request Body --- // IMPORTANT: Adjust this based on the actual Gemini API requirements let requestBody = GeminiImageRequest(prompt: prompt) do { request.httpBody = try JSONEncoder().encode(requestBody) } catch { throw GeminiError.requestFailed(error) // Or a more specific encoding error } // --- End Request Body Creation --- // --- Perform Network Request --- let data: Data let response: URLResponse do { (data, response) = try await URLSession.shared.data(for: request) } catch { throw GeminiError.requestFailed(error) } guard let httpResponse = response as? HTTPURLResponse else { throw GeminiError.invalidResponse } // --- Handle Response --- // IMPORTANT: Adjust status code checking and response parsing based on the actual Gemini API guard (200...299).contains(httpResponse.statusCode) else { // Try to decode potential error message from API let errorString = String(data: data, encoding: .utf8) ?? "Unknown API error" print("API Error Response (\(httpResponse.statusCode)): \(errorString)") throw GeminiError.apiError("API returned status code \(httpResponse.statusCode). \(errorString)") } do { let decodedResponse = try JSONDecoder().decode(GeminiImageResponse.self, from: data) // --- Extract Image Data --- // Prioritize base64 data if available, otherwise use the URL if let base64String = decodedResponse.imageData, let imageData = Data(base64Encoded: base64String) { return imageData } else if let imageUrlString = decodedResponse.imageUrl, let imageUrl = URL(string: imageUrlString) { // If only URL is provided, fetch the image data from the URL do { let (imageData, imageResponse) = try await URLSession.shared.data(from: imageUrl) guard let imageHttpResponse = imageResponse as? HTTPURLResponse, (200...299).contains(imageHttpResponse.statusCode) else { throw GeminiError.invalidResponse // Or specific image download error } return imageData } catch { throw GeminiError.requestFailed(error) // Error fetching image from URL } } else { throw GeminiError.imageDataMissing // Neither URL nor data found } // --- End Image Data Extraction --- } catch let decodingError as DecodingError { print("JSON Decoding Error: \(decodingError)") throw GeminiError.invalidResponse // Or more specific decoding error } catch { throw error // Re-throw other errors (like network errors during image download) } // --- End Response Handling --- } }