🧠 EdgeBrain — AI-Powered Edge Intelligence Platform

──────────────────────────────────────────────────────────────

Autonomous real-world decision systems. Runs locally. No paid APIs required.

https://github.com/rudra496/EdgeBrain

──────────────────────────────────────────────────────────────

---

🎬 Demo

# Clone & run in under 2 minutes
git clone https://github.com/rudra496/EdgeBrain.git && cd EdgeBrain
docker compose up --build -d
# Open http://localhost:3000 — done.

What you'll see:

• 🌡️ Live temperature, humidity, energy, light charts
• 🤖 AI agents making decisions in real-time
• 🚨 Alerts firing when anomalies are detected
• 💡 Lights auto-on with motion, auto-off after 5 min
• 🌀 Fan activation when temperature exceeds 30°C
• 📊 Full agent message pipeline visualization

---

✨ Features

🏠 Smart Device Simulation (11 devices)

• 3 rooms — Living Room (5 sensors), Bedroom (3 sensors), Server Room (3 sensors)
• 5 sensor types — Temperature, Motion, Energy, Humidity, Light
• Realistic patterns — circadian rhythms, Gaussian noise, mean reversion
• Rare event simulation — temperature spikes (1%), energy surges (0.5%)
• Configurable — base values, noise scale, spike probability per device
• Time-aware — energy peaks during work hours, light follows sunrise/sunset

🤖 Multi-Agent AI System

• Data Agent — validates ranges, stores readings, routes to decision pipeline
• Decision Agent — runs all strategies, deduplicates, ranks by confidence
• Action Agent — creates alerts, sends MQTT commands, updates actuator state
• Message Bus — full queryable log with agent performance tracking
• Per-agent metrics — average processing time, message count, error rate

🧠 AI Decision Engine (3 strategies)

• Threshold Rules — configurable per device type, with hysteresis to prevent flapping
• Anomaly Detection — 3 statistical methods vote: Z-Score, IQR, Gradient
  • Requires 2+ methods to agree (reduces false positives by ~70%)
• No-Motion Timeout — turns off lights after configurable inactivity period
• Plugin Architecture — add custom strategies via DecisionStrategy interface
  • Just implement evaluate() and name — that's it

📈 Predictive Analytics

• Trend Prediction — linear regression on recent readings to forecast next value
• Moving Averages — SMA and EMA for smoothing noisy sensor data
• Anomaly Score — combined confidence from all detection methods
• Forecasting — 10-step ahead prediction for temperature and energy

📊 Real-Time Dashboard (5 pages)

• Dashboard — stats overview, live charts, event feed, recent alerts
• Devices — all sensor states with room grouping, online/offline status
• Analytics — historical charts, trend predictions, device comparisons
• Agents — pipeline visualization, message log, strategy config, performance
• Settings — system configuration, thresholds, export data, system info

🔌 Communication & Integration

• MQTT 3.1.1 — topic-based architecture with QoS support
• WebSocket — real-time bidirectional dashboard updates
• REST API — 20+ endpoints with auto-generated OpenAPI docs
• Data Export — CSV and JSON export for any device's readings
• Webhook Support — configurable webhook for alert notifications

🏗️ Production Infrastructure

• Docker Compose — 6 services, health checks, restart policies, volumes
• PostgreSQL — composite indexes, JSONB fields, time-series optimized
• Redis — event queuing, pub/sub, alert cache, live telemetry
• GitHub Actions — CI (tests + lint) + Release (Docker + Git tag)
• Makefile — one-command operations: make dev, make test, make release

🔧 Optional Hardware

• ESP32 firmware — Arduino C++ with WiFi + MQTT
• DHT11 — temperature sensor (code simulated by default, real sensor supported)
• PIR — passive infrared motion sensor
• LED — actuator indicator
• Configurable — just edit WiFi + MQTT credentials

---

⚡ Quick Start

Prerequisites

• Docker v20+ & Docker Compose v2+
• 4GB RAM (8GB recommended)
• ~2GB disk space

Install & Run

git clone https://github.com/rudra496/EdgeBrain.git
cd EdgeBrain
docker compose up --build -d

Or use Make

make setup        # Clone (if not already), build, and start
make dev          # Start in development mode
make test         # Run test suite
make logs         # Tail all service logs
make stop         # Stop all services
make clean        # Stop + remove volumes

Access Points

Service	URL	Description
🖥️ Dashboard	http://localhost:3000	React real-time dashboard
📡 Swagger API	http://localhost:8000/docs	Interactive API explorer
📖 ReDoc	http://localhost:8000/redoc	Alternative API docs
❤️ Health	http://localhost:8000/api/v1/health	System health check
🐝 MQTT	localhost:1883	Mosquitto broker
📡 MQTT/WS	localhost:9001	MQTT over WebSocket

Verify Everything

# Health check
curl -s http://localhost:8000/api/v1/health | python3 -m json.tool

# System statistics
curl -s http://localhost:8000/api/v1/stats | python3 -m json.tool

# Device list
curl -s http://localhost:8000/api/v1/devices | python3 -m json.tool

# Recent alerts
curl -s http://localhost:8000/api/v1/alerts | python3 -m json.tool

# Agent performance
curl -s http://localhost:8000/api/v1/agents/stats | python3 -m json.tool

---

🏗️ Architecture

graph TB
    subgraph Edge["🔌 Edge Layer"]
        S1["🌡️ Temp — Room 1"]
        S2["🏃 Motion — Room 1"]
        S3["⚡ Energy — Room 1"]
        S4["💧 Humidity — Room 1"]
        S5["☀️ Light — Room 1"]
        S6["🌡️ Temp — Room 2"]
        S7["🏃 Motion — Room 2"]
        S8["⚡ Energy — Room 2"]
        S9["🌡️ Temp — Server Room"]
        S10["💧 Humidity — Server Room"]
        S11["⚡ Energy — Server Room"]
        ESP["📡 ESP32 — Optional"]
    end

    subgraph Comms["📡 Communication Layer"]
        MQTT["Mosquitto MQTT 3.1.1"]
    end

    subgraph BE["🧠 Backend (FastAPI)"]
        API["REST + WebSocket API"]
        PIPE["Data Pipeline"]
        EVT["Event Engine"]
        EXEC["Execution Engine"]
        PRED["Prediction Engine"]
    end

    subgraph AI["🤖 AI / Decision Engine"]
        DA["Data Agent"]
        DEC["Decision Agent"]
        AA["Action Agent"]
        RULE["Rule Engine<br/>+ Hysteresis"]
        ANOM["Anomaly Detector<br/>Z-Score · IQR · Gradient"]
        PREDAI["Predictor<br/>Linear Regression"]
        PLUG["Plugin System"]
    end

    subgraph Data["💾 Data Layer"]
        PG[("PostgreSQL 16")]
        REDIS[("Redis 7")]
    end

    subgraph FE["🖥️ Frontend"]
        DASH["React Dashboard<br/>5 Pages · WebSocket"]
    end

    S1 & S2 & S3 & S4 & S5 & S6 & S7 & S8 & S9 & S10 & S11 & ESP -->|"device/+/data"| MQTT
    MQTT -->|"subscribe"| PIPE
    PIPE -->|"store"| PG
    PIPE -->|"route"| DA
    DA -->|"validate"| DEC
    RULE & ANOM & PREDAI --> DEC
    PLUG --> DEC
    DEC -->|"decisions"| AA
    AA -->|"alerts"| EVT
    AA -->|"commands"| EXEC
    EVT --> REDIS
    REDIS -->|"broadcast"| DASH
    EXEC -->|"device/+/command"| MQTT
    MQTT -->|"actuate"| S1 & S2 & S3
    API <-->|"HTTP/WS"| DASH
    API --> PG
    API --> REDIS

Decision Pipeline

sequenceDiagram
    participant S as Sensor
    participant M as MQTT
    participant DA as Data Agent
    participant DB as PostgreSQL
    participant DEC as Decision Agent
    participant AA as Action Agent
    participant A as Alert/Redis
    participant CMD as MQTT Command
    participant ACT as Actuator

    S->>M: device/room-1/data {temp: 35°C}
    M->>DA: route message
    DA->>DA: validate range
    DA->>DB: store reading
    DA->>DEC: evaluate(device, type, value)
    DEC->>DEC: ThresholdStrategy → fan ON
    DEC->>DEC: AnomalyDetector → normal
    DEC->>DEC: deduplicate + rank
    DEC->>AA: [Decision: activate fan]
    AA->>A: push alert to Redis
    AA->>CMD: device/room-1-actuator-fan/command
    CMD->>ACT: activate fan
    A-->>Dashboard: WebSocket broadcast

---

📁 Project Structure

EdgeBrain/
├── .github/
│   ├── ISSUE_TEMPLATE/
│   │   ├── bug_report.yml          # Structured bug reports
│   │   └── feature_request.yml     # Structured feature requests
│   └── PULL_REQUEST_TEMPLATE.md    # PR checklist
│
├── backend/
│   ├── app/
│   │   ├── api/
│   │   │   ├── routes.py           # 20+ REST endpoints + WebSocket
│   │   │   └── schemas.py          # Pydantic request/response models
│   │   ├── core/
│   │   │   ├── config.py           # Settings with env vars
│   │   │   ├── database.py         # SQLAlchemy engine + sessions
│   │   │   ├── mqtt_client.py      # Thread-safe MQTT with reconnect
│   │   │   └── events.py           # Redis queue + pub/sub
│   │   ├── ai/
│   │   │   ├── rules.py            # Rule engine + hysteresis + plugin
│   │   │   ├── anomaly.py          # Z-Score + IQR + Gradient detection
│   │   │   └── prediction.py       # Linear regression + forecasting
│   │   ├── agents/
│   │   │   └── multi_agent.py      # Data → Decision → Action pipeline
│   │   ├── models/
│   │   │   └── models.py           # 5 SQLAlchemy models + indexes
│   │   ├── services/
│   │   │   ├── ingestion.py        # Data validation + storage
│   │   │   └── execution.py        # Commands + alerts + actuator state
│   │   └── main.py                 # FastAPI app + lifespan
│   ├── tests/
│   │   ├── test_ai.py              # 20+ AI engine tests
│   │   └── test_api.py             # API endpoint tests
│   ├── requirements.txt
│   └── Dockerfile
│
├── frontend/
│   ├── src/
│   │   ├── App.js                  # Main app + 5-page router
│   │   ├── App.css                 # Complete dark theme design system
│   │   └── index.js                # React entry point
│   ├── public/index.html
│   ├── package.json
│   └── Dockerfile
│
├── device-simulator/
│   └── simulator.py                # 11 devices, 3 rooms, realistic data
│
├── simulator/
│   └── Dockerfile                  # Simulator container
│
├── esp32-firmware/
│   ├── main/main.ino               # ESP32 Arduino firmware
│   └── README.md                   # Hardware setup guide
│
├── docker/
│   ├── init.sql                    # Database schema + indexes
│   └── mosquitto.conf              # MQTT broker config
│
├── docs/
│   ├── ARCHITECTURE.md             # Detailed system design
│   ├── SETUP.md                    # Setup with/without Docker
│   └── ROADMAP.md                  # Development roadmap
│
├── Makefile                        # Convenience commands
├── docker-compose.yml              # 6-service orchestration
├── LICENSE                         # MIT
├── CODE_OF_CONDUCT.md              # CoC
├── CONTRIBUTING.md                 # Contribution guide
└── README.md                       # This file

---

🔌 Custom Strategies

from app.ai.rules import DecisionStrategy, Decision

class MyCustomStrategy(DecisionStrategy):
    """Example: Alert when humidity exceeds threshold."""

    @property
    def name(self) -> str:
        return "humidity_alert"

    def evaluate(self, device_id, device_type, value, history):
        if device_type != "humidity":
            return []

        if value > 80:
            return [Decision(
                action="activate",
                device_id=device_id,
                params={"actuator": "alarm"},
                reason=f"High humidity: {value:.1f}%",
                confidence=0.85,
                severity="warning",
                source=self.name,
            )]
        return []

# Register:
from app.agents.multi_agent import agents
agents.engine.add_strategy(MyCustomStrategy())

---

🛠️ Tech Stack

Layer	Technology	Version	Purpose
Backend	Python	3.11+	Runtime
Framework	FastAPI	0.110	Async REST + WebSocket
Database	PostgreSQL	16	Time-series storage
Cache	Redis	7.2	Queue, pub/sub, cache
Messaging	Mosquitto	2.0	MQTT broker
AI/Math	NumPy + SciPy	1.26 / 1.12	Anomaly detection, prediction
Frontend	React	18.3	Dashboard UI
Charts	Recharts	2.12	Real-time visualizations
Icons	Lucide	0.344	Icon library
Infra	Docker Compose	2.x	Local orchestration
Hardware	ESP32	Arduino	Optional real sensors

---

📖 Documentation

Document	Description
Architecture	System design, data flow, plugin system
Setup Guide	Docker + native setup instructions
Roadmap	Future plans and milestones
Contributing	How to contribute code
API Reference	Interactive Swagger (when running)

---

🚀 Release Strategy

Releases are versioned using Semantic Versioning:

• v1.x.x — Current stable. Core features complete.
• v2.x.x — Planned: distributed edge, TimescaleDB, Grafana.
• v3.x.x — Planned: plugin marketplace, multi-tenant.

See Releases for changelog.

---

🤝 Contributing

We ❤️ contributions! See CONTRIBUTING.md.

# Fork, branch, code, PR
git clone https://github.com/rudra496/EdgeBrain.git
cd EdgeBrain
make dev      # Start development environment
make test     # Run tests
# Make changes, then open PR

Contributors

Thanks to all contributors! Your PRs make EdgeBrain better.

---

📄 License

MIT — free for personal and commercial use.

---

⭐ Star History

---

---

More Open Source Projects

Project	Stars	Description
StealthHumanizer		Free AI text humanizer — 13 providers, no login
DevRoadmaps		17 career paths, 1700+ free resources
CodeVista		AI code analysis & security scanner
Nexus Agent		Self-evolving local AI agent framework
MindWell		Free mental health support platform
ScienceLab 3D		40+ virtual STEM experiments
SightlineAI		AI smart glasses for the blind

---

Connect

• 
• 
• 
• 
•