Update app.py

app.py

@@ -85,19 +85,20 @@ with gr.Blocks() as block_demo:
 
     def refresh_comparison():
         return  get_random_comparison()
-    def on_load(request, request_head=None):
-        if request_head is None:
-            headers = request.headers
-            host = request.client.host
-            request_state = dict(headers)
-            request_state['host'] = host
-        else :
-            request_state = request_head
-        # print(str(request))
-        print(request_state)
+
+    def prepare_everything_else():
         image, method1, method2, image1, image2, property, image_input = refresh_comparison()
         return image1, image2, f"<h2 style='font-size: 24px;'>Which one <mark class='red'>{property_dict[property]}</mark>?</h2>",\
-          image, method1, method2, property, image_input, request_state
+          image, method1, method2, property, image_input
+    def on_load(request, request_head=None):
+
+        headers = request.headers
+        host = request.client.host
+        request_state = dict(headers)
+        request_state['host'] = host
+
+        return **prepare_everything_else(), request_state
+        
     gr.Markdown("<h2 align='center',style='font-size: 24px;'>Low-light Image Enhancer Arena 🥊</h2>")
     gr.Markdown("<p align='center', style='font-size: 18px;'>LIME-Eval is an arena to ask human-beings to judge the performance of different low-light image enhancers with respect to </p>")
     gr.Markdown("<p align='center', style='font-size: 18px;'>different factors, including Artifact, Color Degradation, Noise, Poor Illumination, Blur, and Overall quality.</p>")
@@ -121,7 +122,7 @@ with gr.Blocks() as block_demo:
     
     prop_text = gr.Markdown(f'###Which one is better in terms of x?')
     image_state, method1_state, method2_state, property_state, ip_state = gr.State(), gr.State(), gr.State(), gr.State(), gr.State()
-    block_demo.load(on_load, inputs=[None], outputs=[img1, img2, prop_text, 
+    block_demo.load(on_load, inputs=[], outputs=[img1, img2, prop_text, 
                                                    image_state, method1_state, method2_state, property_state, img_input, ip_state])
     with gr.Row():
         l_butt = gr.Button("Image 1")