Update app.py

app.py

@@ -2,6 +2,8 @@ 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")
@@ -13,7 +15,7 @@ HEADERS = {
     "x-rapidapi-host": "instagram-scraper-api2.p.rapidapi.com"
 }
 
-# Function to extract all links recursively from JSON
+# Function to extract all links recursively
 def extract_links(obj):
     links = []
     if isinstance(obj, dict):
@@ -46,16 +48,41 @@ if st.button("Fetch Details"):
             st.subheader("📜 JSON Response")
             st.json(data)
 
-            # Extracting Links from Nested JSON
+            # Extract all links
             links = extract_links(data)
 
-            # Display Links Below JSON
-            if links:
-                st.subheader("🔗 Extracted Links:")
-                for link in links:
+            # 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):
+                    st.image(img_url, caption=f"Image {idx + 1}", use_column_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:
                     st.markdown(f"[🔗 {link}]({link})")
-            else:
-                st.info("❌ No links found in the response.")
 
             # **Download Buttons Below Links**
             st.subheader("⬇️ Download Data")
@@ -70,13 +97,6 @@ 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.")