Update app.py

app.py

@@ -29,9 +29,14 @@ def get_backlinks(api_login, api_key, target_url, filters):
     if response.status_code == 200 and 'results' in response.json():
         # Extract the results
         results = response.json()['results']
-        # Convert results to DataFrame
-        df = pd.DataFrame(results)
-        return df
+        # Check if results are not empty
+        if results:
+            # Convert results to DataFrame
+            df = pd.DataFrame(results)
+            return df
+        else:
+            st.error("Received empty data from API.")
+            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}")
@@ -72,14 +77,9 @@ generate_button = st.sidebar.button("Generate All")
 reset_button = st.sidebar.button("Reset")
 
 # Generate CSV and download button
-if generate_button and target_url:  # Check after target_url is defined
+if generate_button and target_url:
     df = get_backlinks(api_login, api_key, target_url, filters)
     if df is not None:
-        csv = df.to_csv(index=False)
-        b64 = base64.b64encode(csv.encode()).decode()
-        href = f'<a href="data:file/csv;base64,{b64}" download="backlinks.csv">Download CSV file</a>'
-        st.markdown(href, unsafe_allow_html=True)
-
         # Display the DataFrame in the second column
         with col2:
             st.header("Output")