index.js

const express = require('express');
const { createServer } = require('http');
const { join } = require('path');
const { Server } = require('socket.io');
const { exec } = require('child_process');
const bodyParser = require('body-parser');
const session = require('express-session');
const passport = require('passport');
const LocalStrategy = require('passport-local').Strategy;
const axios = require('axios');

require('dotenv').config();

const app = express();
const PORT = process.env.PORT || 5500;

const server = createServer(app);
const io = new Server(server);

app.use(express.static('public'));
app.use(bodyParser.json());
app.use(session({ secret: 'secret', resave: false, saveUninitialized: false }));
app.use(passport.initialize());
app.use(passport.session());

const users = [
  { id: 1, username: 'user1', password: 'pass1' },
  { id: 2, username: 'user2', password: 'pass2' }
];

passport.use(new LocalStrategy(
  (username, password, done) => {
    const user = users.find(u => u.username === username && u.password === password);
    if (user) {
      return done(null, user);
    } else {
      return done(null, false, { message: 'Incorrect credentials.' });
    }
  }
));

passport.serializeUser((user, done) => {
  done(null, user.id);
});

passport.deserializeUser((id, done) => {
  const user = users.find(u => u.id === id);
  done(null, user);
});

app.post('/login', passport.authenticate('local'), (req, res) => {
  res.send('Logged in');
  console.log("logged in");
});

let prompts = [];
let imageUrls = [];

let replicateApiKey = '';
let openaiApiKey = '';

io.on('connection', (socket) => {
  console.log('A user connected');

  // Listen for the submission of API keys
  socket.on('submit api keys', (data) => {
    console.log('API Keys received:', data);
    replicateApiKey = data.replicateApiKey;
    openaiApiKey = data.openaiApiKey;
  });

  socket.on('submit title', (data) => {
    const command = `python3 generate_title.py "${data}" "${replicateApiKey}"`;
    exec(command, (error, stdout, stderr) => {
      if (error || stderr) {
        console.error('Error executing Python script:', error, stderr);
        io.emit('error', 'Failed to generate title');
        return;
      }
      io.emit('new title', stdout.trim());
      console.log("Title generated by AI:", stdout.trim());
    });
  });

  socket.on('submit bio', (data) => {
  
    const { name, bio } = data;
    const command = `python3 generate_bio.py "${name}" "${bio}" "${replicateApiKey}"`;
    exec(command, (error, stdout, stderr) => {
      if (error || stderr) {
        console.error('Error executing Python script:', error, stderr);
        io.emit('error', 'Failed to generate bio');
        return;
      }
      io.emit('new bio', stdout.trim());
      console.log("Bio:", stdout.trim());
    });
  });

  socket.on('submit header', (data) => {
    io.emit('new header', data);
  });

  socket.on('submit chat', (data) => {
    io.emit('new chat', data);
  });

  socket.on('submit label', (data) => {
  
    const command = `python3 generate_label.py "${data}" "${replicateApiKey}"`;
    exec(command, (error, stdout, stderr) => {
      if (error || stderr) {
        console.error('Error executing Python script:', error, stderr);
        io.emit('error', 'Failed to generate label');
        return;
      }
      io.emit('new label', stdout.trim());
      console.log("Label:", stdout.trim());
    });
  });

  socket.on('submit concept', (prompt) => {
    prompts.push(prompt);
    io.emit('submit concept', prompt);

    const { theme, imagery } = prompt;
    
    const command_idea = `python3 generate_conceptualart.py "${theme}" "${imagery}" "${replicateApiKey}" "${openaiApiKey}"`;
    const command_idea2 = `python3 generate_conceptualart2.py "${theme}" "${imagery}" "${replicateApiKey}" "${openaiApiKey}"`;

    exec(command_idea, (error, stdout, stderr) => {
      if (error || stderr) {
        console.error('Error executing Python script:', error, stderr);
        io.emit('error', 'Failed to generate image');
        return;
      }
      try {
        const outputParts = stdout.split('\n');
        const url = outputParts[0];
        const titleLine = outputParts.find(line => line.startsWith('Title:'));
        const title = titleLine ? titleLine.split('Title:')[1].trim().replace(/^"|"$/g, '') : 'No title provided';
        io.emit("new label and article title", title);
        
        const descriptionStartIndex = outputParts.findIndex(line => line.startsWith('Description:')) + 1;
        let description = outputParts.slice(descriptionStartIndex).join('\n').trim();
        
        // Use a regular expression to remove text between [ and ]
        description = description.replace(/\[.*?\]/g, '').trim();
        
        console.log("Description:", description);
        io.emit('new description', description);
        

        const matches = url.match(/https:\/\/[^"]+/);
        if (matches && matches[0]) {
          imageUrls.push(matches[0]);
          io.emit('new image', matches[0]);
        } else {
          io.emit('error', 'No image URL found');
          console.error('No URL found in Python script output:', stdout);
        }
      } catch (err) {
        console.error('Error processing output:', err);
        io.emit('error', 'Error processing image data');
      }
    });

    exec(command_idea2, (error, stdout, stderr) => {
      if (error || stderr) {
        console.error('Error executing Python script:', error, stderr);
        io.emit('error', 'Failed to generate image');
        return;
      }
      try {
        const outputParts2 = stdout.split('\n');
        const url2 = outputParts2[0];

        const titleLine2 = outputParts2.find(line => line.startsWith('Title:'));
        const title2 = titleLine2 ? titleLine2.split('Title:')[1].trim().replace(/^"|"$/g, '') : 'No title provided';

        const descriptionStartIndex2 = outputParts2.findIndex(line => line.startsWith('Description:')) + 1;
        const description2 = outputParts2.slice(descriptionStartIndex2).join('\n').trim();

        console.log("Title2:", title2);

        const matches2 = url2.match(/https:\/\/[^"]+/);
        if (matches2 && matches2[0]) {
          io.emit('new image 2', matches2[0]);
        } else {
          console.error('No URL found in Python script output:', stdout);
          io.emit('error', 'No image URL found');
        }
      } catch (err) {
        console.error('Error processing output:', err);
        io.emit('error', 'Error processing image data');
      }
    });
  });

});

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


server.listen(PORT, () => {
  console.log(`Server running at http://localhost:${PORT}`);
});