diff --git a/src/controllers/render_controller.py b/src/controllers/render_controller.py index 2ba12c0736f796fd23b232030b58eab77ce29cf1..af5205abb4daed88d429f95b4bd197b945fe82df 100644 --- a/src/controllers/render_controller.py +++ b/src/controllers/render_controller.py @@ -1,18 +1,24 @@ import os -from gradio_client import Client +from gradio_client import Client, handle_file class RenderController: - def get_render(self, hf_token='', text="Hello!!", - selected_model="Model 1 (Turbo Realism)") -> str: + def get_render(self, hf_token='', base_image_url='', + prompt='A realistic photo of a building', + 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: - client = Client("Jonny001/Text-to-Image") + client = Client("unity/IP-Adapter-Instruct") result = client.predict( - text=text, - selected_model=selected_model, - api_name="/generate_images" + images=handle_file(base_image_url), + prompt=prompt, + negative_prompt=negative_prompt, + instruct_query=instruct_query, + scale=scale, + nfaa_negative_prompt=nfaa_negative_prompt, + api_name="/generate_image" ) - print(result) - - return str(result) + return result diff --git a/src/routes/router.py b/src/routes/router.py index f5ece6bc883c1b9f3a35ed1e3bac16b692fcf077..bb778bf622d211a80b845ef59c63ed2d3e9f837e 100644 --- a/src/routes/router.py +++ b/src/routes/router.py @@ -11,12 +11,12 @@ 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(): - base_image_url = request.args.get('base_image_url', type=str) hf_token = request.args.get('hf_token', '', type=str) + base_image_url = request.args.get('base_image_url', type=str) ss_guidance_strength = request.args.get('ss_guidance_strength', 7.5, type=float) ss_sampling_steps = request.args.get('ss_sampling_steps', 12, type=int) slat_guidance_strength = request.args.get('slat_guidance_strength', 3, type=float) @@ -34,10 +34,15 @@ def setup_routes(app): @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, text, selected_model) + base_image_url = request.args.get('base_image_url', '', type=str) + prompt = request.args.get('prompt', 'A realistic photo of a building', type=str) + negative_prompt = request.args.get('negative_prompt', 'sketch, drawing', type=str) + instruct_query = request.args.get('instruct_query', 'use everything from the image', type=str) + scale = request.args.get('scale', 0.8, type=float) + nfaa_negative_prompt = request.args.get('nfaa_negative_prompt', 'naked, bikini, skimpy, scanty, bare skin, lingerie, swimsuit, exposed, see-through', type=str) + + return render_controller.get_render(hf_token, base_image_url, prompt, negative_prompt, + instruct_query, scale, nfaa_negative_prompt) # Suggest @app.route('/suggest', methods=['GET'])