Update app.py

app.py

@@ -24,7 +24,7 @@ def embed_data(data):
     return model.encode(data)
     
 
-def process_data(df, desc, message, embed=False):
+def process_data(df, desc, message, query, embed=False):
     data = [
         [
             f'Represent the document for retrieval of {x[desc]} information : ',
@@ -37,16 +37,17 @@ def process_data(df, desc, message, embed=False):
     if embed :
         corpus_embeddings = embed_data(data)
 
-    def query_change():
-        query  = [['Represent the question for retrieving supporting documents: ',question]]
-        query_embeddings = model.encode(query)
-
-        with st.spinner('Wait for it...'):
-            similarities = cosine_similarity(query_embeddings,corpus_embeddings)
-            retrieved_doc_id = np.argmax(similarities)
-        st.markdown(f"{data[retrieved_doc_id][-1]}",unsafe_allow_html=True)
+    
+    
+    query_embeddings = model.encode(query)
 
-    question = st.text_input("Question (Press Enter to query) :", on_change=query_change)
+    with st.spinner('Wait for it...'):
+        similarities = cosine_similarity(query_embeddings,corpus_embeddings)
+        retrieved_doc_id = np.argmax(similarities)
+    st.markdown(f"{data[retrieved_doc_id][-1]}",unsafe_allow_html=True)
+    
+    # question = st.text_input("Question (Press Enter to query) :")
+    # query  = [['Represent the question for retrieving supporting documents: ',question]]
     # btn_q = st.button("Submit", key="submit_query")
 
     # if btn_q :
@@ -67,6 +68,7 @@ if opt == "intent.csv":
     process_data(df, desc='description', message='message', embed=True)
 
 else :
+    embed = False
     f = st.file_uploader("Upload CSV File with at least 2 columns", ['xlsx', 'csv'])
     delim = st.text_input('CSV File Delimiter')
     btn = st.button("Submit", key="submit_first")
@@ -80,15 +82,20 @@ else :
             st.write("FAILED! At least 2 columns needed. Please check your dataset")
         else :
             st.session_state.query_bool = True
+            embed=True
 
     if st.session_state.query_bool:
         with st.form("my_form"):
             cols = list(st.session_state.df.columns)
             desc = st.radio("Description Column (e.g. description)", cols)
             message = st.radio("Response Template Column (e.g. message)", cols)
+            question = st.text_input("Question (Press Enter to query) :")
             submitted = st.form_submit_button("submit")
+            
             if submitted:
-                process_data(st.session_state.df, desc, message, embed=True)
+                query  = [['Represent the question for retrieving supporting documents: ',question]]
+                process_data(st.session_state.df, desc, message, query, embed=embed)
+                embed=False