save code

change_app.py

@@ -289,9 +289,14 @@ def change(human_name, cloth_name, c_width, c_height, cloth_url, human_url, no2,
     if queue["queue_running"] or queue["queue_pending"]:
         return None, None, sex_girl, "cur gpu is busy, Shou not happen"
 
-    print(f'开始换衣服 {change_only_name} cloth_len:{cloth_len} cloth_short:{cloth_short}')
-    with open(f'{APP_ROOT}/{change_only_name}', 'r', encoding='utf-8') as file:
-        prompt_text = file.read()
+    if tuodi:
+        print(f'开始换衣服 {change_tuodi_input} cloth_len:{cloth_len} cloth_short:{cloth_short}')
+        with open(f'{APP_ROOT}/{change_tuodi_input}', 'r', encoding='utf-8') as file:
+            prompt_text = file.read()
+    else:
+        print(f'开始换衣服 {change_only_name} cloth_len:{cloth_len} cloth_short:{cloth_short}')
+        with open(f'{APP_ROOT}/{change_only_name}', 'r', encoding='utf-8') as file:
+            prompt_text = file.read()
 
     prompt = json.loads(prompt_text)
 

config.py

@@ -40,11 +40,13 @@ logger.addHandler(console_handler)
 selfComputer = "5090"
 
 if selfComputer == "5090":
+    change_tuodi_input = 'change_tuodi_input_api.json'
     change_only_name = "change_new_only_change.json"
     change_with_add_cloth = "change_new_debug_api.json"
     xiezhen_name = "xiezhen-0812-40s_api.json"
     base64_test_port = 28888
 else:
+    change_tuodi_input = 'change_tuodi_input_3090_api.json'
     change_only_name = "change_new_only_3090.json"
     change_with_add_cloth = "change_new_3090_api.json"
     xiezhen_name = "xiezhen-0812-40s_api.json"