Update app.py

app.py

@@ -13,7 +13,6 @@ def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
 
     payload = {"input": {}}
     
-    
     base_url = "http://0.0.0.0:7860"
     for i, key in enumerate(names):
         value = args[i]
@@ -24,7 +23,6 @@ def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
 
     response = requests.post("http://0.0.0.0:5000/predictions", headers=headers, json=payload)
 
-    
     if response.status_code == 201:
         follow_up_url = response.json()["urls"]["get"]
         response = requests.get(follow_up_url, headers=headers)
@@ -35,7 +33,6 @@ def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
             time.sleep(1)
     if response.status_code == 200:
         json_response = response.json()
-        #If the output component is JSON return the entire output response 
         if(outputs[0].get_config()["name"] == "json"):
             return json_response["output"]
         predict_outputs = parse_outputs(json_response["output"])
@@ -47,16 +44,58 @@ def predict(request: gr.Request, *args, progress=gr.Progress(track_tqdm=True)):
             raise gr.Error(f"Sorry, the Cog image is still processing. Try again in a bit.")
         raise gr.Error(f"The submission failed! Error: {response.status_code}")
 
+def check_password(password):
+    return password == "pixo"
 
 css = '''
-#top{position: fixed;}
+body {
+    font-family: Arial, sans-serif;
+    background-color: #f0f0f0;
+}
+.container {
+    max-width: 1200px;
+    margin: 0 auto;
+    padding: 20px;
+    background-color: white;
+    border-radius: 10px;
+    box-shadow: 0 0 10px rgba(0,0,0,0.1);
+}
+.gradio-slider input[type="range"] {
+    accent-color: #3498db;
+}
+.gradio-button {
+    background-color: #3498db;
+    color: white;
+    border: none;
+    padding: 10px 20px;
+    border-radius: 5px;
+    cursor: pointer;
+    transition: background-color 0.3s;
+}
+.gradio-button:hover {
+    background-color: #2980b9;
+}
+#top {
+    position: fixed;
+    top: 20px;
+    right: 20px;
+    max-width: 300px;
+    border-radius: 10px;
+    box-shadow: 0 0 10px rgba(0,0,0,0.1);
+}
 '''
-with gr.Blocks() as demo:
+
+with gr.Blocks(css=css) as demo:
+    gr.Markdown("# Expression Editor")
+    
     with gr.Column():
-        gr.Markdown("# Expression Editor")
+        password = gr.Textbox(type="password", label="Enter password")
+        login_button = gr.Button("Login")
+    
+    with gr.Column(visible=False) as main_interface:
         gr.Markdown("Demo for expression-editor cog image by fofr")
         with gr.Row():
-            with gr.Column():
+            with gr.Column(elem_classes="container"):
                 image = gr.Image(
                     label="Input image", 
                     type="filepath",
@@ -150,23 +189,25 @@ with gr.Blocks() as demo:
     inputs = [image, rotate_pitch, rotate_yaw, rotate_roll, blink, eyebrow, wink, pupil_x, pupil_y, aaa, eee, woo, smile, src_ratio, sample_ratio, crop_factor, output_format, output_quality]
     outputs = [result_image]
 
+    def login(password):
+        if check_password(password):
+            return gr.update(visible=True), gr.update(visible=False)
+        else:
+            return gr.update(visible=False), gr.update(visible=True, value="")
+
+    login_button.click(
+        fn=login,
+        inputs=[password],
+        outputs=[main_interface, password]
+    )
+
     submit_btn.click(
         fn=predict,
         inputs=inputs,
         outputs=outputs,
     )
 
-    rotate_pitch.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
-    rotate_yaw.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
-    rotate_roll.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
-    blink.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
-    eyebrow.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
-    wink.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
-    pupil_x.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
-    pupil_y.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
-    aaa.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
-    eee.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
-    woo.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
-    smile.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
+    for slider in [rotate_pitch, rotate_yaw, rotate_roll, blink, eyebrow, wink, pupil_x, pupil_y, aaa, eee, woo, smile]:
+        slider.release(fn=predict, inputs=inputs, outputs=outputs, show_progress="minimal")
 
 demo.launch(share=False, show_error=True)