Update app.py

app.py

@@ -3,12 +3,12 @@ import streamlit as st
 from google.oauth2 import service_account
 from google.cloud import language_v1
 
-# Function to count entities with 'mid' and '/g/' in their metadata
+# Function to count entities with 'mid' that contains '/g/' or '/m/' in their metadata
 def count_entities(entities):
     count = 0
     for entity in entities:
         metadata = entity.metadata
-        if 'mid' in metadata and '/g/' in metadata['mid']:
+        if 'mid' in metadata and ('/g/' in metadata['mid'] or '/m/' in metadata['mid']):
             count += 1
     return count
 
@@ -40,10 +40,10 @@ def sample_analyze_entities(text_content):
 
     response = client.analyze_entities(request={"document": document, "encoding_type": encoding_type})
     
-    # Count the entities with 'mid' and '/g/' in their metadata
+    # Count the entities with 'mid' and either '/g/' or '/m/' in their metadata
     entity_count = count_entities(response.entities)
 
-    st.write(f"We found {len(response.entities)} entities - {entity_count} meet your criteria")
+    st.write(f"We found {len(response.entities)} entities - We found {entity_count} Google Entities")
     st.write("---")
 
     for i, entity in enumerate(response.entities):
@@ -55,10 +55,6 @@ def sample_analyze_entities(text_content):
         if entity.metadata:
             st.write("Metadata:")
             st.write(entity.metadata)
-
-        if entity.mentions:
-            st.write("Mentions:")
-            st.write(', '.join([mention.text.content for mention in entity.mentions]))
         
         st.write("---")