Update src/streamlit_app.py

src/streamlit_app.py

@@ -282,10 +282,6 @@ if uploaded is not None:
     else:
         df_input = pd.read_excel(uploaded)
 
-    st.subheader("Uploaded Data Preview")
-    st.dataframe(df_input.head(20), use_container_width=True)
-    st.info(f"{len(df_input)} rows loaded")
-
     # --- Column mapping ---
     st.subheader("Map Columns")
     all_cols = ["(auto-detect)"] + list(df_input.columns)
@@ -311,20 +307,14 @@ if uploaded is not None:
             rename_map[zip_col] = "Zip"
         df_input = df_input.rename(columns=rename_map)
 
-    # --- Row range ---
-    r1, r2 = st.columns(2)
-    with r1:
-        row_start = st.number_input("Start row", min_value=0, value=0)
-    with r2:
-        row_end = st.number_input("End row (0 = all)", min_value=0, value=0)
+    st.subheader("Uploaded Data Preview")
+    st.dataframe(df_input.head(20), use_container_width=True)
+    st.info(f"{len(df_input)} rows loaded")
 
     # --- Run button ---
     if st.button("Run Scraper", type="primary", use_container_width=True):
 
-        if row_end > 0:
-            df_work = df_input.iloc[row_start:row_end].copy()
-        else:
-            df_work = df_input.iloc[row_start:].copy()
+        df_work = df_input.copy()
 
         df_work['query'] = df_work.iloc[:, :4].astype(str).apply(' '.join, axis=1)
         df_work['query'] = df_work['query'] + ' square footage'
@@ -455,8 +445,15 @@ if uploaded is not None:
             unsafe_allow_html=True,
         )
 
-        # --- Download button ---
-        csv_bytes = df_clean.to_csv(index=False, encoding='utf-8-sig').encode('utf-8-sig')
+        # --- Download button (include mapped input columns) ---
+        input_cols = list(df_input.columns[:4])
+        df_download = df_clean.reset_index(drop=True)
+        df_work_reset = df_work[input_cols].reset_index(drop=True)
+        for col in input_cols:
+            if col not in df_download.columns and col in df_work_reset.columns:
+                df_download[col] = df_work_reset[col]
+
+        csv_bytes = df_download.to_csv(index=False, encoding='utf-8-sig').encode('utf-8-sig')
         st.download_button(
             label="Download Results CSV",
             data=csv_bytes,