Update app.py

app.py

@@ -3,6 +3,22 @@ import streamlit as st
 from google.oauth2 import service_account
 from google.cloud import language_v1
 
+# Adding checkbox options for entity types
+entity_types_to_show = [
+    "UNKNOWN", "PERSON", "LOCATION", "ORGANIZATION", "EVENT", "WORK_OF_ART", "CONSUMER_GOOD", "OTHER"
+]
+selected_types = st.multiselect('Select entity types to show:', entity_types_to_show)
+
+def sample_analyze_entities(text_content, selected_types):
+    # Existing code for setting up credentials and client...
+
+    # Create an empty list to hold the results
+    entities_list = []
+
+    for entity in response.entities:
+        entity_type_name = language_v1.Entity.Type(entity.type_).name
+        if entity_type_name in selected_types:
+
 # Header and intro
 st.title("Google Cloud NLP Entity Analyzer")
 st.write("## Introduction to the Knowledge Graph API")