From 59cec853ec0d82f5f0766d170bbf3046839ed94f Mon Sep 17 00:00:00 2001
From: BIRK Renaud <renaud.birk@imt-atlantique.net>
Date: Tue, 25 Feb 2025 15:32:07 +0100
Subject: [PATCH] fix: CORS header

---
 requirements.txt     | 1 +
 src/main.py          | 5 +++++
 src/routes/router.py | 2 +-
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 820099c..10a5b34 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 Flask>=2.0.3
+flask-cors>=5.0.1
 dotenv>=0.9.9
 gradio_client>=1.7.0
 pydantic>=1.8.2
diff --git a/src/main.py b/src/main.py
index a63b876..9eae820 100644
--- a/src/main.py
+++ b/src/main.py
@@ -1,11 +1,16 @@
 from dotenv import load_dotenv
 from flask import Flask
+from flask_cors import CORS
 from routes.router import setup_routes
 
 load_dotenv()
 
 app = Flask(__name__)
 
+CORS(app)
+
+app.config['CORS_HEADERS'] = 'Content-Type'
+
 setup_routes(app)
 
 if __name__ == '__main__':
diff --git a/src/routes/router.py b/src/routes/router.py
index 128917a..f5ece6b 100644
--- a/src/routes/router.py
+++ b/src/routes/router.py
@@ -11,7 +11,7 @@ glb_controller = GlbController()
 render_controller = RenderController()
 suggest_controller = SuggestController()
 
-def setup_routes(app):
+def setup_routes(app):    
     @app.route('/glb', methods=['GET'])
     # Image to model
     def get_glb():
-- 
GitLab