| const express = require('express'); |
| const proxy = require('express-http-proxy'); |
| const app = express(); |
| const targetUrl = process.env.OPENAI_BASE |
| const openaiKey = process.env.OPENAI_API_KEY |
| const port = 7860; |
| const baseUrl = getExternalUrl(process.env.SPACE_ID); |
|
|
| app.use('/api', proxy(targetUrl, { |
| proxyReqOptDecorator: (proxyReqOpts, srcReq) => { |
| |
| proxyReqOpts.headers['Authorization'] = 'Bearer '+openaiKey; |
| return proxyReqOpts; |
| }, |
| })); |
|
|
| app.get("/", (req, res) => { |
| res.send(`Error 404`); |
| }); |
|
|
| function getExternalUrl(spaceId) { |
| try { |
| const [username, spacename] = spaceId.split("/"); |
| return `https://${username}-${spacename.replace(/_/g, "-")}.hf.space/api/v1`; |
| } catch (e) { |
| return ""; |
| } |
| } |
|
|
| app.listen(port, () => { |
| console.log(`Reverse proxy server running on ${baseUrl}`); |
| }); |