README.md

metadata

title: KIA Command Center
emoji: 🛡️
colorFrom: red
colorTo: blue
sdk: docker
pinned: false
app_port: 7860
short_description: Albanian General Staff Military AI Assistant

🛡️ KIA Command Center — v3.0

Komanda e Inteligjencës Artificiale i Shtabit të Përgjithshëm Forcat e Armatosura të Republikës së Shqipërisë

Asistenti AI i parë institucional i specializuar për komandën ushtarake shqiptare — me 4 nivele aksesi, RAG hibrid, OCR, STT/TTS, dhe ndërfaqe profesionale C4ISR.

---

⚡ Kapacitetet Kryesore

🔐 Sistemi Multi-Rol (4 Nivele Aksesi)

Rol	Klasifikimi	Aksesi
Vizitor	🟢 I PAKLASIFIKUAR	Informacion publik
Oficer	🟡 I KUFIZUAR	Operacional i përgjithshëm
Komandant	🟠 KONFIDENCIAL	Plane strategjike
Gjeneral	🔴 SEKRET	Akses i plotë strategjik

• Çdo rol ka sistem prompt unik, suggestions specifike, dhe UI klasifikimi
• Classification badge në çdo mesazh
• Greeting personal sipas rangut

🧠 RAG Engine Hibrid

• ChromaDB — Kërkim semantik vektorial me all-MiniLM-L6-v2
• BM25 — Lexical ranking si complement/fallback
• 14,000+ dokumente nga burime zyrtare (MOD, AAF, NATO, Wikipedia)
• 41 Gold Facts — Fakte të verifikuara me prioritet absolut
• Source Priority: Gold > MOD/AAF (0.95) > NATO (0.90) > Wiki (0.70)
• Recency Boost: Informacione 2025-2026 marrin 30% bonus

🎤 Multimodal (Zë + Dokument)

• STT: Whisper — transkriptim i urdhrave në shqip
• TTS: Edge-TTS — përgjigje zanore automatike
• OCR: Tesseract + pdfplumber — skanim PDF, DOCX, XLSX, imazhe
• Auto-suggest: Pas OCR, ofron pyetje kontekstuale mbi dokumentin

🗺️ Ndërfaqe C4ISR Profesionale

• Dashboard me metrika të animuara (count-up) dhe sparkline charts
• Hartë Operative Leaflet me baza ushtarake dhe misione NATO
• Organigram interaktiv — klikoni për pyetje mbi çdo departament
• PDF Export i formatuar si raport klasifikuar i inteligjencës
• Keyboard Shortcuts (F1) — navigim profesional

🔒 Siguria (OPSEC)

• Prompt injection protection (7+ patterna)
• Refuzim i pyetjeve të ndjeshme (armë, kodet, ORBAT)
• Rate limiting (10 req/min per IP)
• Session memory me TTL 30 min

---

🏗️ Arkitektura

┌─────────────────────────────────────────────────────┐
│                    Frontend (Vite)                    │
│  ┌──────────┐  ┌──────┐  ┌──────┐  ┌────────────┐  │
│  │Role Login│  │ Chat │  │ Map  │  │ Dashboard  │  │
│  └──────────┘  └──────┘  └──────┘  └────────────┘  │
│  ┌──────────┐  ┌──────┐  ┌──────────────────────┐  │
│  │Org Chart │  │ Docs │  │ Intel Panel (Right)  │  │
│  └──────────┘  └──────┘  └──────────────────────┘  │
└──────────────────────┬──────────────────────────────┘
                       │ API
┌──────────────────────┴──────────────────────────────┐
│                Backend (FastAPI)                      │
│  ┌──────┐  ┌──────┐  ┌──────┐  ┌────────────────┐  │
│  │ Chat │  │ STT  │  │ TTS  │  │   OCR Scanner  │  │
│  └──┬───┘  └──────┘  └──────┘  └────────────────┘  │
│     │                                                │
│  ┌──┴───────────────────────────────────────────┐   │
│  │          Hybrid RAG Engine                    │   │
│  │  ┌────────┐  ┌────────┐  ┌──────────────┐   │   │
│  │  │ChromaDB│  │  BM25  │  │  Gold Facts  │   │   │
│  │  └────────┘  └────────┘  └──────────────┘   │   │
│  └──────────────────────────────────────────────┘   │
│     │                                                │
│  ┌──┴──────────┐                                    │
│  │ Qwen-72B    │  (Hugging Face Inference API)      │
│  └─────────────┘                                    │
└─────────────────────────────────────────────────────┘

---

🚀 Setup & Deployment

Lokal

# Backend
pip install -r requirements-space.txt
uvicorn app.api:app --port 8001

# Frontend
cd frontend && npm install && npm run dev

Docker (Hugging Face Spaces)

docker build -t kia .
docker run -e HF_TOKEN=your_token -p 7860:7860 kia

Data Pipeline

python main.py all  # Scrape → Clean → Build Dataset → Upload to HF

---

📁 Struktura e Projektit

KIA/
├── app/
│   ├── api.py           # FastAPI endpoints + multi-role system
│   ├── rag.py           # Hybrid RAG engine (ChromaDB + BM25)
│   ├── ocr.py           # Document scanner (PDF, DOCX, images)
│   ├── stt.py           # Speech-to-text (Whisper)
│   └── tts.py           # Text-to-speech (Edge-TTS)
├── frontend/
│   ├── index.html       # SPA with role selection + tactical HUD
│   └── src/
│       ├── main.js      # App logic, multi-role, streaming, animations
│       └── style.css    # NATO C4ISR design system
├── scraper/             # Web scraping pipeline
├── processing/          # Data cleaning + dataset builder
├── finetune/            # QLoRA fine-tuning scripts
├── data/
│   ├── gold/            # Verified critical facts (41 entries)
│   ├── dataset/         # Train/validation JSONL (~28MB)
│   └── chroma_db/       # Vector store
├── Dockerfile           # Multi-stage build
└── main.py              # Pipeline orchestrator

---

🎖️ Teknologjitë

Komponent	Teknologji
Model AI	Qwen2.5-72B-Instruct (via HF inference)
Embeddings	all-MiniLM-L6-v2 (Sentence Transformers)
Vector DB	ChromaDB
Keyword Search	BM25 (rank_bm25)
Backend	FastAPI + Uvicorn
Frontend	Vite + Vanilla JS
STT	OpenAI Whisper
TTS	Edge-TTS (Albanian)
OCR	Tesseract + pdfplumber
Maps	Leaflet.js
Container	Docker (multi-stage)

---

📄 Licenca

Zhvilluar për përdorim institucional — Shtabi i Përgjithshëm i FA-së.

---

Zhvilluar me krenarinë shqiptare 🇦🇱 — KIA v3.0 Multi-Role Edition