import gradio
import os

# basic storage using /dat
def store_data(data):
    if not data:
        return "no request", 400
    if not data["filename"]:
        return "no filename", 400
    if not data["content"]:
        # delete the file
        os.remove(f"/dat/{data['filename']}.txt")
        return "file deleted", 200
    with open(f"/dat/{data['filename']}.txt", "w") as f:
        f.write(data["content"])
        return "file saved", 200
    return "error", 500 # should not happen

def load_data(data):
    if not data:
        return "no request", 400
    if not data["filename"]:
        return "no filename", 400
    try:
        with open(f"/dat/{data['filename']}.txt", "r") as f:
            return f.read(), 200
    except FileNotFoundError:
        return "", 404 # file not found, it was last set to empty string
    return "error", 500 # should not happen

# that's it, now we host the API
iface = gradio.Interface(
    fn=store_data,
    inputs=[
        gradio.inputs.Textbox(label="filename"),
        gradio.inputs.Textbox(label="content")
    ],
    outputs=gradio.outputs.Textbox(label="status")
)
iface.launch()

iface = gradio.Interface(
    fn=load_data,
    inputs=[
        gradio.inputs.Textbox(label="filename")
    ],
    outputs=gradio.outputs.Textbox(label="content")
)
iface.launch()

# now you can use the API to store and load data
# POST /store_data {"filename": "test", "content": "hello world"}
# GET /load_data?filename=test