app.py

from __future__ import annotations

from flask import Flask, jsonify, render_template, request
from dotenv import load_dotenv

load_dotenv()

import traceback
import config
from agent.orchestrator import PiForgeOrchestrator
from agent.tools import safe_read_file, safe_write_file, list_project_tree

app = Flask(__name__)
@app.errorhandler(Exception)
def handle_exception(e):
    return jsonify({
        "ok": False,
        "error": str(e),
        "trace": traceback.format_exc()
    }), 500
orchestrator = PiForgeOrchestrator(config.PROJECTS_DIR)


@app.get("/")
def index():
    return render_template("index.html")


@app.get("/api/health")
def health():
    return jsonify({
        "ok": True,
        "backend": config.MODEL_BACKEND,
        "projects_dir": str(config.PROJECTS_DIR),
    })


@app.get("/api/projects")
def projects():
    items = sorted([p.name for p in config.PROJECTS_DIR.iterdir() if p.is_dir()])
    return jsonify({"projects": items})


@app.post("/api/create_project")
def create_project():
    payload = request.get_json(force=True)
    result = orchestrator.create_project(
        name=payload.get("name", "").strip(),
        template_name=payload.get("template", "python_cli").strip(),
    )
    return jsonify(result)


@app.post("/api/generate")
def generate():
    payload = request.get_json(force=True)
    result = orchestrator.generate_from_prompt(
        project_name=payload.get("project_name", "").strip(),
        prompt=payload.get("prompt", "").strip(),
    )
    return jsonify(result)


@app.post("/api/run")
def run_command():
    payload = request.get_json(force=True)
    result = orchestrator.run_command(
        project_name=payload.get("project_name", "").strip(),
        command=payload.get("command", "").strip(),
    )
    return jsonify(result)


@app.get("/api/project_tree/<project_name>")
def project_tree(project_name: str):
    return jsonify({"tree": list_project_tree(config.PROJECTS_DIR / project_name)})


@app.get("/api/file")
def read_file():
    project = request.args.get("project", "").strip()
    relpath = request.args.get("path", "").strip()
    content = safe_read_file(config.PROJECTS_DIR / project, relpath)
    return jsonify({"content": content})


@app.post("/api/file")
def write_file():
    payload = request.get_json(force=True)
    result = safe_write_file(
        config.PROJECTS_DIR / payload.get("project", "").strip(),
        payload.get("path", "").strip(),
        payload.get("content", ""),
    )
    return jsonify(result)


if __name__ == "__main__":
    app.run(host=config.HOST, port=config.PORT, debug=config.DEBUG)