Function predict_from_file

server/main.py:180–203

Endpoint to make predictions using an uploaded file

(file: UploadFile = File(...), prompt: str = Form(...))

@app.post("/predict_from_file", response_model=MagmaResponse)
async def predict_from_file(file: UploadFile = File(...), prompt: str = Form(...)):
    """Endpoint to make predictions using an uploaded file"""
    if model is None:
        raise HTTPException(status_code=503, detail="Model not loaded yet")

    try:
        # Read the image file
        image_data = await file.read()
        image = Image.open(io.BytesIO(image_data)).convert("RGB")

        # Generate the response
        normalized_actions, delta_values, text_response = generate_response(
            image, prompt
        )

        return {
            "text_response": text_response,
            "normalized_actions": normalized_actions,
            "delta_values": delta_values,
        }
    except Exception as e:
        raise HTTPException(
            status_code=500, detail=f"Error processing request: {str(e)}"
        )


@app.get("/health")