quantum-pixel 2.0.2__tar.gz → 2.0.4__tar.gz

{quantum_pixel-2.0.2 → quantum_pixel-2.0.4}/Cargo.lock

@@ -773,7 +773,7 @@ dependencies = [
 
 [[package]]
 name = "quantum-pixel"
-version = "2.0.2"
+version = "2.0.4"
 dependencies = [
  "pyo3",
  "stegano-core",

{quantum_pixel-2.0.2 → quantum_pixel-2.0.4}/Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 edition = "2024"
 name = "quantum-pixel"
-version = "2.0.2"
+version = "2.0.4"
 authors = ["Linos1391 <linos.coding@gmail.com>"]
 homepage = "https://github.com/Linos1391/quantum-pixel.git"
 description = "When both YES and NO are existed. Only human with emotions can comprehend."

{quantum_pixel-2.0.2 → quantum_pixel-2.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quantum-pixel
-Version: 2.0.2
+Version: 2.0.4
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Natural Language :: English
 Classifier: Programming Language :: Rust

{quantum_pixel-2.0.2 → quantum_pixel-2.0.4}/quantum_pixel/generator.py

@@ -6,6 +6,7 @@ Generating layers from an image file.
 import logging
 import os
 from collections.abc import Generator as gen
+import asyncio
 from random import shuffle, randint
 
 import numpy as np
@@ -45,28 +46,32 @@ class Generator:
         """
         assert 0 <= intensity <= 1, "Invalid intensity"
 
-        # [[i, j] for i in range(img_data.shape[0]) for j in range(img_data.shape[1])]
-        available_location = np.stack(np.meshgrid(np.arange(self.img_data.shape[0]),
-            np.arange(self.img_data.shape[1])), axis=-1).reshape(-1, 2).tolist()
-        shuffle(available_location)
+        try:
+            # [[i, j] for i in range(img_data.shape[0]) for j in range(img_data.shape[1])]
+            available_location = np.stack(np.meshgrid(np.arange(self.img_data.shape[0]),
+                np.arange(self.img_data.shape[1])), axis=-1).reshape(-1, 2).tolist()
+            shuffle(available_location)
 
-        # do the shit.
-        self._allowance = int(int(np.sum(self.img_data) * intensity))
-        self._remain_allowance = self._allowance
+            # do the shit.
+            self._allowance = int(int(np.sum(self.img_data) * intensity))
+            self._remain_allowance = self._allowance
 
-        layer: np.ndarray = np.zeros_like(self.img_data)
-        while self._remain_allowance > 0:
-            try:
-                location = tuple(available_location.pop())
-            except IndexError:
-                break
-            for current_value in range(3): # RGBA channels
-                value=min(randint(0, self.img_data[location][current_value]),self._remain_allowance)
-                layer[location][current_value] = value
-                self._remain_allowance -= value
-                yield int(100*(1-self._remain_allowance/self._allowance))
-        os.makedirs(os.path.dirname(output_path), exist_ok=True)
-        Image.fromarray(layer.astype(np.uint8), "RGB").save(output_path)
+            layer: np.ndarray = np.zeros_like(self.img_data)
+            while self._remain_allowance > 0:
+                try:
+                    location = tuple(available_location.pop())
+                except IndexError:
+                    break
+                for current_value in range(3): # RGBA channels
+                    value=min (randint(0, self.img_data[location][current_value]),
+                                self._remain_allowance)
+                    layer[location][current_value] = value
+                    self._remain_allowance -= value
+                    yield int(100*(1-self._remain_allowance/self._allowance))
+            os.makedirs(os.path.dirname(output_path), exist_ok=True)
+            Image.fromarray(layer.astype(np.uint8), "RGB").save(output_path)
+        except asyncio.exceptions.CancelledError:
+            yield -1
 
 if __name__ == "__main__":
     generator = Generator("assets/material.png")

{quantum_pixel-2.0.2 → quantum_pixel-2.0.4}/quantum_pixel/static/encode.js

@@ -29,10 +29,12 @@
             result.style.display = "";
             result.textContent = 'Loading, please wait patiently.';
 
+            const TIMEOUT = 300_000; // 5 minutes - NodeJS default anyway.
             try {
                 const r = await fetch(location.pathname, {
                     method: 'POST',
-                    body: fd
+                    body: fd,
+                    signal: AbortSignal.timeout(TIMEOUT)
                 });
 
                 const reader = r.body.getReader();
@@ -47,6 +49,7 @@
 
                             const error = document.getElementById("error");
                             if (json.error) {
+                                error.style.display = "";
                                 error.innerHTML = json.error;
                             } else {
                                 error.style.display = "none";
@@ -74,7 +77,13 @@
                 }
 
             } catch (err) {
-                console.error(err);
+                if (err.name.startsWith("TimeoutError") || err.name.startsWith("AbortError")) {
+                    const error = document.getElementById("error");
+                    error.style.display = "";
+                    error.innerHTML = "TimeoutError: The system has reach timeout of 5 minutes. This might be due to RAM limit exceed, use local system instead.";
+                } else {
+                    console.error(err);
+                }
             }
         });
     })
@@ -87,3 +96,7 @@
         });
     });
 })();
+
+
+// TODO
+// - error did not display initially! --> error.html wraps everything.

{quantum_pixel-2.0.2 → quantum_pixel-2.0.4}/quantum_pixel/templates/decode.html

@@ -21,7 +21,17 @@
                 </form>
             </section>
 
-            <div id="error" style="display: none;">{{ error | safe }}</div>
+            <br>
+
+            {% if error %}
+                <div id="error">
+                    <h2>Error</h2>
+                    <p>{{ error }}</p>
+                    <p><a href="/">Return to Home</a></p>
+                </div>
+            {% else %}
+                <div id="error" style="display: none;"></div>
+            {% endif %}
 
             <div id="result" style="display: none;"></div>
         </main>

{quantum_pixel-2.0.2 → quantum_pixel-2.0.4}/quantum_pixel/templates/encode.html

@@ -15,7 +15,17 @@
                 <img id="uploaded-image" src="/static/images/{{ uid }}/encode_input.png" alt="uploaded image">
             </div>
 
-            <div id="error" style="display: none;">{{ error | safe }}</div>
+            <br>
+
+            {% if error %}
+                <div id="error">
+                    <h2>Error</h2>
+                    <p>{{ error }}</p>
+                    <p><a href="/">Return to Home</a></p>
+                </div>
+            {% else %}
+                <div id="error" style="display: none;"></div>
+            {% endif %}
             
             <section id="options-panel" class="options-panel">
                 <h3>Options</h3>

{quantum_pixel-2.0.2 → quantum_pixel-2.0.4}/quantum_pixel/web.py

@@ -6,6 +6,7 @@ import os
 import time
 import asyncio
 import json
+import types
 from concurrent.futures.thread import ThreadPoolExecutor
 from contextlib import asynccontextmanager
 
@@ -120,9 +121,8 @@ async def start_encode(request: Request, uid: str):
     """Start the encode html."""
     if not os.path.exists(_join_uid(uid, "encode_input.png")):
         return templates.TemplateResponse("encode.html", {"request": request,
-                "error": templates.get_template("error.html").render({"error":
-                    ("The image file cannot be found, you might just exit and got auto cleanup! "
-                    "Otherwise, there might be an attack/error, please issue me if that's so.")})})
+            "error": ("The image file cannot be found, you might just exit and got auto cleanup! "
+                     "Otherwise, there might be an attack/error, please issue me if that's so.")})
     return templates.TemplateResponse("encode.html", {"request": request, "uid": uid})
 
 @app.post("/encode/{uid}")
@@ -144,16 +144,16 @@ async def end_encode(request: Request, uid: str):
                     future_item = asyncio.get_event_loop().run_in_executor(executor,
                                     Generator(input_path).preview, float(form.get("intensity")),
                                     _join_uid(uid, "encode_preview.png"))
-                    future_item.add_done_callback(lambda _: _remove_from_list(uid))
                     background_task.update({uid: future_item})
 
                     async def _progress_streaming():
                         for progress in await future_item:
                             yield f"{progress}\n"
+                        _remove_from_list(uid)
                         yield json.dumps({"result": templates.get_template("result.html")
                                         .render({"path": f"{uid}/encode_preview.png",
                                                  "download": "encode-preview",})})
-                        return StreamingResponse(_progress_streaming(), media_type="text/plain")
+                    return StreamingResponse(_progress_streaming(), media_type="text/plain")
 
                 except asyncio.exceptions.CancelledError:
                     return_error = "User exited."
@@ -162,14 +162,21 @@ async def end_encode(request: Request, uid: str):
 
         case "panel_resize":
             if input_path and form.get("image") and form.get("width") and form.get("length"):
-                Image.open(form.get("image").file)\
+                def _resize() -> None:
+                    Image.open(form.get("image").file)\
                             .resize((int(form.get("width")), int(form.get("length"))))\
                             .save(_join_uid(uid, "encode_resize.png"), optimize=True)
-                return JSONResponse(content={
-                        "result": templates.get_template("result.html").render({
-                            "path": f"{uid}/encode_resize.png",
-                            "download": "encode-resize",
-                        })})
+                try:
+                    future_item=asyncio.get_event_loop().run_in_executor(executor, _resize)
+                    background_task.update({uid: future_item})
+                    await future_item
+                    return JSONResponse(content={
+                            "result": templates.get_template("result.html").render({
+                                "path": f"{uid}/encode_resize.png",
+                                "download": "encode-resize",
+                            })})
+                except asyncio.exceptions.CancelledError:
+                    return_error = "User exited."
 
         case "panel_steganography":
             if input_path and form.get("disguise"):
@@ -206,9 +213,8 @@ async def start_decode(request: Request, uid: str):
     """Start the decode html."""
     if not os.path.exists(_join_uid(uid, "decode_input.png")):
         return templates.TemplateResponse("decode.html", {"request": request,
-                "error": templates.get_template("error.html").render({"error":
-                    ("The image file cannot be found, you might just exit and got auto cleanup! "
-                    "Otherwise, there might be an attack/error, please issue me if that's so.")})})
+            "error": ("The image file cannot be found, you might just exit and got auto cleanup! "
+                    "Otherwise, there might be an attack/error, please issue me if that's so.")})
     return templates.TemplateResponse("decode.html", {"request": request})
 
 @app.post("/decode/{uid}", response_class=JSONResponse)
@@ -262,6 +268,14 @@ async def end_decode(request: Request, uid: str):
 async def remove(uid: str):
     """Remove file when user exit."""    
     shutil.rmtree(os.path.join(IMAGE_DIR, uid), ignore_errors=True)
-    task = background_task.get(uid)
+    task: asyncio.Future = background_task.get(uid)
     if task:
-        task.cancel()
+        try:
+            result = task.result()
+            if isinstance(result, types.GeneratorType):
+                result.throw(asyncio.exceptions.CancelledError("Task cancelled."))
+                _remove_from_list(uid)
+            else:
+                task.cancel()
+        except (asyncio.exceptions.InvalidStateError, asyncio.exceptions.CancelledError):
+            task.cancel()

{quantum_pixel-2.0.2 → quantum_pixel-2.0.4}/uv.lock

@@ -125,7 +125,7 @@ wheels = [
 
 [[package]]
 name = "fastapi"
-version = "0.129.0"
+version = "0.131.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "annotated-doc" },
@@ -134,9 +134,9 @@ dependencies = [
     { name = "typing-extensions" },
     { name = "typing-inspection" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/48/47/75f6bea02e797abff1bca968d5997793898032d9923c1935ae2efdece642/fastapi-0.129.0.tar.gz", hash = "sha256:61315cebd2e65df5f97ec298c888f9de30430dd0612d59d6480beafbc10655af", size = 375450, upload-time = "2026-02-12T13:54:52.541Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/91/32/158cbf685b7d5a26f87131069da286bf10fc9fbf7fc968d169d48a45d689/fastapi-0.131.0.tar.gz", hash = "sha256:6531155e52bee2899a932c746c9a8250f210e3c3303a5f7b9f8a808bfe0548ff", size = 369612, upload-time = "2026-02-22T16:38:11.252Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/9e/dd/d0ee25348ac58245ee9f90b6f3cbb666bf01f69be7e0911f9851bddbda16/fastapi-0.129.0-py3-none-any.whl", hash = "sha256:b4946880e48f462692b31c083be0432275cbfb6e2274566b1be91479cc1a84ec", size = 102950, upload-time = "2026-02-12T13:54:54.528Z" },
+    { url = "https://files.pythonhosted.org/packages/ff/94/b58ec24c321acc2ad1327f69b033cadc005e0f26df9a73828c9e9c7db7ce/fastapi-0.131.0-py3-none-any.whl", hash = "sha256:ed0e53decccf4459de78837ce1b867cd04fa9ce4579497b842579755d20b405a", size = 103854, upload-time = "2026-02-22T16:38:09.814Z" },
 ]
 
 [package.optional-dependencies]
@@ -739,7 +739,7 @@ wheels = [
 
 [[package]]
 name = "typer"
-version = "0.24.0"
+version = "0.24.1"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "annotated-doc" },
@@ -747,9 +747,9 @@ dependencies = [
     { name = "rich" },
     { name = "shellingham" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/5a/b6/3e681d3b6bb22647509bdbfdd18055d5adc0dce5c5585359fa46ff805fdc/typer-0.24.0.tar.gz", hash = "sha256:f9373dc4eff901350694f519f783c29b6d7a110fc0dcc11b1d7e353b85ca6504", size = 118380, upload-time = "2026-02-16T22:08:48.496Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/f5/24/cb09efec5cc954f7f9b930bf8279447d24618bb6758d4f6adf2574c41780/typer-0.24.1.tar.gz", hash = "sha256:e39b4732d65fbdcde189ae76cf7cd48aeae72919dea1fdfc16593be016256b45", size = 118613, upload-time = "2026-02-21T16:54:40.609Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/85/d0/4da85c2a45054bb661993c93524138ace4956cb075a7ae0c9d1deadc331b/typer-0.24.0-py3-none-any.whl", hash = "sha256:5fc435a9c8356f6160ed6e85a6301fdd6e3d8b2851da502050d1f92c5e9eddc8", size = 56441, upload-time = "2026-02-16T22:08:47.535Z" },
+    { url = "https://files.pythonhosted.org/packages/4a/91/48db081e7a63bb37284f9fbcefda7c44c277b18b0e13fbc36ea2335b71e6/typer-0.24.1-py3-none-any.whl", hash = "sha256:112c1f0ce578bfb4cab9ffdabc68f031416ebcc216536611ba21f04e9aa84c9e", size = 56085, upload-time = "2026-02-21T16:54:41.616Z" },
 ]
 
 [[package]]