update consistency value

app.py

@@ -24,8 +24,6 @@ TOKEN = os.environ.get("TOKEN")
 
 
 
-
-
 server_address = remote_url
 client_id = str(uuid.uuid4())
 
@@ -52,10 +50,10 @@ def get_queue_status(server_address):
         print(f"Error retrieving queue info: {e}")
         return {'queue_running': 0, 'queue_pending': 0}  # Return 0s if there is an error or connection issue
 
-def queue_prompt(workflow, bg_color, pos_prompt, password, images_base64, id_strength, consistency_magic_input):
+def queue_prompt(workflow, bg_color, pos_prompt, password, images_base64, id_strength, consistency_magic):
     p = {"prompt": {"dummy":"dummy"}, "client_id": client_id, "api":"silosnap_portrait", "workflow":workflow, 
          "bg_color":bg_color, "pos_prompt":pos_prompt, "pass":password, "upimages":images_base64,
-         "id_strength":id_strength, "consistency_magic_input":consistency_magic_input}
+         "id_strength":id_strength, "consistency_magic":consistency_magic}
     data = json.dumps(p).encode('utf-8')
     req =  urllib.request.Request("http://{}/prompt?token={}".format(server_address, TOKEN), data=data)
     # req =  urllib.request.Request("http://{}/prompt".format(server_address), data=data, headers={'Content-Type': 'application/json'})
@@ -73,8 +71,8 @@ def get_history(prompt_id):
     with urllib.request.urlopen("http://{}/history/{}?token={}".format(server_address, prompt_id, TOKEN)) as response:
         return json.loads(response.read())
 
-def get_images(ws, workflow, bg_color, pos_prompt, password, images_base64, id_strength, consistency_magic_input):
-    prompt_id = queue_prompt(workflow, bg_color, pos_prompt, password, images_base64, id_strength, consistency_magic_input)['prompt_id']
+def get_images(ws, workflow, bg_color, pos_prompt, password, images_base64, id_strength, consistency_magic):
+    prompt_id = queue_prompt(workflow, bg_color, pos_prompt, password, images_base64, id_strength, consistency_magic)['prompt_id']
     output_images = {}
     while True:
         out = ws.recv()
@@ -105,7 +103,7 @@ def get_images(ws, workflow, bg_color, pos_prompt, password, images_base64, id_s
 #------------------------------------------------------------
 
 def main():
-    st.title("SILO.AI - AMDsnap ver.1.1.leon ")
+    st.title("SILO.AI - AMDsnap ver.1.2.leon ✨")
     
     workflow = 'amdsnap_style'
     bg_color = "white"
@@ -152,8 +150,8 @@ def main():
 
 
     # Number input for consistency magic value    
-    consistency_magic_input = st.number_input(
-        "🦄 Consistency Magic Value (1 ~ 4,294,967,294):",
+    consistency_magic = st.number_input(
+        "🦄 Style Consistency Magic Value (1 ~ 4,294,967,294):",
         min_value=1,
         max_value=4294967294,
         value=None,
@@ -200,7 +198,7 @@ def main():
                     st.error(f"Please check your server address: {server_address}")
                     return
 
-                images = get_images(ws, workflow, bg_color, text_prompt, password, images_base64, id_strength, consistency_magic_input)
+                images = get_images(ws, workflow, bg_color, text_prompt, password, images_base64, id_strength, consistency_magic)
                 ws.close()
                 gif_runner.empty()  # Remove the GIF after processing is done
                 
@@ -221,5 +219,5 @@ def main():
                 st.write("No images returned.")
 
 if __name__ == "__main__":
-    # main()
-    st.title("Updating.. wiat a 10~20 min, sorry! ")
+    main()
+    # st.title("Updating.. wiat a 10~20 min, sorry! ")