Update app.py

app.py

@@ -43,12 +43,28 @@ def sample_analyze_entities(text_content, your_query=""):
     # Create an empty list to hold the results
     entities_list = []
 
-    for entity in response.entities:
-        entity_type_name = language_v1.Entity.Type(entity.type_).name
-        if not selected_types or entity_type_name in selected_types:
-            # Your existing code for creating and populating the entity_details dictionary...
+for entity in response.entities:
+    entity_type_name = language_v1.Entity.Type(entity.type_).name
+    if not selected_types or entity_type_name in selected_types:
+        entity_details = {
+            "Name": entity.name,
+            "Type": entity_type_name,
+            "Salience Score": entity.salience,
+            "Metadata": [],
+            "Mentions": []
+        }
+
+        for metadata_name, metadata_value in entity.metadata.items():
+            entity_details["Metadata"].append({metadata_name: metadata_value})
+
+        for mention in entity.mentions:
+            entity_details["Mentions"].append({
+                "Text": mention.text.content,
+                "Type": language_v1.EntityMention.Type(mention.type_).name
+            })
+
+        entities_list.append(entity_details)  # Append the dictionary to the list
 
-            entities_list.append(entity_details)  # Append the dictionary to the list
 
     # Streamlit UI
     if your_query:
@@ -73,7 +89,8 @@ def sample_analyze_entities(text_content, your_query=""):
                 else:
                     st.write(value)
         
-        st.write("----")
+        st.write("----")  # Line separator
+        st.write(" ")  # Extra line
 
     st.write(f"### Language of the text: {response.language}")