app.py

import os
import psycopg2
from flask import Flask, request, jsonify

app = Flask(__name__)

DB_HOST = os.environ.get("DB_HOST")
DB_USER = os.environ.get("DB_USER")
DB_PASS = os.environ.get("DB_PASS")
DB_NAME = os.environ.get("DB_NAME")

def get_connection():
    return psycopg2.connect(
        host=DB_HOST,
        user=DB_USER,
        password=DB_PASS,
        dbname=DB_NAME
    )

@app.route("/")
def home():
    return "Task API Running"

@app.route("/task", methods=["POST"])
def add_task():
    data = request.json
    task = data["task"]

    conn = get_connection()
    cur = conn.cursor()

    cur.execute("INSERT INTO tasks (task) VALUES (%s)", (task,))
    conn.commit()

    cur.close()
    conn.close()

    return "Task added"

@app.route("/tasks")
def get_tasks():

    conn = get_connection()
    cur = conn.cursor()

    cur.execute("SELECT * FROM tasks")
    rows = cur.fetchall()

    cur.close()
    conn.close()

    return jsonify(rows)

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=8080)