Update app.py

app.py

@@ -5,23 +5,15 @@ from google.cloud import language_v1
 import urllib.parse
 import urllib.request
 
+# Function to query Google's Knowledge Graph API
 # Function to query Google's Knowledge Graph API
 def query_knowledge_graph(entity_id):
     try:
-        service_url = 'https://kgsearch.googleapis.com/v1/entities:search'
-        params = {
-            'ids': entity_id,
-            'limit': 1,
-            'indent': True,
-            'key': st.secrets["kg_api"],
-        }
-        url = service_url + '?' + urllib.parse.urlencode(params)
-        response = json.loads(urllib.request.urlopen(url).read())
-        for element in response['itemListElement']:
-            google_search_link = f"https://www.google.com/search?kgmid={entity_id}"
-            st.markdown(f'[Open in Google Search]({google_search_link})', unsafe_allow_html=True)
+        google_search_link = f"https://www.google.com/search?kgmid={entity_id}"
+        st.markdown(f'[Open in Google Search]({google_search_link})', unsafe_allow_html=True)
     except Exception as e:
-        st.write(f"An error occurred while querying the Knowledge Graph: {e}")
+        st.write(f"An error occurred: {e}")
+
 
 # Function to count entities with 'mid' that contains '/g/' or '/m/' in their metadata
 def count_entities(entities):