Улучшен дизайн кнопок SpinBox: добавлены современные стили, улучшена визуальная обратная связь, добавлены SVG иконки, оптимизированы размеры и отступы, улучшена читаемость значени
This commit is contained in:
1 parent
2b9165771b
commit
29dce906c5
6110 files changed
+775979
No files matched your search
@@ -0,0 +1,100 @@
|
||||
const express = require('express');
|
||||
const cors = require('cors');
|
||||
const axios = require('axios');
|
||||
|
||||
const app = express();
|
||||
const port = 3000;
|
||||
|
||||
// Конфигурация API Stable Diffusion
|
||||
const SD_API_URL = 'http://localhost:7860';
|
||||
|
||||
app.use(cors());
|
||||
app.use(express.json());
|
||||
|
||||
// Прокси для txt2img с отслеживанием прогресса
|
||||
app.post('/api/txt2img', async (req, res) => {
|
||||
try {
|
||||
// Отправляем запрос на генерацию
|
||||
const generationPromise = axios.post(`${SD_API_URL}/sdapi/v1/txt2img`, req.body);
|
||||
|
||||
// Начинаем отслеживать прогресс сразу
|
||||
let isGenerating = true;
|
||||
let lastProgress = 0;
|
||||
|
||||
while (isGenerating) {
|
||||
try {
|
||||
const progressResponse = await axios.get(`${SD_API_URL}/sdapi/v1/progress`);
|
||||
const progress = progressResponse.data;
|
||||
|
||||
if (progress.progress >= 1.0) {
|
||||
isGenerating = false;
|
||||
}
|
||||
|
||||
// Если прогресс изменился, обновляем lastProgress
|
||||
if (progress.progress !== lastProgress) {
|
||||
lastProgress = progress.progress;
|
||||
console.log(`Прогресс генерации: ${Math.round(progress.progress * 100)}%`);
|
||||
}
|
||||
|
||||
await new Promise(resolve => setTimeout(resolve, 100));
|
||||
} catch (error) {
|
||||
console.error('Ошибка при получении прогресса:', error);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// Ждем завершения генерации
|
||||
const response = await generationPromise;
|
||||
res.json(response.data);
|
||||
} catch (error) {
|
||||
console.error('Ошибка при запросе к Stable Diffusion API:', error);
|
||||
res.status(500).json({ error: 'Ошибка при генерации изображения' });
|
||||
}
|
||||
});
|
||||
|
||||
// Прокси для проверки прогресса
|
||||
app.get('/api/progress', async (req, res) => {
|
||||
try {
|
||||
const response = await axios.get(`${SD_API_URL}/sdapi/v1/progress`);
|
||||
const progress = response.data;
|
||||
|
||||
// Добавляем дополнительную информацию о прогрессе
|
||||
const result = {
|
||||
...progress,
|
||||
percentage: Math.round(progress.progress * 100),
|
||||
current_step: progress.state?.sampling_step || 0,
|
||||
total_steps: progress.state?.sampling_steps || 1
|
||||
};
|
||||
|
||||
res.json(result);
|
||||
} catch (error) {
|
||||
console.error('Ошибка при получении прогресса:', error);
|
||||
res.status(500).json({ error: 'Ошибка при получении прогресса' });
|
||||
}
|
||||
});
|
||||
|
||||
// Прокси для получения моделей
|
||||
app.get('/api/sd-models', async (req, res) => {
|
||||
try {
|
||||
const response = await axios.get(`${SD_API_URL}/sdapi/v1/sd-models`);
|
||||
res.json(response.data);
|
||||
} catch (error) {
|
||||
console.error('Ошибка при получении списка моделей:', error);
|
||||
res.status(500).json({ error: 'Ошибка при получении списка моделей' });
|
||||
}
|
||||
});
|
||||
|
||||
// Прокси для получения сэмплеров
|
||||
app.get('/api/samplers', async (req, res) => {
|
||||
try {
|
||||
const response = await axios.get(`${SD_API_URL}/sdapi/v1/samplers`);
|
||||
res.json(response.data);
|
||||
} catch (error) {
|
||||
console.error('Ошибка при получении списка сэмплеров:', error);
|
||||
res.status(500).json({ error: 'Ошибка при получении списка сэмплеров' });
|
||||
}
|
||||
});
|
||||
|
||||
app.listen(port, () => {
|
||||
console.log(`Сервер запущен на порту ${port}`);
|
||||
});
|
||||
Reference in new issue
Block a user