Update app.py

app.py

@@ -38,6 +38,22 @@ import json
 import yaml
 from slugify import slugify
 
+from collections import deque
+
+log_buffer = deque(maxlen=1000)
+
+class InMemoryLogHandler(logging.Handler):
+    def emit(self, record):
+        msg = self.format(record)
+        log_buffer.append(msg)
+
+# Setup logging to buffer
+handler = InMemoryLogHandler()
+handler.setFormatter(logging.Formatter("%(asctime)s | %(levelname)s | %(message)s"))
+logging.getLogger().addHandler(handler)
+
+
+
 sys.path.insert(0, "ai-toolkit")
 from toolkit.job import get_job
 
@@ -77,6 +93,24 @@ app.add_middleware(
 def health_check():
     return {"status": "✅ FastAPI running on Hugging Face Spaces!"}
 
+
+
+
+
+@app.get("/jobs")
+def all_jobs():
+    return list(JOB_QUEUE.values())
+
+
+@app.get("/logs")
+def get_logs(limit: int = 50):
+    return list(log_buffer)[-limit:]
+
+@app.get("/jobs/running")
+def running_jobs():
+    return [job for job in JOB_QUEUE.values() if job.status in ("pending", "running")]
+
+
 REPO_ID = "rahul7star/ohamlab"
 FOLDER = "demo"
 BASE_URL = f"https://huggingface.co/{REPO_ID}/resolve/main/"