Spaces:
Running
Running
File size: 2,351 Bytes
e21f117 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 | from fastapi import FastAPI, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
import uuid
import time
app = FastAPI(title="NoobAI RAM Storage")
# WAJIB: Buka CORS agar UI HTML kamu bisa nge-fetch API ini
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# ==========================================
# 🧠 IN-MEMORY STORAGE (PURE RAM)
# Struktur: { "image_id": {"base64": "...", "timestamp": 12345} }
# ==========================================
memory_db = {}
# Proteksi RAM: Asumsi 1 gambar Base64 = ~1.5MB.
# 5000 gambar = ~7.5GB RAM. Sangat aman untuk RAM 16/18GB.
MAX_IMAGES = 5000
class ImagePayload(BaseModel):
image_base64: str
@app.get("/")
def read_root():
"""Health check endpoint"""
return {
"status": "Online",
"ram_images_count": len(memory_db),
"max_capacity": MAX_IMAGES
}
@app.post("/save")
def save_image(payload: ImagePayload):
"""Simpan gambar ke RAM"""
# Jika RAM penuh, hapus gambar paling lama (FIFO)
if len(memory_db) >= MAX_IMAGES:
oldest_id = min(memory_db, key=lambda k: memory_db[k]["timestamp"])
del memory_db[oldest_id]
img_id = str(uuid.uuid4())
memory_db[img_id] = {
"base64": payload.image_base64,
"timestamp": time.time()
}
return {"status": "success", "image_id": img_id}
@app.get("/images")
def get_all_images():
"""Ambil daftar ID gambar (hanya ID dan waktu, bukan base64 nya agar enteng)"""
return {
"images": [
{"id": k, "timestamp": v["timestamp"]}
for k, v in sorted(memory_db.items(), key=lambda item: item[1]["timestamp"], reverse=True)
]
}
@app.get("/image/{img_id}")
def get_image(img_id: str):
"""Ambil Base64 dari gambar spesifik berdasarkan ID"""
if img_id not in memory_db:
raise HTTPException(status_code=404, detail="Image not found in RAM or expired")
return {"image_id": img_id, "base64": memory_db[img_id]["base64"]}
@app.delete("/image/{img_id}")
def delete_image(img_id: str):
"""Hapus gambar secara manual dari RAM"""
if img_id in memory_db:
del memory_db[img_id]
return {"status": "deleted"}
return {"status": "not_found"}
|