Update app.py

app.py

@@ -48,9 +48,6 @@ spaBERT_model.eval()
 
 
 
-
-
-
 st.title("SpaGAN Demo")
 st.write("Enter a text, and the system will highlight the geo-entities within it.")
 
@@ -68,16 +65,28 @@ for label, (color, description) in COLOR_MAP.items():
     st.markdown(f"- **{label}**: <span style='color:{color}'>{color}</span> - {description}", unsafe_allow_html=True)
 
 # Text input
-user_input = st.text_area("Input Text", height=200)
+#user_input = st.text_area("Input Text", height=200)
+
+# Define example reviews for testing
+example_reviews = {
+    "Review 1": "I visited the Empire State Building in New York last summer, and it was amazing!",
+    "Review 2": "Google, headquartered in Mountain View, is a leading tech company in the United States.",
+}
+
+# Dropdown for selecting an example review
+user_input = st.selectbox("Select an example review", options=list(example_reviews.keys()))
+
+# Get the selected review text
+selected_review = example_reviews[user_input]
 
 # Process the text when the button is clicked
 if st.button("Highlight Geo-Entities"):
-    if user_input.strip():
+    if selected_review.strip():
         # Process the text using spaCy
-        doc = nlp(user_input)
+        doc = nlp(selected_review)
 
         # Highlight geo-entities with different colors
-        highlighted_text = user_input
+        highlighted_text = selected_review
         for ent in reversed(doc.ents):
             if ent.label_ in COLOR_MAP:
                 color = COLOR_MAP[ent.label_][0]
@@ -90,4 +99,4 @@ if st.button("Highlight Geo-Entities"):
         # Display the highlighted text with HTML support
         st.markdown(highlighted_text, unsafe_allow_html=True)
     else:
-        st.error("Please enter some text.")
+        st.error("Please select a review.")