Update app.py

app.py

@@ -164,6 +164,8 @@ st.sidebar.write("DELHI TECHNOLOGICAL UNIVERSITY")
 st.sidebar.title("Navigation")
 st.sidebar.write("Team Name: sadhya")
 app_mode = st.sidebar.selectbox("Choose the task", ["Welcome","Project Details", "Task 1","Team Details"])
+
+
 if app_mode == "Welcome":
     # Navigation Menu
     st.write("# Welcome to Amazon Smbhav! 🎉")
@@ -250,12 +252,12 @@ elif app_mode=="Project Details":
 
     """)
     
-
 elif app_mode == "Team Details":
     st.write("## Meet Our Team:")
     display_team_members(team_members)
     st.write("Delhi Technological University")
 
+
 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.")
@@ -271,15 +273,21 @@ elif app_mode == "Task 1":
     post_url = st.text_input("Enter Instagram Post URL:")
 
     if post_url:
-        caption, image_url = get_instagram_post_details(post_url)
+    caption, image_path = get_instagram_post_details(post_url)
+    
+    if image_path and os.path.exists(image_path):
+        st.subheader("Caption:")
+        st.write(caption)
+        st.subheader("Image:")
         
-        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.")
+        # Load and display the image
+        image = Image.open(image_path)
+        st.image(image, use_column_width=True)
+        
+        # Clean up (optional)
+        os.remove(image_path)
+    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)
@@ -327,8 +335,8 @@ elif app_mode == "Task 1":
                             product_name = clean_text_line
                         else:
                             product_details += clean_text_line + " "
-
             return product_name, product_details.strip()
+            
         if st.button("Start Analysis"):
             simulate_progress()
             # Loop through each uploaded image and process them
@@ -344,6 +352,10 @@ elif app_mode == "Task 1":
                 st.write(f"Extracting details from {uploaded_image.name}...")
                 result = ocr.ocr(img_array, cls=True)
 
+                #############################
+                #OCR result text to be parsed here through LLM and get product listing content.
+                #############################
+
                 # Process the OCR result to extract product name and properties
                 product_name, product_details = extract_product_info(result)