Create colora.js
Browse files
colora.js
ADDED
|
@@ -0,0 +1,67 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
// On définit la classe de ton modèle Colara
|
| 2 |
+
class ColaraModel {
|
| 3 |
+
|
| 4 |
+
// Cette méthode est requise par Hugging Face, mais ne fait rien
|
| 5 |
+
static async getInstance(progress_callback = null) {
|
| 6 |
+
return new ColaraModel();
|
| 7 |
+
}
|
| 8 |
+
|
| 9 |
+
// La méthode 'generate' qui va prendre une image en entrée
|
| 10 |
+
async generate(imageBlob) {
|
| 11 |
+
// On va charger l'image
|
| 12 |
+
const img = await new Promise((resolve) => {
|
| 13 |
+
const reader = new FileReader();
|
| 14 |
+
reader.onload = (e) => {
|
| 15 |
+
const image = new Image();
|
| 16 |
+
image.onload = () => resolve(image);
|
| 17 |
+
image.src = e.target.result;
|
| 18 |
+
};
|
| 19 |
+
reader.readAsDataURL(imageBlob);
|
| 20 |
+
});
|
| 21 |
+
|
| 22 |
+
// Créer un canvas temporaire pour manipuler les pixels de l'image
|
| 23 |
+
const tempCanvas = document.createElement('canvas');
|
| 24 |
+
tempCanvas.width = img.width;
|
| 25 |
+
tempCanvas.height = img.height;
|
| 26 |
+
const tempCtx = tempCanvas.getContext('2d');
|
| 27 |
+
tempCtx.drawImage(img, 0, 0);
|
| 28 |
+
|
| 29 |
+
const imageData = tempCtx.getImageData(0, 0, img.width, img.height);
|
| 30 |
+
const data = imageData.data;
|
| 31 |
+
|
| 32 |
+
// --- Le cœur de la logique de Colara ---
|
| 33 |
+
|
| 34 |
+
// L'IA choisit une couleur de filtre aléatoire
|
| 35 |
+
const randomRed = Math.floor(Math.random() * 256);
|
| 36 |
+
const randomGreen = Math.floor(Math.random() * 256);
|
| 37 |
+
const randomBlue = Math.floor(Math.random() * 256);
|
| 38 |
+
|
| 39 |
+
// Boucler sur chaque pixel pour appliquer le filtre
|
| 40 |
+
for (let i = 0; i < data.length; i += 4) {
|
| 41 |
+
const originalRed = data[i];
|
| 42 |
+
const originalGreen = data[i + 1];
|
| 43 |
+
const originalBlue = data[i + 2];
|
| 44 |
+
|
| 45 |
+
const brightness = 0.2126 * originalRed + 0.7152 * originalGreen + 0.0722 * originalBlue;
|
| 46 |
+
|
| 47 |
+
data[i] = randomRed * (brightness / 255);
|
| 48 |
+
data[i + 1] = randomGreen * (brightness / 255);
|
| 49 |
+
data[i + 2] = randomBlue * (brightness / 255);
|
| 50 |
+
}
|
| 51 |
+
|
| 52 |
+
// Mettre à jour les données de pixels
|
| 53 |
+
tempCtx.putImageData(imageData, 0, 0);
|
| 54 |
+
|
| 55 |
+
// Convertir le canvas en un blob pour le résultat
|
| 56 |
+
const resultBlob = await new Promise(resolve => tempCanvas.toBlob(resolve, 'image/png'));
|
| 57 |
+
|
| 58 |
+
// Le format de sortie pour une image est différent du texte
|
| 59 |
+
return {
|
| 60 |
+
image: resultBlob,
|
| 61 |
+
generated_text: "Image générée par Colara"
|
| 62 |
+
};
|
| 63 |
+
}
|
| 64 |
+
}
|
| 65 |
+
|
| 66 |
+
// On exporte la classe pour qu'elle puisse être utilisée
|
| 67 |
+
export default ColaraModel;
|