Create app.py

app.py

@@ -0,0 +1,68 @@
+from fastapi import FastAPI, HTTPException, Response
+from pydantic import BaseModel
+import io
+import base64
+import blosc
+import numpy as np
+import matplotlib.pyplot as plt
+
+# --- App and Pydantic Model Setup ---
+
+app = FastAPI()
+
+class BinaryPayload(BaseModel):
+    file_data: str
+    dtype: str = 'float32'
+    shape: tuple[int, int] = (720, 1440)
+    cmap: str = 'plasma'
+
+# --- Helper Function ---
+
+def normalize(arr: np.ndarray, lo: float, hi: float) -> np.ndarray:
+    """Normalizes a NumPy array to the 0-1 range for visualization."""
+    return np.clip((arr - lo) / (hi - lo), 0, 1)
+
+# --- API Endpoints ---
+
+@app.get("/")
+def home():
+    return {"message": "Binary to PNG Conversion API is live."}
+
+@app.post("/render-from-binary")
+async def render_from_binary(payload: BinaryPayload):
+    """
+    Accepts a base64-encoded, Blosc-compressed binary chunk
+    and returns it as a rendered PNG image.
+    """
+    try:
+        # Decode the base64 string
+        compressed_data = base64.b64decode(payload.file_data)
+        
+        # Decompress the data using Blosc
+        decompressed_data = blosc.decompress(compressed_data)
+        
+        # Interpret bytes as a NumPy array and reshape it
+        image_array = np.frombuffer(decompressed_data, dtype=payload.dtype)
+        image_array = image_array.reshape(payload.shape)
+        
+        # Normalize the array for proper color mapping
+        vmin = float(np.nanmin(image_array))
+        vmax = float(np.nanmax(image_array))
+        normalized_frame = normalize(image_array, vmin, vmax)
+        
+        # Generate the PNG image using Matplotlib
+        img_buf = io.BytesIO()
+        fig, ax = plt.subplots(figsize=(6, 3), dpi=240)
+        ax.imshow(normalized_frame, cmap=payload.cmap, origin="upper")
+        ax.axis("off")
+        plt.savefig(img_buf, format="png", bbox_inches="tight", pad_inches=0)
+        plt.close(fig)
+        img_buf.seek(0)
+        
+        # Return the image as a response
+        return Response(content=img_buf.read(), media_type="image/png")
+        
+    except base64.binascii.Error as e:
+        raise HTTPException(status_code=400, detail=f"Invalid base64 string: {e}")
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f"Failed to process binary data: {e}")