Update app.py

app.py

@@ -2,8 +2,6 @@ import streamlit as st
 import requests
 import json
 import os
-import zipfile
-import io
 
 # Load API Key from Hugging Face Secrets
 API_KEY = os.getenv("INSTA_API")
@@ -15,7 +13,7 @@ HEADERS = {
     "x-rapidapi-host": "instagram-scraper-api2.p.rapidapi.com"
 }
 
-# Function to extract all links recursively
+# Function to extract all links recursively from JSON
 def extract_links(obj):
     links = []
     if isinstance(obj, dict):
@@ -31,7 +29,7 @@ def extract_links(obj):
 
 # Streamlit UI
 st.set_page_config(page_title="Instagram Scraper", layout="wide")
-st.title("📸 Instagram Scraper")
+st.title("📸 Instagram Scraper API")
 
 # User Input
 username = st.text_input("Enter Instagram Username:", "mrbeast")
@@ -48,41 +46,16 @@ if st.button("Fetch Details"):
             st.subheader("📜 JSON Response")
             st.json(data)
 
-            # Extract all links
+            # Extracting Links from Nested JSON
             links = extract_links(data)
 
-            # Separate image links and other links
-            image_links = [link for link in links if any(ext in link for ext in [".jpg", ".jpeg", ".png", ".webp"])]
-            other_links = [link for link in links if link not in image_links]
-
-            # Display Images
-            if image_links:
-                st.subheader("🖼️ Extracted Images:")
-                img_bytes_list = []
-                img_filenames = []
-                
-                for idx, img_url in enumerate(image_links):  # Use image_links instead of links
-                    st.image(img_url, caption=f"Image {idx + 1}", use_container_width=True)
-
-                    # Download images as bytes
-                    img_bytes = requests.get(img_url).content
-                    img_bytes_list.append(img_bytes)
-                    img_filenames.append(f"{username}_image_{idx+1}.jpg")
-
-                # **Download All Images as ZIP**
-                if img_bytes_list:
-                    with io.BytesIO() as zip_buffer:
-                        with zipfile.ZipFile(zip_buffer, "w") as zip_file:
-                            for filename, img_data in zip(img_filenames, img_bytes_list):
-                                zip_file.writestr(filename, img_data)
-                        zip_buffer.seek(0)
-                        st.download_button("📥 Download All Images", data=zip_buffer, file_name=f"{username}_images.zip", mime="application/zip")
-
-            # Display Other Links
-            if other_links:
-                st.subheader("🔗 Other Links:")
-                for link in other_links:
+            # Display Links Below JSON
+            if links:
+                st.subheader("🔗 Extracted Links:")
+                for link in links:
                     st.markdown(f"[🔗 {link}]({link})")
+            else:
+                st.info("❌ No links found in the response.")
 
             # **Download Buttons Below Links**
             st.subheader("⬇️ Download Data")
@@ -97,9 +70,16 @@ if st.button("Fetch Details"):
             # Download Text File
             st.download_button(label="Download Text", data=text_data, file_name=f"{username}_details.txt", mime="text/plain")
 
+            # Download Profile Picture (If available)
+            profile_pic = data.get("profile_picture")
+            if profile_pic:
+                st.image(profile_pic, caption="Profile Picture")
+                img_bytes = requests.get(profile_pic).content
+                st.download_button(label="Download Profile Picture", data=img_bytes, file_name=f"{username}_profile.jpg", mime="image/jpeg")
+
         else:
             st.error("❌ Failed to retrieve data. Please check the username.")
 
 # Footer
 st.markdown("---")
-st.caption("Powered by Streamlit")
+st.caption("Powered by RapidAPI & Streamlit")