app.py: refined arguments preprocessing.

app.py

@@ -8,7 +8,7 @@ from io import BytesIO
 import PIL.Image
 from typing import Tuple
 from novita_client import NovitaClient, V3TaskResponseStatus
-from time import sleep
+from time import time
 
 from style_template import styles
 
@@ -118,6 +118,8 @@ CONTROLNET_DICT = dict(
 	},
 )
 
+last_check = 0
+
 
 
 def get_novita_client (novita_key):
@@ -242,12 +244,13 @@ def generate_image (
 		raise gr.Error(f'Cannot find any input face image! Please refer to step 1️⃣')
 
 	#print('novita_key:', novita_key1)
-	print('face_image_path:', face_image_path)
+	#print('face_image_path:', face_image_path)
 	if not novita_key1:
 		raise gr.Error(f'Please input your Novita Key!')
 	try:
 		client = get_novita_client(novita_key1)
 		prompt, negative_prompt = apply_style(style_name, prompt, negative_prompt)
+		prompt = prompt[:1024]
 		#print('prompt:', prompt)
 		#print('negative_prompt:', negative_prompt)
 		#print('seed:', seed)
@@ -296,11 +299,16 @@ def generate_image (
 
 		print('task_id:', res['task_id'])
 		def progress (x):
-			print('progress:', x.task.status)
+			global last_check
+			t = time()
+			if t > last_check + 5:
+				last_check = t
+				print('progress:', t, x.task.status)
 		final_res = client.wait_for_task_v3(res['task_id'], callback=progress)
 		if final_res is None or final_res.task.status == V3TaskResponseStatus.TASK_STATUS_FAILED:
 			raise RuntimeError(f'Novita task failed: {final_res and final_res.task.status}')
 		print('status:', final_res.task.status)
+		print('returned images:', final_res.images)
 
 		final_res.download_images()
 	except Exception as e:
@@ -452,8 +460,8 @@ with gr.Blocks(css=css) as demo:
 				)
 				guidance_scale = gr.Slider(
 					label='Guidance scale',
-					minimum=0.1,
-					maximum=20.0,
+					minimum=1.,
+					maximum=30.0,
 					step=0.1,
 					value=0.0 if enable_lcm_arg else 5.0,
 				)