Update app.py

app.py

@@ -18,11 +18,11 @@ def query_knowledge_graph(entity_id):
         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']}")
+            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}")
 
-
 # Function to count entities with 'mid' that contains '/g/' or '/m/' in their metadata
 def count_entities(entities):
     count = 0
@@ -81,8 +81,9 @@ def sample_analyze_entities(text_content):
                 query_knowledge_graph(entity_id)
 
         if entity.mentions:
-            st.write("Mentions:")
-            st.write(', '.join([mention.text.content for mention in entity.mentions]))
+            mention_count = len(entity.mentions)
+            plural = "s" if mention_count > 1 else ""
+            st.write(f"Mentions: {mention_count} mention{plural}")
 
         st.write("---")