From 3a70053b1796ff1ded431c688f491697b49044df Mon Sep 17 00:00:00 2001
From: BIRK Renaud <renaud.birk@imt-atlantique.net>
Date: Tue, 25 Feb 2025 13:58:24 +0100
Subject: [PATCH] feat: API GET /render
---
src/controllers/render_controller.py | 14 +++++++++++++-
src/routes/router.py | 7 ++++++-
2 files changed, 19 insertions(+), 2 deletions(-)
diff --git a/src/controllers/render_controller.py b/src/controllers/render_controller.py
index d362dd5..92cc236 100644
--- a/src/controllers/render_controller.py
+++ b/src/controllers/render_controller.py
@@ -2,5 +2,17 @@ import os
from gradio_client import Client, handle_file
class RenderController:
- def get_render(self, hf_token='') -> str:
+ def get_render(self, hf_token='', text="Hello!!",
+ selected_model="Model 1 (Turbo Realism)") -> str:
+
+ client = Client("Jonny001/Text-to-Image")
+
+ result = client.predict(
+ text=text,
+ selected_model=selected_model,
+ api_name="/generate_images"
+ )
+
+ print(result)
+
return "Hello, World!"
diff --git a/src/routes/router.py b/src/routes/router.py
index 9293e2d..128917a 100644
--- a/src/routes/router.py
+++ b/src/routes/router.py
@@ -13,6 +13,7 @@ suggest_controller = SuggestController()
def setup_routes(app):
@app.route('/glb', methods=['GET'])
+ # Image to model
def get_glb():
base_image_url = request.args.get('base_image_url', type=str)
hf_token = request.args.get('hf_token', '', type=str)
@@ -29,12 +30,16 @@ def setup_routes(app):
slat_sampling_steps, multiimage_algo,
mesh_simplify, texture_size)
+ # Text to image
@app.route('/render', methods=['GET'])
def get_render():
hf_token = request.args.get('hf_token', '', type=str)
+ text = request.args.get('text', 'Hello!!', type=str)
+ selected_model = request.args.get('selected_model', 'Model 1 (Turbo Realism)', type=str)
- return render_controller.get_render(hf_token)
+ return render_controller.get_render(hf_token, text, selected_model)
+ # Suggest
@app.route('/suggest', methods=['GET'])
def get_suggest():
hf_token = request.args.get('hf_token', '', type=str)
--
GitLab