init

app.py

@@ -43,8 +43,12 @@ def is_http_resource_accessible(url):
         return False
 
 
-def onPoseChange(prompt_text, source_image, request: gr.Request):
+def onPoseChange(prompt_text, source_image, token_input, request: gr.Request):
     """Handle pose change request"""
+    # Check token first
+    if token_input != POSEToken:
+        return "please input the correct token!", None, None, None
+    
     if source_image is None:
         return "Please provide source image first!", None, None, None
     
@@ -228,7 +232,17 @@ with gr.Blocks() as demo:
 
     # Add pose changer module
     with gr.Accordion('pose changer', open=False):
-        # Top: text box and button
+        # Top: token input
+        with gr.Row():
+            token_input = gr.Textbox(
+                value="",
+                label="Access Token",
+                placeholder="请输入token...",
+                type="password",
+                scale=1
+            )
+        
+        # Middle: text box and button
         with gr.Row():
             pose_prompt = gr.Textbox(
                 value="Change the pose: hands on hips.#Change the pose: arms extended.",
@@ -257,7 +271,7 @@ with gr.Blocks() as demo:
     # Bind pose changer change button
     change_button.click(
         fn=onPoseChange,
-        inputs=[pose_prompt, pose_changer_image],
+        inputs=[pose_prompt, pose_changer_image, token_input],
         outputs=[pose_change_info, pose_result_1, pose_result_2, pose_result_3]
     )
 

utils.py

@@ -15,6 +15,7 @@ OssUrl = "https://selfit-deploy-1256039085.cos.accelerate.myqcloud.com/"
 Regions = "IndiaPakistanBengal"
 TOKEN = os.environ['TOKEN']
 UKAPIURL = os.environ['UKAPIURL']
+POSEToken = os.environ['POSEToken']
 
 
 proj_dir = os.path.dirname(os.path.abspath(__file__))