from flask import Flask, send_from_directory
import os

app = Flask(__name__)
REPO_DIR = "/workspace/apt-repo"

@app.route('/<path:path>')
def serve_file(path):
    return send_from_directory(REPO_DIR, path)

@app.route('/')
def index():
    return send_from_directory(REPO_DIR, 'index.html') if os.path.exists(os.path.join(REPO_DIR, 'index.html')) else 'APT Repo server is running!'

@app.route('/pgp-key.public')
def serve_public_key():
    return send_from_directory('/workspace/depot', 'pgp-key.public')