AI-Documentation-Agent

DOZAi is a full-stack AI platform that automates technical documentation from code, APIs, and prompts. It runs entirely on an AMD GPU Droplet (AMD Developer Cloud), hosting: Spring Boot backend (REST API, auth, processing), Python Flask app (visualization engine), PostgreSQL (metadata & history), and Ollama with TinyLlama (local LLM inference). An integration layer connects to cloud AI (Gemini API, OpenRouter). How It Works: Input Collection – Users upload files (PDF, images, text). Spring Boot handles uploads and JWT authentication. Document Analysis – Python Flask extracts text/code blocks, identifies APIs/patterns, prepares data for AI. AI-Powered Generation – Dual-path: local (AMD GPU + TinyLlama for fast, offline gen) or cloud (Gemini 3 Pro via OpenRouter for complex docs). Visualization & Output – Python renders architecture diagrams (PlantUML, Mermaid), flow charts, API docs, and PDF reports with embedded visuals. Storage & Retrieval – PostgreSQL stores docs, metadata, and user history. Users retrieve, regenerate, or export to PDF/DOCX/Markdown. The agent can reverse-engineer code into docs, generate API references, create diagrams from prompts, summarize papers, and translate comments. Running all components on a single AMD GPU Droplet reduces costs while leveraging both local and cloud AI. This demonstrates a complete, production-ready AI documentation workflow.