managarten/maerchenzauber/apps/mobile/server/replicateProxy.js

const express = require('express');
const cors = require('cors');
const fetch = require('node-fetch');
const dotenv = require('dotenv');
const path = require('path');

// Load environment variables from the root .env file
const envPath = path.join(__dirname, '..', '.env');
console.log('Loading .env from:', envPath);
dotenv.config({ path: envPath });

const app = express();
app.use(cors());
app.use(express.json());

const MAERCHENZAUBER_REPLICATE_API_KEY = process.env.MAERCHENZAUBER_REPLICATE_API_KEY;
console.log('Environment variables loaded:', {
  MAERCHENZAUBER_REPLICATE_API_KEY_EXISTS: !!MAERCHENZAUBER_REPLICATE_API_KEY,
  ENV_KEYS: Object.keys(process.env).filter(key => key.includes('REPLICATE') || key.includes('EXPO'))
});

app.post('/api/generate-image', async (req, res) => {
  try {
    const { prompt } = req.body;
    console.log('Received prompt:', prompt);
    
    if (!MAERCHENZAUBER_REPLICATE_API_KEY) {
      throw new Error('Replicate API key is not configured');
    }
    
    // Start the prediction
    const startResponse = await fetch('https://api.replicate.com/v1/models/black-forest-labs/flux-schnell/predictions', {
      method: 'POST',
      headers: {
        'Authorization': `Bearer ${MAERCHENZAUBER_REPLICATE_API_KEY}`,
        'Content-Type': 'application/json',
        'Prefer': 'wait'
      },
      body: JSON.stringify({
        input: {
          prompt,
          go_fast: true,
          megapixels: "1",
          num_outputs: 1,
          aspect_ratio: "1:1",
          output_format: "webp",
          output_quality: 80,
          num_inference_steps: 4
        }
      })
    });

    if (!startResponse.ok) {
      const errorText = await startResponse.text();
      console.error('Replicate API error:', errorText);
      throw new Error(`Failed to start image generation: ${errorText}`);
    }

    const result = await startResponse.json();
    console.log('Generation completed:', result);

    if (result.error) {
      throw new Error(result.error);
    }

    res.json({ imageUrl: result.output[0] });

  } catch (error) {
    console.error('Error generating image:', error);
    res.status(500).json({ error: `Failed to generate image: ${error.message}` });
  }
});

const PORT = process.env.PORT || 3001;
app.listen(PORT, () => {
  console.log(`Server running on port ${PORT}`);
});