From 308a3e24e089451c8732617b6ab1b46fe8b4d73f Mon Sep 17 00:00:00 2001 From: BIRK Renaud <renaud.birk@imt-atlantique.net> Date: Tue, 18 Feb 2025 20:13:08 +0100 Subject: [PATCH] feat: API GET /render --- src/controllers/render_controller.py | 6 ++++++ src/routes/router.py | 7 +++++++ 2 files changed, 13 insertions(+) create mode 100644 src/controllers/render_controller.py diff --git a/src/controllers/render_controller.py b/src/controllers/render_controller.py new file mode 100644 index 0000000..d362dd5 --- /dev/null +++ b/src/controllers/render_controller.py @@ -0,0 +1,6 @@ +import os +from gradio_client import Client, handle_file + +class RenderController: + def get_render(self, hf_token='') -> str: + return "Hello, World!" diff --git a/src/routes/router.py b/src/routes/router.py index ea9e058..00fdca4 100644 --- a/src/routes/router.py +++ b/src/routes/router.py @@ -1,11 +1,14 @@ from flask import Blueprint, request from controllers.glb_controller import GlbController +from controllers.render_controller import RenderController from controllers.suggest_controller import SuggestController glb_bp = Blueprint('glb', __name__) +render_bp = Blueprint('render', __name__) suggest_bp = Blueprint('suggest', __name__) glb_controller = GlbController() +render_controller = RenderController() suggest_controller = SuggestController() def setup_routes(app): @@ -26,6 +29,10 @@ def setup_routes(app): slat_sampling_steps, multiimage_algo, mesh_simplify, texture_size) + @app.route('/render', methods=['GET']) + def get_render(): + return render_controller.get_render() + @app.route('/suggest', methods=['GET']) def get_suggest(): hf_token = request.args.get('hf_token', '', type=str) -- GitLab