feature to directly paste link of insta post and get the product listing details added. (#7)

- feature to directly paste link of insta post and get the product listing details added. (f7c3ac83f49325e86794156ac5325119e16a8bbb)


Co-authored-by: Ratan Prakash Mishra <RatanPrakash@users.noreply.huggingface.co>

app.py

@@ -11,6 +11,22 @@ import dateparser
 import os
 import matplotlib.pyplot as plt
 
+# Function to get Instagram post details
+import instaloader
+def get_instagram_post_details(post_url):
+    try:
+        shortcode = post_url.split('/')[-2]
+        post = instaloader.Post.from_shortcode(L.context, shortcode)
+
+        # Retrieve caption and image URL
+        caption = post.caption
+        image_url = post.url
+        
+        return caption, image_url
+    except Exception as e:
+        return str(e), None
+
+
 # Initialize PaddleOCR model
 ocr = PaddleOCR(use_angle_cls=True, lang='en')
 
@@ -243,6 +259,27 @@ elif app_mode == "Team Details":
 elif app_mode == "Task 1":
     st.write("## Task 1: 🖼️ OCR to Extract Details 📄")
     st.write("Using OCR to extract details from product packaging material, including brand name and pack size.")
+
+    
+    # Instantiate Instaloader
+    L = instaloader.Instaloader()
+    
+    # Streamlit UI
+    st.title("Instagram Post Details Extractor")
+    
+    # Text input for Instagram post URL
+    post_url = st.text_input("Enter Instagram Post URL:")
+
+    if post_url:
+        caption, image_url = get_instagram_post_details(post_url)
+        
+        if image_url:
+            st.subheader("Caption:")
+            st.write(caption)
+            st.subheader("Image:")
+            st.image(image_url, use_column_width=True)
+        else:
+            st.error("Failed to retrieve the post details. Please check the URL.")
     
     # File uploader for images (supports multiple files)
     uploaded_files = st.file_uploader("Upload images of products", type=["jpeg", "png", "jpg"], accept_multiple_files=True)