| --- |
| title: PsyPredict [Backend] |
| emoji: 🧠 |
| colorFrom: indigo |
| colorTo: purple |
| sdk: docker |
| pinned: false |
| --- |
| |
| # PsyPredict - Backend |
|
|
| **FastAPI** backend for PsyPredict — production-grade multimodal clinical AI system. |
|
|
| ## What Runs Here |
|
|
| | Service | Technology | |
| |---------|-----------| |
| | API Framework | FastAPI + Uvicorn | |
| | LLM Inference | Ollama / Llama3 (local) | |
| | Text Emotion | DistilBERT (`bhadresh-savani/distilbert-base-uncased-emotion`) | |
| | Crisis Detection | Zero-shot NLI (MiniLM) | |
| | Face Emotion | Keras CNN (custom trained, `emotion_model_trained.h5`) | |
| | Remedies | CSV lookup (`MEDICATION.csv`) | |
|
|
| ## Endpoints |
|
|
| | Method | Path | Description | |
| |--------|------|-------------| |
| | `POST` | `/api/chat` | Main therapist — returns `PsychReport` | |
| | `POST` | `/api/predict/emotion` | Facial emotion detection | |
| | `GET` | `/api/get_advice` | Remedy/condition lookup | |
| | `POST` | `/api/analyze/text` | Text emotion + crisis score | |
| | `GET` | `/api/health` | System health check | |
|
|
| ## Running Locally |
|
|
| ```bash |
| # 1. Install Ollama + LLaMA 3 (one-time) |
| winget install Ollama.Ollama |
| ollama pull llama3 |
| |
| # 2. Install dependencies |
| pip install -r requirements.txt |
| |
| # 3. Start server |
| uvicorn app.main:app --host 0.0.0.0 --port 7860 --reload |
| ``` |
|
|
| Swagger docs: http://localhost:7860/docs |
|
|
| ## Key Files |
|
|
| ``` |
| app/ |
| ├── main.py # FastAPI app factory |
| ├── config.py # Pydantic Settings |
| ├── schemas.py # All request/response models (PsychReport etc.) |
| ├── services/ |
| │ ├── ollama_engine.py # LLaMA 3 async client |
| │ ├── text_emotion_engine.py# DistilBERT classifier |
| │ ├── crisis_engine.py # Zero-shot NLI crisis detection |
| │ ├── fusion_engine.py # Multimodal weighted fusion |
| │ ├── emotion_engine.py # Keras CNN face emotion (preserved) |
| │ └── remedy_engine.py # CSV remedy lookup (preserved) |
| └── api/endpoints/ |
| ├── therapist.py # POST /api/chat |
| ├── facial.py # POST /api/predict/emotion |
| ├── remedies.py # GET /api/get_advice |
| └── analysis.py # POST /api/analyze/text + GET /api/health |
| ``` |
|
|
