Update app.py

app.py

@@ -8,6 +8,7 @@ KG_name = st.text_input("Give Kuzu-KG name")
 
   
 if (token != '' and KG_name != ""):
+    index = None
     if(os.path.exists(KG_name)):
         index = load_index(token,KG_name)
     else:
@@ -19,28 +20,28 @@ if (token != '' and KG_name != ""):
                 links = text.split(",")   
                 index = get_index(links,token,KG_name)
         
+    if (index != None):
+        get_network_graph(index)
+        emb = get_embeddings(index)
+        fig = get_visualize_embeddings(emb)
             
-    get_network_graph(index)
-    emb = get_embeddings(index)
-    fig = get_visualize_embeddings(emb)
-        
-    # Plotly Chart
-    st.plotly_chart(fig, use_container_width=True)
-        
-    # Render the h1 block, contained in a frame of size 200x200.
-    HtmlFile = open("kuzugraph_draw3.html", 'r', encoding='utf-8')
-    # Read the HTML file
-    with open("kuzugraph_draw3.html", 'r', encoding='utf-8') as HtmlFile:
-        source_code = HtmlFile.read()
-    # st.markdown(f'<div style="width: 800px; height: 600px">{source_code}</div>', unsafe_allow_html=True)
-    components.html(source_code,width=800, height=600, scrolling=False)
-
-    with st.form("my_form"):
-        user_query = st.text_input("Ask the KG ','")
-          
-        submitted = st.form_submit_button("Submit")
-        if submitted:
-            res = query_model(index,user_query)
-            st.write(res)
+        # Plotly Chart
+        st.plotly_chart(fig, use_container_width=True)
+            
+        # Render the h1 block, contained in a frame of size 200x200.
+        HtmlFile = open("kuzugraph_draw3.html", 'r', encoding='utf-8')
+        # Read the HTML file
+        with open("kuzugraph_draw3.html", 'r', encoding='utf-8') as HtmlFile:
+            source_code = HtmlFile.read()
+        # st.markdown(f'<div style="width: 800px; height: 600px">{source_code}</div>', unsafe_allow_html=True)
+        components.html(source_code,width=800, height=600, scrolling=False)
+    
+        with st.form("my_form"):
+            user_query = st.text_input("Ask the KG ','")
+              
+            submitted = st.form_submit_button("Submit")
+            if submitted:
+                res = query_model(index,user_query)
+                st.write(res)