Update app.py

app.py

@@ -144,8 +144,8 @@ elif selected == "Upload Data":
 elif selected == "Define Query":
     st.header("Define Your Custom Query")
 
-    entity = st.text_input("Enter the entity name")
-    query = st.text_input("Enter the query (e.g., 'contact information for {entity}')")
+    #entity = st.text_input("Enter the entity name")
+    #query = st.text_input("Enter the query (e.g., 'contact information for {entity}')")
 
     if st.session_state["data"] is not None:
         column_selection = st.selectbox("Select the primary column for entities", options=st.session_state["data"].columns)
@@ -161,10 +161,10 @@ elif selected == "Define Query":
             st.code(example_query)
     else:
         st.warning("Please upload data first.")
-    if st.button("Extract Information"):
-        if entity and query:
-            response_text = get_llm_response(entity, query)
-            st.write(response_text)
+   # if st.button("Extract Information"):
+    #    if entity and query:
+     #       response_text = get_llm_response(entity, query)
+      #      st.write(response_text)
 
 # Extract Information Section with Progress Bar
 elif selected == "Extract Information":