Update app.py

app.py

@@ -1,33 +1,24 @@
 import streamlit as st
-from bs4 import BeautifulSoup
 import requests
-from selenium import webdriver
 import time
-import io
 from PIL import Image
+from io import BytesIO
 
 def download_instagram_image(link):
-    options = webdriver.ChromeOptions()
-    options.add_argument('--headless')
-    options.add_argument('--disable-gpu')
-    options.add_argument('--no-sandbox')
-
-    with webdriver.Chrome('chromedriver', options=options) as driver:
-        driver.get(link)
-        time.sleep(2)  # Allow time for the page to load
+    response = requests.get(link)
+    
+    if response.status_code == 200:
+        data = response.json()
+        img_url = data["graphql"]["shortcode_media"]["display_url"]
         
-        soup = BeautifulSoup(driver.page_source, 'lxml')
-        img = soup.find('img', class_='FFVAD')
-
-        if img:
-            img_url = img['src']
-            img_data = requests.get(img_url).content
-
-            image = Image.open(io.BytesIO(img_data))
-            image.save(f"instagram_{int(time.time())}.png")
-            st.success('Image downloaded successfully!')
-        else:
-            st.error('Could not find the image on the provided URL.')
+        img_response = requests.get(img_url)
+        img_data = BytesIO(img_response.content)
+
+        image = Image.open(img_data)
+        image.save(f"instagram_{int(time.time())}.png")
+        st.success('Image downloaded successfully!')
+    else:
+        st.error('Error fetching Instagram data. Please check the provided URL.')
 
 def main():
     st.title("Instagram Image Downloader")