Update app.py

app.py

@@ -85,11 +85,14 @@ with gr.Blocks() as block_demo:
 
     def refresh_comparison():
         return  get_random_comparison()
-    def on_load(request: gr.Request):
-        headers = request.headers
-        host = request.client.host
-        request_state = dict(headers)
-        request_state['host'] = host
+    def on_load(request: gr.Request=None, 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)
         image, method1, method2, image1, image2, property, image_input = refresh_comparison()
@@ -135,7 +138,7 @@ with gr.Blocks() as block_demo:
         # if type(choice) is not str : choice = choice.value
         print(choice, image, method1, method2, property, ip)
         send_message_to_mongodb(image, property, method1, method2, choice, ip)
-        img1, img2, prop_text, image_state, method1_state, method2_state, property_state, img_input, ip_state = on_load()
+        img1, img2, prop_text, image_state, method1_state, method2_state, property_state, img_input, ip_state = on_load(request_head=ip)
         # new_image, new_method1, new_method2, new_image1, new_image2, new_property = get_random_comparison()
         return [
             img1, img2, prop_text, image_state, method1_state, method2_state, property_state, img_input, ip_state