Added query

app.py

@@ -16,10 +16,13 @@ embed_model_name = st.text_input(
 llm_model_name = st.text_input(
     'Embed Model name', "mistralai/Mistral-7B-Instruct-v0.2")
 
+query = st.text_input(
+    'Query', "What is the price of the product?")
+
 html_file = st.file_uploader("Upload a html file", type=["html"])
 
 if st.button('Start Pipeline'):
-    if html_file is not None and embed_model_name is not None and llm_model_name is not None:
+    if html_file is not None and embed_model_name is not None and llm_model_name is not None and query is not None:
         st.write('Running Pipeline')
         llm = HuggingFaceInferenceAPI(
             model_name=llm_model_name, token=inference_api_key)
@@ -54,7 +57,7 @@ if st.button('Start Pipeline'):
         retriever = index.as_retriever()
 
         ranked_nodes = retriever.retrieve(
-            "Get me all the information about the product")
+            query)
 
         with st.expander("Ranked Nodes"):
             for node in ranked_nodes:
@@ -63,8 +66,7 @@ if st.button('Start Pipeline'):
         query_engine = index.as_query_engine(
             filters=filters, service_context=service_context)
 
-        response = query_engine.query(
-            "Get me all the information about the product")
+        response = query_engine.query(query)
 
         st.write(response)