README.md

---
title: '🛡️PhishGuard AI '
emoji: 🧠
colorFrom: blue
colorTo: green
sdk: gradio
sdk_version: 5.49.1
app_file: app_gradio.py
pinned: true
license: mit
---
<p align="center">
  <img src="https://huggingface.co/MujMej/phishguard-ai/resolve/main/banner.png" alt="PhishGuard AI Banner" width="100%">
</p>


# 🛡️ PhishGuard AI 

PhishGuard AI je pametan alat za analizu phishing e-mailova, razvijen u **Pythonu**, koristeći **NLP modele (Transformers)** i **heurističke analize**.  
Prepoznaje phishing, spam i legitimne poruke na osnovu jezičkih i tehničkih indikatora.

---

## ⚙️ Karakteristike

- 🧠 **Hibridni model:** Kombinuje Hugging Face NLP model (`firoj/phishing-email-detection`) sa heuristikom (SPF, DKIM, DMARC, gustoća URL-ova, ključne riječi).  
- 📊 **Nivoi rizika:** Klasifikuje poruke kao **Critical**, **High**, **Medium** ili **Low**.  
- 📬 **Analiza e-maila:** Uploaduj `.eml` fajl direktno kroz web interfejs.  
- 🌈 **XDR-inspirisan dashboard:** Vizuelni prikaz doprinosa AI modela i heuristike.  
- 💾 **Offline/Online rad:** Radi lokalno ili direktno na Hugging Face Space-u.  

---

## 🚀 Kako koristiti

1. Uploaduj `.eml` fajl e-mail poruke  
2. Klikni na **Analyze Email**  
3. Dobit ćeš detaljan izvještaj sa:
   - AI sigurnosnim skorom  
   - Heurističkom ocjenom rizika  
   - Izdvojenim URL adresama  
   - Pie chart vizualizacijom  

---

## 🧩 Korištene tehnologije

| Kategorija | Alat / Biblioteka |
|-------------|-------------------|
| Mašinsko učenje | Hugging Face Transformers |
| Backend | Python 3.10+ |
| Web UI | Gradio |
| Vizualizacija | Plotly |
| Parsiranje HTML-a | BeautifulSoup4 |
| ML Framework | PyTorch |

---

## 📈 Primjer izlaza

Klasifikacija: PHISH
Nivo rizika: Critical
HF Score: 0.91
Heuristički score: 0.72
Pošiljalac: [email protected]

Naslov: "Urgent Account Verification Required"


---

## 👩‍💻 Autor  
**Autor:** Mejra Mujanović  
**Projekat:** Diplomski rad — *Mašinsko učenje u Pythonu (FIT Apeiron, 2025)*  
**Tagovi:** `cybersecurity`, `nlp`, `phishing`, `xdr`, `soc`, `ai`  

---

<p align="center">
  💡 *Detektuj. Klasifikuj. Zaštiti.*
</p>