Update app.py

app.py

@@ -162,32 +162,45 @@ if reset:
     st.session_state.clear()
 
 # Generate and reset button logic
-generate_button = st.sidebar.button("Generate All")
+generate_pressed = st.sidebar.button("Generate All")
+if generate_pressed:
+    st.session_state['generate_pressed'] = True
 
 # Main app layout
 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, col2 = st.columns(2)
+    col1, _ = st.columns(2)
     target_url_key = f"target_url_{i}"
-    
+
     with col1:
         st.text_input(f"Enter the target URL {i + 1}", key=target_url_key)
-    
-    df_key = f"df_{i}"
-    df = st.session_state.get(df_key)
+
     with col2:
-        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',
-            )
-        elif df is None and generate_button:
-            st.error(f"Failed to generate CSV for URL {i + 1}: No data returned from the API or data processing error.")
+        # 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_button:
+if 'generate_pressed' in st.session_state and st.session_state['generate_pressed']:
     jobs = Queue()
     results = Queue()
     workers = [BacklinkWorker(jobs, results, api_login, api_key) for _ in range(num_concurrent_calls)]
@@ -210,6 +223,22 @@ if generate_button:
         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):
+        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.")
+
 # Display and download logic for each row
 for i in range(row_count):
     df_key = f"df_{i}"