Update app.py

app.py

@@ -67,10 +67,10 @@ if upload_file is not None:
             st.error("Error: The file contains missing values.")
             st.stop()
         else:
-            st.session_state.df_original = df_original  # Store the original dataframe
+            st.session_state.df_original = df_original  
 
             all_columns = df_original.columns.tolist()
-            st.subheader("Select columns for the Tree Map", divider="blue")
+            st.subheader("Select columns from your data to visualize it in a tree map", divider="blue")
             parent_column = st.selectbox("Select the parent column:", all_columns)
             value_column = st.selectbox("Select the value column:", all_columns)
 
@@ -78,15 +78,15 @@ if upload_file is not None:
                 if parent_column == value_column:
                     st.warning("Warning: You have selected the same column for both the parent and value column. Please select two different columns from your data.")
                 else:
-                    df_treemap = df_original.copy()  # Create a copy for treemap
+                    df_treemap = df_original.copy()  
                     path_columns = [px.Constant("all"), parent_column, value_column]
                     fig = px.treemap(df_treemap,
                                      path=path_columns)
                     fig.update_layout(margin=dict(t=50, l=25, r=25, b=25))
-                    st.subheader("Tree map", divider="red")
+                    st.subheader("Tree map", divider="blue")
                     st.plotly_chart(fig)
 
-            st.subheader("Uploaded Data", divider="green")
+            st.subheader("Uploaded File", divider="blue")
             st.dataframe(df_original, key="uploaded_dataframe")
             st.write(f"_Number of rows_: {df_original.shape[0]}")
             st.write(f"_Number of columns_: {df_original.shape[1]}")
@@ -122,7 +122,7 @@ if st.button("Retrieve your answer"):
     
     with st.spinner("Wait for it...", show_time=True):
         time.sleep(5)          
-        if df1 is not None:
+        if df is not None:
             tqa = pipeline(task="table-question-answering", model="microsoft/tapex-large-finetuned-wtq")
             st.write(tqa(table=df, query=question)['answer'])