diff --git a/src/controllers/glb_controller.py b/src/controllers/glb_controller.py index 2289ca676f6227a8c4ab03cbdb83ac9c5582c315..84e859ba7fed5ff744e54cb4c3f83ecfa9390730 100644 --- a/src/controllers/glb_controller.py +++ b/src/controllers/glb_controller.py @@ -1,11 +1,12 @@ import os from gradio_client import Client, handle_file +from flask import jsonify, Response class GlbController: def get_glb(self, hf_token='', base_image_url='', ss_guidance_strength=7.5, ss_sampling_steps=12, slat_guidance_strength=3, slat_sampling_steps=12, multiimage_algo='stochastic', - mesh_simplify=0.95, texture_size=1024) -> str: + mesh_simplify=0.95, texture_size=1024) -> Response: if hf_token == '': hf_token = os.getenv("hf_token") @@ -44,4 +45,4 @@ class GlbController: api_name="/extract_glb" ) - return extract_glb + return jsonify({"path": extract_glb}) diff --git a/src/controllers/render_controller.py b/src/controllers/render_controller.py index af5205abb4daed88d429f95b4bd197b945fe82df..5c2e9bacc2b0e78bec23d74e307db3fb5e6a7a5c 100644 --- a/src/controllers/render_controller.py +++ b/src/controllers/render_controller.py @@ -1,4 +1,5 @@ import os +from flask import jsonify, Response from gradio_client import Client, handle_file class RenderController: @@ -7,7 +8,7 @@ class RenderController: negative_prompt="sketch, drawing", instruct_query='use everything from the image', scale=0.8, - nfaa_negative_prompt='naked, bikini, skimpy, scanty, bare skin, lingerie, swimsuit, exposed, see-through') -> str: + nfaa_negative_prompt='naked, bikini, skimpy, scanty, bare skin, lingerie, swimsuit, exposed, see-through') -> Response: client = Client("unity/IP-Adapter-Instruct") @@ -21,4 +22,4 @@ class RenderController: api_name="/generate_image" ) - return result + return jsonify({"result": result}) diff --git a/src/controllers/suggest_controller.py b/src/controllers/suggest_controller.py index 25aa1b250047bc916961ba9e58f0d2e0acc304a6..a881edee8041e6954ac6e1f2a9ee84ece1f8c82e 100644 --- a/src/controllers/suggest_controller.py +++ b/src/controllers/suggest_controller.py @@ -1,10 +1,11 @@ import os +from flask import jsonify, Response from gradio_client import Client class SuggestController: def get_suggest(self, hf_token='', message='', system_prompt='', temperature=.0, max_new_tokens=0, top_k_sampling=0, repetition_penalty=.0, - top_p_sampling=.0) -> str: + top_p_sampling=.0) -> Response: if hf_token == '': hf_token = os.getenv("hf_token") @@ -23,4 +24,4 @@ class SuggestController: print(result) - return result + return jsonify({"message": result})