File size: 2,184 Bytes
496ef98
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
const path = require('path');
// Load .env from repo root when running locally; on HF, secrets are injected as env vars
require('dotenv').config({ path: path.join(__dirname, '../../.env') });
require('dotenv').config({ path: path.join(__dirname, '.env') }); // local override

const express = require('express');
const captionGenerator = require('./caption_generator');

const app = express();
app.use(express.json());
app.use(express.static(path.join(__dirname, 'public')));

// ── Routes ────────────────────────────────────────────────────────────────────

app.get('/', (req, res) => {
    res.sendFile(path.join(__dirname, 'public', 'index.html'));
});

app.post('/api/generate', async (req, res) => {
    try {
        const { tool, creation, beforeAfter, vibe } = req.body;
        if (!tool || !creation || !beforeAfter || !vibe) {
            return res.status(400).json({ error: 'Missing required fields: tool, creation, beforeAfter, vibe' });
        }
        const result = await captionGenerator.generateAll(tool, creation, beforeAfter, vibe);
        res.json({ ok: true, ...result });
    } catch (e) {
        console.error('Generate error:', e.message);
        res.status(500).json({ error: e.message });
    }
});

app.post('/api/track', (req, res) => {
    try {
        const entry = captionGenerator.trackPost(req.body);
        res.json({ ok: true, entry });
    } catch (e) {
        res.status(500).json({ error: e.message });
    }
});

app.get('/api/insights', (req, res) => {
    try {
        res.json(captionGenerator.getInsights());
    } catch (e) {
        res.status(500).json({ error: e.message });
    }
});

// ── Start ─────────────────────────────────────────────────────────────────────

const PORT = process.env.PORT || process.env.CAPTION_PORT || 3100;
app.listen(PORT, () => {
    console.log(`✅ Caption Generator running at http://localhost:${PORT}`);
});