Update app.py

app.py

@@ -8,20 +8,13 @@ from io import StringIO
 import csv
 
 def get_backlinks(api_login, api_key, target_url, filters):
-    encoded_credentials = base64.b64encode(f"{api_login}:{api_key}".encode()).decode()
-    headers = {
-        'Authorization': f'Basic {encoded_credentials}'
-    }
-    post_data = {
-        0: {
-            "target": target_url,
-            "limit": 1000,
-            "mode": "as_is",
-            "filters": filters
-        }
-    }
-    response = requests.post("https://api.dataforseo.com/v3/backlinks/backlinks/live", json=post_data, headers=headers)
-    if response.status_code == 200:
+    try:
+        encoded_credentials = base64.b64encode(f"{api_login}:{api_key}".encode()).decode()
+        headers = {'Authorization': f'Basic {encoded_credentials}'}
+        post_data = {0: {"target": target_url, "limit": 1000, "mode": "as_is", "filters": filters}}
+        response = requests.post("https://api.dataforseo.com/v3/backlinks/backlinks/live", json=post_data, headers=headers)
+        response.raise_for_status()  # Raise an exception for HTTP errors
+
         response_data = response.json()
         if 'tasks' in response_data:
             task_result = response_data['tasks'][0]['result']
@@ -35,9 +28,8 @@ def get_backlinks(api_login, api_key, target_url, filters):
         else:
             st.error(f"No 'tasks' key in response JSON. Full response: {response_data}")
             return None
-    else:
-        error_message = response.json().get('status_message', 'No specific error message provided')
-        st.error(f"Error: Code: {response.status_code} Message: {error_message}")
+    except requests.RequestException as e:
+        st.error(f"Request error: {e}")
         return None
 
 def convert_df_to_csv(df):
@@ -139,24 +131,8 @@ if reset:
     st.session_state.clear()
 
 row_count = st.session_state.get("row_count", 1)
-for i in range(row_count):
-    cols = st.columns(2)
-    target_url_key = f"target_url_{i}"
-    target_url = cols[0].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)
-    if df is not None:
-        csv = convert_df_to_csv(df)
-        cols[1].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:
-        cols[1].error(f"Failed to generate CSV for URL {i + 1}: No data returned from the API or data processing error.")
 
-generate_button = st.sidebar.button("Generate All")
+generate_button = st.sidebar.button("Generate All")  # Define generate_button before its first use
 
 if generate_button:
     jobs = Queue()
@@ -180,3 +156,20 @@ if generate_button:
     while not results.empty():
         url_id, df = results.get()
         st.session_state[f"df_{url_id}"] = df
+
+for i in range(row_count):
+    cols = st.columns(2)
+    target_url_key = f"target_url_{i}"
+    target_url = cols[0].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)
+    if df is not None:
+        csv = convert_df_to_csv(df)
+        cols[1].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:
+        cols[1].error(f"Failed to generate CSV for URL {i + 1}: No data returned from the API or data processing error.")