From 50221af55a838f08bb6a12b93fee6fa01827247c Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Thu, 5 Jan 2023 17:24:33 -0600
Subject: [PATCH] feat(api): add endpoint to get previous outputs

---
 api/serve.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/api/serve.py b/api/serve.py
index 684c3ff5..4519d22e 100644
--- a/api/serve.py
+++ b/api/serve.py
@@ -8,7 +8,7 @@ from diffusers import (
     LMSDiscreteScheduler,
     PNDMScheduler,
 )
-from flask import Flask, make_response, request, send_file
+from flask import Flask, make_response, request, send_file, send_from_directory
 from stringcase import spinalcase
 from io import BytesIO
 from os import environ, path, makedirs
@@ -135,3 +135,7 @@ def txt2img():
     res = make_response(send_file(img_io, mimetype='image/png'))
     res.headers.add('Access-Control-Allow-Origin', '*')
     return res
+
+@app.route('/output/<path:filename>')
+def output(filename):
+    return send_from_directory(output_path, filename, as_attachment=False)