from flask import Flask, render_template, request, jsonify
import os

app = Flask(__name__)

mock_docs = [
    {'id': 1, 'filename': 'Veiligheid op de werkvloer (mock).pdf', 'category': 'Algemeen', 'pages': 10},
    {'id': 2, 'filename': 'Risicoanalyse-methodiek.pdf', 'category': 'Risicobeheersing', 'pages': 25},
    {'id': 3, 'filename': 'Hitte & Warmte Arbeidsomstandigheden.pdf', 'category': 'Hitte', 'pages': 18}
]

@app.route('/')
def index():
    categories = {}
    for doc in mock_docs:
        cat = doc['category']
        categories[cat] = categories.get(cat, 0) + 1
    
    return render_template('index.html', 
                          total_docs=len(mock_docs),
                          categories=categories.items(),
                          recent_docs=mock_docs)

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5052, debug=False)