| | const express = require('express'); |
| | const bodyparser = require('body-parser'); |
| | const txt2image = require("./txt2images") |
| | const fs = require('fs'); |
| |
|
| | |
| | const PORT = 7860; |
| | const HOST = '0.0.0.0'; |
| |
|
| | |
| | const app = express(); |
| | app.use(bodyparser.urlencoded({ extended: true })) |
| | app.post("/txt2pic", async (req, res) => { |
| | var prompt = req.body.prompt; |
| | var size = req.body.size; |
| | var url = await txt2image.txt2img(prompt, size) |
| | res.send(url) |
| | }) |
| | app.get('/', (req, res) => { |
| | res.writeHead(200, { 'Content-Type': 'text/html' }) |
| | var html = fs.readFileSync(__dirname + '/static/index.html', 'utf-8'); |
| | res.end(html); |
| | }); |
| | app.get('/jquery', (req, res) => { |
| | res.writeHead(200, { "Content-Type": "application/javascript" }) |
| | var js = fs.readFileSync(__dirname + '/static/jquery.js', 'utf-8'); |
| | res.end(js); |
| | }); |
| |
|
| | app.listen(PORT, HOST, () => { |
| | if (HOST == '0.0.0.0') { |
| | console.log(`Running on http://127.0.0.1:${PORT}`); |
| | } else { |
| | console.log(`Running on http://${HOST}:${PORT}`); |
| | } |
| |
|
| | }); |