Update app.py

app.py

@@ -3,7 +3,6 @@ import numpy as np
 from sklearn.decomposition import PCA
 from sklearn.cluster import KMeans
 import umap
-import hdbscan
 import plotly.graph_objects as go
 from sentence_transformers import SentenceTransformer
 import torch
@@ -1340,7 +1339,7 @@ def main():
                             stats.update(label=f"Show clusters {column} complete", expanded=False)
 
     if st.session_state['data_processed']:
-        parsed2 = st.session_state.get('new_data', pd.DataFrame())
+        parsed2 = st.session_state.get('dataset', pd.DataFrame())
         parsed2 = filter_dataframe(parsed2)
         col1, col2 = st.columns(2)
         st.dataframe(parsed2)
@@ -1350,7 +1349,7 @@ def main():
             GEMINI_KEY = st.text_input('Gemini API Key', GEMINI_KEY, type='password', help="Enter your Gemini API key")
         if col_parsed and GEMINI_KEY:
             selected_column_data2 = parsed2[col_parsed2].tolist()
-            question2 = st.text_input("Ask a question or leave empty for summarization")
+            question2 = st.text_input("Ask a question / leave empty for summarization")
             if st.button("Generate Query") and selected_column_data2:
                 st.write(gemini_query(question2, selected_column_data2, GEMINI_KEY))