Update app.py

app.py

@@ -2,7 +2,15 @@ import json
 import streamlit as st
 from google.oauth2 import service_account
 from google.cloud import language_v1
+import requests
 
+# Adding checkbox options for entity types
+entity_types_to_show = ["PERSON", "ORGANIZATION", "EVENT"]
+selected_types = st.multiselect('Select entity types to show:', entity_types_to_show)
+
+# Function for querying Google Knowledge Graph API
+def query_google_knowledge_graph(entity_name):
+    return {"info": "Knowledge Graph info for {}".format(entity_name)}
 
 # Header and intro
 st.title("Google Cloud NLP Entity Analyzer")
@@ -31,6 +39,7 @@ def sample_analyze_entities(text_content, your_query=""):
     entities_list = []
     for entity in response.entities:
         entity_type_name = language_v1.Entity.Type(entity.type_).name
+        if entity_type_name in selected_types:
             entity_details = {
                 "Name": entity.name,
                 "Type": entity_type_name,
@@ -53,6 +62,11 @@ def sample_analyze_entities(text_content, your_query=""):
                 st.write(f"**{key}:**")
                 st.write(value)
                 
+        # Query Google Knowledge Graph API for each entity
+        kg_info = query_google_knowledge_graph(entity['Name'])
+        st.write("### Google Knowledge Graph Information")
+        st.write(kg_info)
+
         st.write("----")
 
     st.write(f"### Language of the text: {response.language}")