Update spacy_app.py

spacy_app.py

@@ -9,23 +9,20 @@ spacy.cli.download("en_core_web_sm")
 nlp = spacy.load("en_core_web_sm")
 
 def create_knowledge_graph(text):
-    # Process the text using spaCy
     doc = nlp(text)
     entities = [(ent.text, ent.label_) for ent in doc.ents]
     
-    # Create a graph
     G = nx.Graph()
     for entity, label in entities:
         G.add_node(entity, label=label)
 
-    # Add edges (for simplicity, connect each entity to the next)
     for i in range(len(entities) - 1):
         G.add_edge(entities[i][0], entities[i + 1][0])
     
     return G, entities
 
 def plot_graph(G):
-    pos = nx.spring_layout(G)  # positions for all nodes
+    pos = nx.spring_layout(G)
     edges = G.edges()
     edge_x = []
     edge_y = []
@@ -35,10 +32,10 @@ def plot_graph(G):
         x1, y1 = pos[edge[1]]
         edge_x.append(x0)
         edge_x.append(x1)
-        edge_x.append(None)  # None to break line
+        edge_x.append(None)
         edge_y.append(y0)
         edge_y.append(y1)
-        edge_y.append(None)  # None to break line
+        edge_y.append(None)
 
     node_x = []
     node_y = []
@@ -47,17 +44,14 @@ def plot_graph(G):
         node_x.append(x)
         node_y.append(y)
 
-    # Create Plotly figure
     fig = go.Figure()
     
-    # Add edges to figure
     fig.add_trace(go.Scatter(
         x=edge_x, y=edge_y,
         line=dict(width=0.5, color='black'),
         hoverinfo='none',
         mode='lines'))
 
-    # Add nodes to figure
     fig.add_trace(go.Scatter(
         x=node_x, y=node_y,
         mode='markers+text',
@@ -82,16 +76,21 @@ def plot_graph(G):
 # Streamlit app
 # st.title("Knowledge Graph Generator")
 
-text_input = st.text_area("Enter your text here:")
+# File uploader
+uploaded_file = st.file_uploader("Upload a text file", type="txt")
+if uploaded_file is not None:
+    text_input = uploaded_file.read().decode("utf-8")
+else:
+    text_input = st.text_area("Or enter your text here:")
+
 if st.button("Generate Knowledge Graph"):
     if text_input:
         G, entities = create_knowledge_graph(text_input)
         
-        # Plot the graph using Plotly
         fig = plot_graph(G)
         st.plotly_chart(fig)
         
         st.write("Extracted Entities:")
         st.write(entities)
     else:
-        st.warning("Please enter some text.")
+        st.warning("Please enter some text or upload a file.")