Update app.py - add sample example

app.py

@@ -1,6 +1,16 @@
 import streamlit as st
 from ispice import Spice
 
+
+# Sample example captions
+sample_examples = {
+    "ID - Example 1": {
+        "generated": "P1 sits on the couch. P1 puts her hands on the couch and sits up. P2 sits beside her.",
+        "reference": "P1 straightens her posture. P1 sits up. P2 sits beside her on the couch."
+    }
+}
+
+
 # Function to compute score
 def preprocess_captions(generated_captions, reference_captions):
     hypotheses = {'image'+str(i): [generated_captions[i]] for i in range(len(generated_captions))}
@@ -16,16 +26,29 @@ def main():
     
     spice_scorer = Spice(mode=mode)
     
+    # Sample examples dropdown
+    st.sidebar.subheader("Choose Sample Example:")
+    example_choice = st.sidebar.selectbox("Select Example", [""] + list(sample_examples.keys()), index=0)
+
+    # Initialize with empty strings
+    generated_caption = ""
+    reference_caption = ""
+    
+    if example_choice:
+        generated_caption = sample_examples[example_choice]["generated"]
+        reference_caption = sample_examples[example_choice]["reference"]
+    
      # Description
     st.write("You can either input single caption or multiple captions separated by new line.")
     # Input text boxes
-    generated_caption = st.text_area("Generated Caption:", "")
-    reference_caption = st.text_area("Reference Caption:", "")
+    generated_caption_input = st.text_area("Generated Caption:", value=generated_caption)
+    reference_caption_input = st.text_area("Reference Caption:", value=reference_caption)
+    
     
     # Compute score button
     if st.button("Compute Score"):
-        generated_captions = generated_caption.split("\n")
-        reference_captions = reference_caption.split("\n")
+        generated_captions = generated_caption_input.split("\n")
+        reference_captions = reference_caption_input.split("\n")
         
         print(generated_captions, len(generated_captions))
         print(reference_captions, len(reference_captions))