Update app.py

app.py

@@ -59,30 +59,36 @@ def evaluate_caption(image, caption):
     # ,gr.Image(type="pil", value="mukherjee_kushin_WIDPICS1.jpg")
 
 callback = gr.HuggingFaceDatasetSaver('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', "gradioTest")
-with gr.Blocks() as demo:
+
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
   gr.Markdown(
     """
     # Welcome to our Human vs. AI game!
-    
-    ### You and an AI agent are trying to convince a third AI agent that each of you are better at describing the visual world. \n
-    ### In order to win, describe this image in one sentence. Then the second AI agent will also generate a description and the third agent will decide a winner.
-    ### You win if the AI says that "Player 1 wins!"
+
+   You and an AI agent are trying to convince a third AI agent that each of you are better at describing the visual world. \n
+   In order to win, describe this image in one sentence. Then the second AI agent will also generate a description and the third agent will decide a winner.
+   You win if the AI says that "Player 1 wins!"
     """)
-  im_path_str = 'n01677366_12918.JPEG'
+  # im_path_str = 'n03418158_2886.JPEG'
+  im_path_str = random.choice(os.listdir(im_dir))
+  
   im_path = gr.Textbox(label="Image fname",value=im_path_str,interactive=False, visible=False)
   # fn=evaluate_caption,
   # inputs=["image", "text"]
- 
-  
-  with gr.Column():
-    im = gr.Image(label="Target Image", interactive = False, type="pil",value =os.path.join(im_dir,im_path_str),height=500)
-    caps = gr.Textbox(label="Player 1 Caption")
-    submit_btn = gr.Button("Submit!!")
+
+
+  with gr.Row():
+    im = gr.Image(label="Target Image", interactive = False, type="pil",value =os.path.join(im_dir,im_path_str),height=400)
+    with gr.Column():
+      caps = gr.Textbox(label="Player 1 Caption")
+      submit_btn = gr.Button("Submit!!")
+      out1 = gr.Textbox(label="Player 2 (Machine) Caption",interactive=False)
+    
   # outputs=["text","text"],
-  with gr.Column():
-    out1 = gr.Textbox(label="Player 2 (Machine) Caption",interactive=False)
-    out2 = gr.Textbox(label="Winner",interactive=False)
-    reload_btn = gr.Button("Next Image")
+  with gr.Row():
+    with gr.Column():
+      out2 = gr.Textbox(label="Winner",interactive=False)
+      reload_btn = gr.Button("Next Image")
 
 
   # live=False,
@@ -95,4 +101,4 @@ with gr.Blocks() as demo:
   #     btn = gr.Button("Flag")
   # btn.click(lambda *args: callback.flag(args), [im, caps, out1, out2], None, preprocess=False)
 
-demo.launch()
+demo.launch()