Update app.py

app.py

@@ -3,8 +3,6 @@ import pandas as pd
 import requests
 import base64
 
-# Function to call the DataForSEO API with Basic Authentication
-
 def get_backlinks(api_login, api_key, target_url, filters):
     # Encoding credentials
     encoded_credentials = base64.b64encode(f"{api_login}:{api_key}".encode()).decode()
@@ -29,7 +27,11 @@ def get_backlinks(api_login, api_key, target_url, filters):
 
     # Check if the response contains 'results' key
     if response.status_code == 200 and 'results' in response.json():
-        return response.json()['results']
+        # Extract the results
+        results = response.json()['results']
+        # Convert results to DataFrame
+        df = pd.DataFrame(results)
+        return df
     else:
         error_message = response.json().get('status_message', 'No specific error message provided')
         st.error(f"Error: Code: {response.status_code} Message: {error_message}")
@@ -71,20 +73,16 @@ reset_button = st.sidebar.button("Reset")
 
 # Generate CSV and download button
 if generate_button and target_url:  # Check after target_url is defined
-    data = get_backlinks(api_login, api_key, target_url, filters)
-    if data:
-        df = pd.DataFrame(data)
+    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)
 
-with col2:
-    st.header("Output")
-    if generate_button and target_url:
-        data = get_backlinks(api_login, api_key, target_url, filters)
-        if data:
-            df = pd.DataFrame(data)
+        # Display the DataFrame in the second column
+        with col2:
+            st.header("Output")
             st.dataframe(df)
 
 # Reset functionality