Update app.py

app.py

@@ -7,18 +7,22 @@ import urllib.request
 
 # Function to query Google's Knowledge Graph API
 def query_knowledge_graph(entity_id):
-    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']:
-        st.write(f"Knowledge Graph name: {element['result']['name']}")
-        st.write(f"Knowledge Graph score: {element['resultScore']}")
+    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']:
+            st.write(f"Knowledge Graph name: {element['result']['name']}")
+            st.write(f"Knowledge Graph score: {element['resultScore']}")
+    except Exception as e:
+        st.write(f"An error occurred while querying the Knowledge Graph: {e}")
+
 
 # Function to count entities with 'mid' that contains '/g/' or '/m/' in their metadata
 def count_entities(entities):