Proxy Fix and requirements.txt

1 files changed, 4 insertions, 0 deletions

diff --git a/main.py b/main.py
index 7c530fc..05f645d 100755
--- a/main.py
+++ b/main.py
@@ -1,8 +1,12 @@
 from flask import Flask, render_template, send_file
 from markdown import markdown
 from db import DB
+from werkzeug.middleware.proxy_fix import ProxyFix
 
 app = Flask(__name__)
+app.wsgi_app = ProxyFix(
+    app.wsgi_app, x_for=1, x_proto=1, x_host=1, x_prefix=1
+)
 r = DB()
 
 @app.route('/')