Update app.py

app.py

@@ -55,8 +55,12 @@ def sample_analyze_entities(text_content):
     # 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 - We found {entity_count} Google Entities")
-    st.write("---")
+    if entity_count == 0:
+        st.write(f"We found {len(response.entities)} entities - We found no Google Entities")
+    elif entity_count == 1:
+        st.write(f"We found {len(response.entities)} entities - We found 1 Google Entity")
+    else:
+        st.write(f"We found {len(response.entities)} entities - We found {entity_count} Google Entities")
 
     for i, entity in enumerate(response.entities):
         st.write(f"Entity {i+1} of {len(response.entities)}")