main.py

import os
import subprocess
import json
from fastapi import FastAPI, UploadFile, File, HTTPException, BackgroundTasks
from fastapi.responses import FileResponse, HTMLResponse
from pydantic import BaseModel

app = FastAPI()
WORKSPACE_DIR = os.getenv("WORKSPACE_DIR", "/workspace")
HISTORY_FILE = os.path.join(WORKSPACE_DIR, "chat_history.json")
os.makedirs(WORKSPACE_DIR, exist_ok=True)

class ChatRequest(BaseModel):
    prompt: str
    model: str

class EditRequest(BaseModel):
    filename: str
    content: str

def load_history():
    if os.path.exists(HISTORY_FILE):
        try:
            with open(HISTORY_FILE, "r") as f:
                return json.load(f)
        except:
            pass
    return [{"role": "opencode", "text": "How can I help you code today?"}]

def save_history(history):
    with open(HISTORY_FILE, "w") as f:
        json.dump(history, f)

def run_opencode_bg(prompt: str, model: str):
    env = os.environ.copy()
    if "deepseek" in model.lower():
        env["OPENAI_API_BASE"] = "https://openrouter.ai/api/v1"
        env["OPENAI_API_KEY"] = env.get("OPENROUTER_API_KEY", "")
    
    # We rely on the full LLM dependency to process the instructions and generate files
    cmd = ["opencode", "--prompt", prompt, "--cwd", WORKSPACE_DIR]
    try:
        result = subprocess.run(cmd, env=env, capture_output=True, text=True, timeout=300)
        output = result.stdout if result.returncode == 0 else result.stderr
    except Exception as e:
        output = f"Error running OpenCode: {str(e)}"
        
    history = load_history()
    history.append({"role": "opencode", "text": output})
    save_history(history)

@app.get("/api/chat/history")
def get_history():
    return {"history": load_history()}

@app.post("/api/chat")
def chat(req: ChatRequest, background_tasks: BackgroundTasks):
    history = load_history()
    history.append({"role": "user", "text": req.prompt})
    save_history(history)
    
    # Background task allows app closure without interrupting the AI
    background_tasks.add_task(run_opencode_bg, req.prompt, req.model)
    return {"status": "started"}

@app.get("/api/files")
def list_files():
    files = [f for f in os.listdir(WORKSPACE_DIR) if f != "chat_history.json"]
    return {"files": files}

@app.get("/api/files/download/{filename}")
def download_file(filename: str):
    path = os.path.join(WORKSPACE_DIR, filename)
    if os.path.exists(path):
        return FileResponse(path, filename=filename)
    raise HTTPException(status_code=404)

@app.delete("/api/files/{filename}")
def delete_file(filename: str):
    path = os.path.join(WORKSPACE_DIR, filename)
    if os.path.exists(path):
        os.remove(path)
        return {"status": "deleted"}
    raise HTTPException(status_code=404)

@app.post("/api/files/upload")
async def upload_file(file: UploadFile = File(...)):
    path = os.path.join(WORKSPACE_DIR, file.filename)
    with open(path, "wb") as f:
        f.write(await file.read())
    return {"status": "uploaded"}

@app.post("/api/files/edit")
def edit_file(req: EditRequest):
    path = os.path.join(WORKSPACE_DIR, req.filename)
    with open(path, "w") as f:
        f.write(req.content)
    return {"status": "saved"}

@app.get("/api/files/read/{filename}")
def read_file(filename: str):
    path = os.path.join(WORKSPACE_DIR, filename)
    if os.path.exists(path):
        with open(path, "r") as f:
            return {"content": f.read()}
    raise HTTPException(status_code=404)

@app.get("/", response_class=HTMLResponse)
def root():
    with open("index.html", "r") as f:
        return f.read()