Update app.py

app.py

@@ -12,7 +12,7 @@ from torch.utils.data import DataLoader
 from PIL import Image
 
 device = torch.device('cpu')
-dev_mode = True
+dev_mode = False
 
 
 #Spacy Initialization Section
@@ -240,11 +240,21 @@ for label, (color, description) in COLOR_MAP.items():
 review_file_path = "models/spabert/datasets/SampleReviews.txt"
 example_reviews = load_reviews_from_file(review_file_path)
 
+# Define labels
+review_labels = {
+    "Review 1": "Real",
+    "Review 2": "Spam",
+}
+
+# Create options with labels for the dropdown
+dropdown_options = [f"{key} ({review_labels.get(key, 'Unknown')})" for key in example_reviews.keys()]
+
 # Dropdown for selecting an example review
-user_input = st.selectbox("Select an example review", options=list(example_reviews.keys()))
+user_selection = st.selectbox("Select an example review", options=dropdown_options)
 
-# Get the selected review text
-selected_review = example_reviews[user_input]
+# Extract the original review key from the selected option
+selected_key = user_selection.split(" (")[0]  # Remove the label part
+selected_review = example_reviews[selected_key]
 
 # Process the text when the button is clicked
 if st.button("Highlight Geo-Entities"):