Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from annotated_text import annotated_text
 from refined.inference.processor import Refined
 import requests
+import json
 
 # Sidebar
 st.sidebar.image("logo-wordlift.png")
@@ -63,6 +64,13 @@ if text_input:
         # Prepare a list to hold the final output
         final_text = []
 
+        # JSON-LD data
+        json_ld_data = {
+            "@context": "https://schema.org",
+            "@type": "WebPage",
+            "mentions": []
+        }
+
         # Replace each entity in the text with its annotated version
         for entity_string, entity_info in entities_map.items():
             entity_data = entities_data.get(entity_string, None)
@@ -87,6 +95,10 @@ if text_input:
 
             entity_annotation = (entity_string, entity_info["id"], color)
             text_input = text_input.replace(entity_string, f'{{{str(entity_annotation)}}}', 1)
+            
+            # Add the entity to JSON-LD data
+            entity_json_ld = combined_entity_info_dictionary[entity_string][1]
+            json_ld_data["mentions"].append(entity_json_ld)
 
         # Split the modified text_input into a list
         text_list = text_input.split("{")
@@ -103,5 +115,8 @@ if text_input:
         # Pass the final_text to the annotated_text function
         annotated_text(*final_text)
 
+        with st.expander("Copy the JSON-LD")
+            st.write(json.dumps(json_ld_data, indent=2))  # Output JSON-LD
+        
         with st.expander("See annotations"):
             st.write(combined_entity_info_dictionary)