Update app.py

app.py

@@ -19,4 +19,26 @@ if user_input:
     with st.spinner('Generating response...'):
         response = pipe(contexted_ipnut)
         st.write("Response:")
-        st.text(response)
+        # Initialize the highlighted text
+        highlighted_text = ""
+        last_position = 0
+
+        # Process each entity and highlight the labeled words
+        for entity in response:
+            start, end = entity['start'], entity['end']
+            label = entity['entity']
+
+            # Add text before the entity
+            highlighted_text += user_input[last_position:start]
+            
+            # Add the highlighted entity
+            highlighted_text += f'<mark style="background-color: #FFFF00;">{user_input[start:end]}</mark><sup>{label}</sup>'
+
+            # Update the last position
+            last_position = end
+
+        # Add remaining text after the last entity
+        highlighted_text += user_input[last_position:]
+
+        # Display the highlighted text using st.markdown
+        st.markdown(highlighted_text, unsafe_allow_html=True)