Update app.py

app.py

@@ -27,44 +27,35 @@ def get_backlinks(api_login, api_key, target_url, filters):
 
     # Log the full response for debugging
     st.text("API Response:")
-    # st.json(response.json())  # This will display the full JSON response in the Streamlit app
-
-    # For debugging: Display the status code and any text
     st.text(f"Response Status Code: {response.status_code}")
     st.text(f"Response Headers: {response.headers}")
     try:
-        st.text(f"Response Body: {response.json()}")
-    except ValueError:
-        st.text("Response Body: <Not a JSON response>")
-
-    # Check if the response contains 'results' key
-    if response.status_code == 200:
-        response_data = response.json()
-        
-        # Check if 'results' key is in the response
-        if 'results' in response_data:
-            results = response_data['results']
-            if results:
-                # Adjust the following line based on the actual JSON structure
-                df = pd.json_normalize(results)
-                return df
-            else:
-                st.error("Received empty data from API.")
-                return None
+        response_json = response.json()
+        st.text(f"Response Body: {response_json}")
+    except ValueError as e:
+        st.text(f"Response Body: <Not a JSON response>\nError: {e}")
+
+    # Check if the response contains 'results' key and handle the JSON structure appropriately
+    if response.status_code == 200 and 'results' in response_json:
+        results = response_json['results']
+        if results:
+            # Check if the results actually contain the backlink data you expect
+            st.text(f"Results: {results}")  # Debugging line to show the results structure
+
+            # If the structure is as expected, convert to DataFrame
+            df = pd.json_normalize(results)
+            return df
         else:
-            # Handle other internal status codes here
-            internal_status_code = response_data.get('status_code', None)
-            internal_status_message = response_data.get('status_message', 'No specific message provided')
-            st.error(f"Internal Status Code: {internal_status_code}, Message: {internal_status_message}")
+            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}")
+        # Handle API errors
+        st.error(f"Error: Code: {response.status_code} Message: {response_json.get('status_message', 'No specific error message provided')}")
         return None
 
 def convert_df_to_csv(df):
     # Convert DataFrame to CSV
-    return df.to_csv().encode('utf-8')
+    return df.to_csv(index=False).encode('utf-8')
 
 # Streamlit layout
 st.sidebar.title("DataForSEO API Parameters")
@@ -107,7 +98,7 @@ col1, col2 = st.columns(2)
 
 with col1:
     st.header("Input")
-    target_url = st.text_input("Enter the target URL")  # Define target_url here
+    target_url = st.text_input("Enter the target URL")
 
 generate_button = st.sidebar.button("Generate All")
 reset_button = st.sidebar.button("Reset")
@@ -116,16 +107,26 @@ reset_button = st.sidebar.button("Reset")
 if generate_button and target_url:
     df = get_backlinks(api_login, api_key, target_url, filters)
     if df is not None:
-        # Convert DataFrame to CSV
         csv = convert_df_to_csv(df)
-        # Create download link
         st.download_button(
             label="Download data as CSV",
             data=csv,
             file_name='backlinks.csv',
             mime='text/csv',
         )
+    else:
+        st.error("Failed to generate CSV: No data returned from the API or data processing error.")
 
 # Reset functionality
 if reset_button:
     st.experimental_rerun()
+
+# Ensure that the download button is only visible if the DataFrame is not None
+if df is not None:
+    csv = convert_df_to_csv(df)
+    st.download_button(
+        label="Download data as CSV",
+        data=csv,
+        file_name='backlinks.csv',
+        mime='text/csv',
+    )