Update app.py

app.py

@@ -172,34 +172,12 @@ row_count = st.session_state.get("row_count", 1)
 # Define placeholders for each row in the second column
 download_placeholders = [st.empty() for _ in range(row_count)]
 
-for i in range(row_count):
-    col1, _ = st.columns(2)
+for i, (col1_placeholder, col2_placeholder) in enumerate(placeholders):
     target_url_key = f"target_url_{i}"
 
-    with col1:
+    with col1_placeholder.container():
         st.text_input(f"Enter the target URL {i + 1}", key=target_url_key)
 
-    with col2:
-        # Use placeholders for download buttons
-        download_placeholder = st.empty()
-
-        # Conditionally display download buttons or error messages
-        if 'generate_pressed' in st.session_state:
-            df_key = f"df_{i}"
-            df = st.session_state.get(df_key)
-
-            if df is not None:
-                csv = convert_df_to_csv(df)
-                download_placeholder.download_button(
-                    label=f"Download data as CSV for URL {i + 1}",
-                    data=csv,
-                    file_name=f'backlinks_{i + 1}.csv',
-                    mime='text/csv',
-                )
-            else:
-                # This will only show an error if the generation process has been initiated
-                download_placeholder.error(f"Failed to generate CSV for URL {i + 1}: No data returned from the API or data processing error.")
-
 if 'generate_pressed' in st.session_state and st.session_state['generate_pressed']:
     jobs = Queue()
     results = Queue()
@@ -223,21 +201,22 @@ if 'generate_pressed' in st.session_state and st.session_state['generate_pressed
         url_id, df = results.get()
         st.session_state[f"df_{url_id}"] = df
 
-    # Update placeholders with download buttons or error messages
-    for i, placeholder in enumerate(download_placeholders):
+    # Update placeholders in the second column
+    for i, (_, col2_placeholder) in enumerate(placeholders):
         df_key = f"df_{i}"
         df = st.session_state.get(df_key)
 
-        if df is not None:
-            csv = convert_df_to_csv(df)
-            placeholder.download_button(
-                label=f"Download data as CSV for URL {i + 1}",
-                data=csv,
-                file_name=f'backlinks_{i + 1}.csv',
-                mime='text/csv',
-            )
-        else:
-            placeholder.error(f"Failed to generate CSV for URL {i + 1}: No data returned from the API or data processing error.")
+        with col2_placeholder.container():
+            if df is not None:
+                csv = convert_df_to_csv(df)
+                st.download_button(
+                    label=f"Download data as CSV for URL {i + 1}",
+                    data=csv,
+                    file_name=f'backlinks_{i + 1}.csv',
+                    mime='text/csv',
+                )
+            else:
+                st.error(f"Failed to generate CSV for URL {i + 1}: No data returned from the API or data processing error.")
 
 # Display and download logic for each row
 for i in range(row_count):