Update README.md - add FastAPI REST endpoints + Gemini AI

README.md

@@ -7,32 +7,39 @@ sdk: docker
 app_file: app.py
 pinned: false
 license: mit
+short_description: CyberForge AI - ML Training & Inference Platform
 ---
 
-# CyberForge AI - ML Training Platform
+# 🔐 CyberForge AI - ML Training & Inference Platform
 
-Train cybersecurity ML models and run Jupyter notebooks on Hugging Face.
+Gradio UI for notebook execution, model training, and inference + FastAPI REST API for the Heroku backend.
 
-## Features
+## REST API Endpoints (for backend mlService.js)
 
-- **Notebooks Tab**: Run ML pipeline notebooks directly in the cloud
-- **Train Model**: Upload CSV data and train security models
-- **Inference**: Load trained models and make predictions
-- **API**: REST API for integration
+| Endpoint | Method | Description |
+|----------|--------|-------------|
+| `/health` | GET | Health check |
+| `/analyze` | POST | AI chat (Gemini) |
+| `/analyze-url` | POST | URL threat analysis |
+| `/scan-threats` | POST | Threat scanning |
+| `/api/insights/generate` | POST | AI insights |
+| `/api/models/predict` | POST | ML model prediction |
+| `/models` | GET | List available models |
+| `/api/analysis/network` | POST | Network traffic analysis |
+| `/api/ai/execute-task` | POST | AI task execution |
 
-## Notebooks
+## Environment Secrets
 
-| # | Notebook | Purpose |
-|---|----------|---------|
-| 00 | environment_setup | System validation |
-| 01 | data_acquisition | Data collection |
-| 02 | feature_engineering | Feature extraction |
-| 03 | model_training | Train models |
-| 04 | agent_intelligence | AI reasoning |
-| 05 | model_validation | Testing |
-| 06 | backend_integration | API packaging |
-| 07 | deployment_artifacts | Deployment |
+Set these in your Space settings → Secrets:
 
-## Links
+- `GEMINI_API_KEY` - Google Gemini API key
+- `HF_TOKEN` - HuggingFace token (for model downloads)
+- `HF_MODEL_REPO` - Model repository (default: `Che237/cyberforge-models`)
 
-- [Datasets](https://huggingface.co/datasets/Che237/cyberforge-datasets)
+## Backend Integration
+
+Your Heroku backend connects to this Space:
+
+```bash
+heroku config:set AI_SERVICE_URL=https://che237-cyberforge.hf.space -a cyberforge
+```