Update app.py

app.py

@@ -176,7 +176,7 @@ model = ModelManager(
     )
 
 image_type = 'pil' # filepath' 
-def predict(input, img_enhancer):
+def predict(input, img_enhancer):    
     if input is None:
         return None
     if image_type == 'filepath':
@@ -231,7 +231,7 @@ css = '''
     display: none !important;
 }
 '''
-set_page_elements = """async () => {
+set_page_elements = """async() => {
     function isMobile() {
         try {
             document.createEvent("TouchEvent"); return true;
@@ -260,13 +260,12 @@ set_page_elements = """async () => {
         const enhancer = gradioEl.querySelectorAll('#enhancer-checkbox')[0];  
         enhancer.style.display = "none";
         const share_btn_container = gradioEl.querySelectorAll('#share-btn-container')[0];  
-        share_btn_container.setAttribute('style', 'width: 0px; height:0px;');        
+        share_btn_container.setAttribute('style', 'width: 0px; height:0px;'); 
     } else {
         max_height = 800;
         image_upload.setAttribute('style', 'min-height: ' + max_height + 'px; overflow-x: overlay');
         data_image.setAttribute('style', 'height: ' + max_height + 'px');
         data_image_div.setAttribute('style', 'min-height: ' + max_height + 'px');
-        image_output.setAttribute('style', 'width: ' + max_height + 'px'); 
     }
     group1.style.display = "none";
     group2.style.display = "block";
@@ -277,19 +276,32 @@ image_blocks = gr.Blocks(css=css, title='修图神器')
 with image_blocks as demo:
     with gr.Group(elem_id="group_1", visible=True) as group_1:
         with gr.Box():
-            with gr.Row():
-                with gr.Column():
+            with gr.Row(elem_id="gallery_row"):
+                with gr.Column(elem_id="gallery_col"):
                     gallery = gr.Gallery(value=['./sample_00.jpg','./sample_00_e.jpg'], show_label=False)
-                    gallery.style(grid=[2], width=320)                                 
+                    gallery.style(grid=[2], height='500px')                                 
             with gr.Row():
                 with gr.Column():
                     begin_button = gr.Button("Let's GO!", elem_id="begin-btn", visible=True) 
+            with gr.Row():
+                with gr.Column():
+                    gr.HTML("""
+                            <div style='margin: 0 auto; text-align: center;color:red;'>
+                                <p>
+                                Solemnly promise: this application will not collect any user information and image resources.
+                                </p>
+                                <p>
+                                【郑重承诺：本应用不收集任何用户信息和图片资源。】
+                                </p>                                
+                            </div>
+                        """
+                    )
 
     with gr.Group(elem_id="group_2", visible=False) as group_2:    
         with gr.Box(elem_id="work-container"):
             with gr.Row(elem_id="input-container"):
                 with gr.Column():
-                    image = gr.Image(source='upload', elem_id="image_upload",tool='sketch', type=f'{image_type}', 
+                    image_input = gr.Image(source='upload', elem_id="image_upload",tool='sketch', type=f'{image_type}', 
                                      label="Upload(载入图片)", show_label=False).style(mobile_collapse=False)
             with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
                 with gr.Column(elem_id="erase-container"):
@@ -308,8 +320,8 @@ with image_blocks as demo:
                         community_icon = gr.HTML(community_icon_html, visible=False)
                         loading_icon = gr.HTML(loading_icon_html, visible=False)
                         share_button = gr.Button("Share to community", elem_id="share-btn", visible=False)                    
-                    
-            btn_erase.click(fn=predict, inputs=[image, img_enhancer], outputs=[image_out, community_icon, loading_icon, share_button])
+   
+            btn_erase.click(fn=predict, inputs=[image_input, img_enhancer], outputs=[image_out, community_icon, loading_icon, share_button])
             share_button.click(None, [], [], _js=share_js)
     
         begin_button.click(fn=None, inputs=[], outputs=[group_1, group_2], _js=set_page_elements)