Update app.py

app.py

@@ -13,8 +13,8 @@ import pandas as pd
 rekaeval = "RekaAI/VibeEval"
 dataset = load_dataset(rekaeval, split="test")
 df = pd.DataFrame(dataset)
-df = df[['media_url', 'prompt']]
-df_markdown = df.copy()
+df = df[['media_url', 'prompt', 'reference']]
+df_markdown = df[['media_url', 'prompt']].copy()
 
 # Function to convert URL to HTML img tag
 def mediaurl_to_img_tag(url):
@@ -132,6 +132,8 @@ with gr.Blocks(fill_height=True, css=CSS) as demo:
             chatbot=chatbot,
         )
     with gr.Column():
+      with gr.Accordion('Open for looking at Ground Truth:', open=False):
+          refernce = gr.Markdown()
       with gr.Row():
         b1 = gr.Button("Previous", interactive=False)
         b2 = gr.Button("Next")
@@ -145,10 +147,11 @@ with gr.Blocks(fill_height=True, css=CSS) as demo:
       x = evt.index[0] + start
       image = df.iloc[x, 0]  
       prompt = df.iloc[x, 1]
+      refrence = df.iloc[x, 2]
       print(f'image = {image}')
       print(f'prompt = {prompt}')
       example = {"text": prompt, "files": [image]}
-      return example
+      return example, refernce
 
   def display_next(dataframe, end):
     print(f'initial value of end = {end}')
@@ -168,7 +171,7 @@ with gr.Blocks(fill_height=True, css=CSS) as demo:
     print(f'returned value of end = {end}')
     return df_images, end, start, gr.Button(interactive=False) if start==0 else gr.Button(interactive=True)
 
-  reka.select(get_example, [reka,num_start], chat_input, show_progress="hidden")
+  reka.select(get_example, [reka,num_start], [chat_input, refernce], show_progress="hidden")
   b2.click(fn=display_next, inputs= [reka, num_end ], outputs=[reka, num_end, num_start, b1], api_name="next_rows", show_progress=False)
   b1.click(fn=display_previous, inputs= [reka, num_start ], outputs=[reka, num_end, num_start, b1], api_name="previous_rows")