FerrellSyntheticIntelligence's picture
Upload folder using huggingface_hub
d2a5f5a verified
Raw
History Blame Contribute Delete
799 Bytes
from flask import Flask, request, jsonify
import json
import os
from pydantic import BaseModel, ValidationError
from typing import Literal, Optional
app = Flask(__name__)
WORKSPACE = os.getcwd()
TASK_FILE = os.path.join(WORKSPACE, "workspace_tasks.json")
class TaskPayload(BaseModel):
intent: Literal["scaffold", "write"]
module_name: Optional[str] = None
file: Optional[str] = None
code: Optional[str] = None
@app.route('/execute', methods=['POST'])
def execute_task():
try:
data = TaskPayload.model_validate(request.json or {})
except ValidationError as exc:
return jsonify({"error": exc.errors()}), 400
with open(TASK_FILE, "w") as f:
json.dump(data.model_dump(), f)
return jsonify({"status": "Task Queued", "intent": data.intent}), 202