|
|
|
|
|
import express from 'express'; |
|
|
import cors from 'cors'; |
|
|
import multer from 'multer'; |
|
|
import { uploadFile } from '@huggingface/hub'; |
|
|
import path from 'path'; |
|
|
import { fileURLToPath } from 'url'; |
|
|
import dotenv from 'dotenv'; |
|
|
dotenv.config() |
|
|
|
|
|
const __filename = fileURLToPath(import.meta.url); |
|
|
const __dirname = path.dirname(__filename); |
|
|
|
|
|
const app = express(); |
|
|
|
|
|
const port = process.env.PORT || 7860; |
|
|
|
|
|
|
|
|
app.use(cors()); |
|
|
app.use(express.json()); |
|
|
|
|
|
|
|
|
const storage = multer.memoryStorage(); |
|
|
const upload = multer({ storage: storage }); |
|
|
|
|
|
|
|
|
|
|
|
const SERVER_CONFIG = { |
|
|
TOKEN: process.env.HF_TOKEN || '', |
|
|
REPO: process.env.HF_REPO || 'TwanAPI/DataTwan', |
|
|
TYPE: 'dataset' |
|
|
}; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
app.post('/api/upload', upload.single('file'), async (req, res) => { |
|
|
try { |
|
|
const file = req.file; |
|
|
const filePath = req.body.path; |
|
|
|
|
|
if (!file) { |
|
|
return res.status(400).json({ error: 'No file provided' }); |
|
|
} |
|
|
|
|
|
if (!filePath) { |
|
|
return res.status(400).json({ error: 'No file path provided' }); |
|
|
} |
|
|
|
|
|
console.log(`[SERVER] Uploading: ${filePath} (${file.size} bytes)`); |
|
|
|
|
|
|
|
|
const response = await uploadFile({ |
|
|
credentials: { |
|
|
accessToken: SERVER_CONFIG.TOKEN, |
|
|
}, |
|
|
repo: { |
|
|
type: SERVER_CONFIG.TYPE, |
|
|
name: SERVER_CONFIG.REPO |
|
|
}, |
|
|
file: { |
|
|
path: filePath, |
|
|
content: new Blob([file.buffer]), |
|
|
}, |
|
|
}); |
|
|
|
|
|
const commitHash = response.commit.oid; |
|
|
const urlPrefix = "https://huggingface.co/datasets"; |
|
|
const publicUrl = `${urlPrefix}/${SERVER_CONFIG.REPO}/blob/${commitHash}/${filePath}`; |
|
|
|
|
|
console.log(`[SERVER] Success: ${publicUrl}`); |
|
|
|
|
|
res.json({ |
|
|
success: true, |
|
|
url: publicUrl |
|
|
}); |
|
|
|
|
|
} catch (error) { |
|
|
console.error('[SERVER] Upload Error:', error); |
|
|
res.status(500).json({ |
|
|
success: false, |
|
|
error: error.message || 'Internal Server Error' |
|
|
}); |
|
|
} |
|
|
}); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
app.use(express.static(path.join(__dirname, 'dist'))); |
|
|
|
|
|
|
|
|
app.get('*', (req, res) => { |
|
|
res.sendFile(path.join(__dirname, 'dist', 'index.html')); |
|
|
}); |
|
|
|
|
|
|
|
|
app.listen(port, () => { |
|
|
console.log(`--------------------------------------------------`); |
|
|
console.log(`✅ Server running on port ${port}`); |
|
|
console.log(` Target Repo: ${SERVER_CONFIG.REPO}`); |
|
|
console.log(`--------------------------------------------------`); |
|
|
}); |