Update app.py

app.py

@@ -4,44 +4,60 @@ from logic import *
 
 token = st.text_input("Open-AI-api-key")
 
-KG_name = st.text_input("Give Kuzu-KG name")
 
-  
-if (token != '' and KG_name != ""):
-    index = None
+input_text = st.radio(
+    "Input Options for text",
+    ["PDF", "Links","No_Input(Already Created)"])
+index = None
+if(input_text == "PDF"):
+    st.subheader("PDF")
+    with st.form("PDF_form"):
+        uploaded_file = st.file_uploader('Choose your .pdf file', type="pdf")
+        KG_name = st.text_input("Give Kuzu-KG name")
+        sub = st.form_submit_button("Submit")
+        if sub: 
+            save_uploadedfile(uploaded_file)
+            index = get_index_pdf(token,KG_name)
+    
+elif(input_text == "Links"):
+    st.subheader("LINKS")
+    with st.form("links_form"):
+        text = st.text_input("Input Links Seperated by ','")
+        KG_name = st.text_input("Give Kuzu-KG name")    
+        submitted = st.form_submit_button("Submit")
+        if submitted:
+            links = text.split(",")   
+            index = get_index(links,token,KG_name)
+
+elif(input_text == "No_Input(Already Created)"):
+    st.subheader("NO INPUT")
+    KG_name = st.text_input("Give Kuzu-KG name")
     if(os.path.exists(KG_name)):
         index = load_index(token,KG_name)
     else:
-        with st.form("links_form"):
-            text = st.text_input("Input Links Seperated by ','")
-              
-            submitted = st.form_submit_button("Submit")
-            if submitted:
-                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)
+        st.write("NO FOLDER BY NAME")
+
+     
+if (index != None):
+    get_network_graph(index)
+    emb = get_embeddings(index)
+    fig = get_visualize_embeddings(emb)
             
-        # Plotly Chart
-        st.plotly_chart(fig, use_container_width=True)
+    # 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)
+    # 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 ','")
+    with st.form("my_form"):
+        user_query = st.text_input("Ask the KG ','")
               
-            new_submitted = st.form_submit_button("Submit")
-            if new_submitted:
-                res = query_model(index,user_query)
-                st.write(res)
-
-
+        new_submitted = st.form_submit_button("Submit")
+        if new_submitted:
+            res = query_model(index,user_query)
+            st.write(res)