Spaces:

fabiochiu
/

text-to-kb

Build error

App Files Files Community

fabiochiusano commited on May 30, 2022

Commit

cc1a8be

1 Parent(s): 5ca37e8

max num chars and manage url errors

Browse files

Files changed (2) hide show

app.py +13 -5
networks/network.html +116 -0

app.py CHANGED Viewed

@@ -82,7 +82,7 @@ if option == "Text":
         if 'text' not in st.session_state:
             st.session_state.text = ""
         text = st.session_state.text
-    text = st.text_area('Text:', value=text, height=300, disabled=disabled)
 elif option == "Article at URL":
     url_options = [
         "Crypto",
@@ -114,8 +114,6 @@ else:
         st.session_state.news_option = news_options[0]
     news_option = st.selectbox('Use articles about:', news_options, index=news_options.index(st.session_state.news_option))
-placeholder = st.empty()
 def generate_kb():
     st.session_state.option = option
     st.session_state.text_option = text_option
@@ -142,7 +140,12 @@ def generate_kb():
         elif url_option == "Rome":
             kb = utils.load_kb("networks/network_2_rome.p")
         else:
-            kb = utils.from_url_to_kb(url, model, tokenizer)
     else:
         if news_option == "Google":
             kb = utils.load_kb("networks/network_3_google.p")
@@ -157,6 +160,7 @@ def generate_kb():
     utils.save_network_html(kb, filename="networks/network.html")
     st.session_state.kb_chart = "networks/network.html"
     st.session_state.kb_text = kb.get_textual_representation()
 st.session_state.option = option
@@ -178,9 +182,13 @@ if 'kb_chart' not in st.session_state:
     st.session_state.kb_chart = None
 if 'kb_text' not in st.session_state:
     st.session_state.kb_text = None
 # show graph
-if st.session_state.kb_chart:
     with st.container():
         st.subheader("Generated KB")
         st.markdown("*You can interact with the graph and zoom.*")

         if 'text' not in st.session_state:
             st.session_state.text = ""
         text = st.session_state.text
+    text = st.text_area('Text:', value=text, height=300, disabled=disabled, max_chars=1500)
 elif option == "Article at URL":
     url_options = [
         "Crypto",
         st.session_state.news_option = news_options[0]
     news_option = st.selectbox('Use articles about:', news_options, index=news_options.index(st.session_state.news_option))
 def generate_kb():
     st.session_state.option = option
     st.session_state.text_option = text_option
         elif url_option == "Rome":
             kb = utils.load_kb("networks/network_2_rome.p")
         else:
+            try:
+                kb = utils.from_url_to_kb(url, model, tokenizer)
+            except Exception as e:
+                print("Couldn't extract article from URL")
+                st.session_state.error_url = "Couldn't extract article from URL"
+                return
     else:
         if news_option == "Google":
             kb = utils.load_kb("networks/network_3_google.p")
     utils.save_network_html(kb, filename="networks/network.html")
     st.session_state.kb_chart = "networks/network.html"
     st.session_state.kb_text = kb.get_textual_representation()
+    st.session_state.error_url = None
 st.session_state.option = option
     st.session_state.kb_chart = None
 if 'kb_text' not in st.session_state:
     st.session_state.kb_text = None
+if 'error_url' not in st.session_state:
+    st.session_state.error_url = None
 # show graph
+if st.session_state.error_url:
+    st.markdown(st.session_state.error_url)
+elif st.session_state.kb_chart:
     with st.container():
         st.subheader("Generated KB")
         st.markdown("*You can interact with the graph and zoom.*")

networks/network.html ADDED Viewed

	@@ -0,0 +1,116 @@

+<html>
+<head>
+<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/vis-network@latest/styles/vis-network.css" type="text/css" />
+<script type="text/javascript" src="https://cdn.jsdelivr.net/npm/vis-network@latest/dist/vis-network.min.js"> </script>
+<center>
+<h1></h1>
+</center>
+<!-- <link rel="stylesheet" href="../node_modules/vis/dist/vis.min.css" type="text/css" />
+<script type="text/javascript" src="../node_modules/vis/dist/vis.js"> </script>-->
+<style type="text/css">
+        #mynetwork {
+            width: 700px;
+            height: 700px;
+            background-color: #ffffff;
+            border: 1px solid lightgray;
+            position: relative;
+            float: left;
+        }
+</style>
+</head>
+<body>
+<div id = "mynetwork"></div>
+<script type="text/javascript">
+    // initialize global variables.
+    var edges;
+    var nodes;
+    var network;
+    var container;
+    var options, data;
+    // This method is responsible for drawing the graph, returns the drawn network
+    function drawGraph() {
+        var container = document.getElementById('mynetwork');
+        // parsing and collecting nodes and edges from the python
+        nodes = new vis.DataSet([{"color": "#00FF00", "id": "Napoleon", "label": "Napoleon", "shape": "circle"}, {"color": "#00FF00", "id": "French Revolution", "label": "French Revolution", "shape": "circle"}, {"color": "#00FF00", "id": "France", "label": "France", "shape": "circle"}]);
+        edges = new vis.DataSet([{"arrows": "to", "from": "Napoleon", "label": "participant in", "title": "participant in", "to": "French Revolution"}, {"arrows": "to", "from": "French Revolution", "label": "participant", "title": "participant", "to": "Napoleon"}, {"arrows": "to", "from": "French Revolution", "label": "country", "title": "country", "to": "France"}]);
+        // adding nodes and edges to the graph
+        data = {nodes: nodes, edges: edges};
+        var options = {
+    "configure": {
+        "enabled": false
+    },
+    "edges": {
+        "color": {
+            "inherit": true
+        },
+        "smooth": {
+            "enabled": true,
+            "type": "dynamic"
+        }
+    },
+    "interaction": {
+        "dragNodes": true,
+        "hideEdgesOnDrag": false,
+        "hideNodesOnDrag": false
+    },
+    "physics": {
+        "enabled": true,
+        "repulsion": {
+            "centralGravity": 0.2,
+            "damping": 0.09,
+            "nodeDistance": 200,
+            "springConstant": 0.05,
+            "springLength": 200
+        },
+        "solver": "repulsion",
+        "stabilization": {
+            "enabled": true,
+            "fit": true,
+            "iterations": 1000,
+            "onlyDynamicEdges": false,
+            "updateInterval": 50
+        }
+    }
+};
+        network = new vis.Network(container, data, options);
+        return network;
+    }
+    drawGraph();
+</script>
+</body>
+</html>