Update app.py

app.py

@@ -50,6 +50,8 @@ elif model_choice == "GPT-4o":
 # Initialize session state for data persistence
 if "df" not in st.session_state:
     st.session_state.df = None
+if "show_preview" not in st.session_state:
+    st.session_state.show_preview = False  
 
 # Dataset Input
 input_option = st.radio("Select Dataset Input:", ["Use Hugging Face Dataset", "Upload CSV File"])
@@ -61,6 +63,7 @@ if input_option == "Use Hugging Face Dataset":
             with st.spinner("Loading dataset..."):
                 dataset = load_dataset(dataset_name, split="train")
                 st.session_state.df = pd.DataFrame(dataset)
+                st.session_state.show_preview = True  # Show preview after loading
                 st.success(f"Dataset '{dataset_name}' loaded successfully!")
         except Exception as e:
             st.error(f"Error: {e}")
@@ -70,16 +73,16 @@ elif input_option == "Upload CSV File":
     if uploaded_file:
         try:
             st.session_state.df = pd.read_csv(uploaded_file)
+            st.session_state.show_preview = True  # Show preview after loading
             st.success("File uploaded successfully!")
         except Exception as e:
             st.error(f"Error loading file: {e}")
 
-# Display Dataset Preview 
-if st.session_state.df is not None:
+# Show Dataset Preview Only After Loading
+if st.session_state.df is not None and st.session_state.show_preview:
     st.subheader("📂 Dataset Preview")
     st.dataframe(st.session_state.df.head())
 
-
 # SQL-RAG Analysis
 if st.session_state.df is not None:
     temp_dir = tempfile.TemporaryDirectory()