Update app.py

app.py

@@ -120,17 +120,14 @@ if upload_option == "Single Image Upload":
         with open(image_path, "wb") as f:
             f.write(uploaded_image.getbuffer())
 
-        # Process the image through the pipeline
-        output = pipeline.invoke({"image_path": image_path})
+        # Add button to trigger information extraction
+        if st.button("Extract Vehicle Information"):
+            # Process the image through the pipeline
+            output = pipeline.invoke({"image_path": image_path})
 
-        # Show the results in a user-friendly format
-        st.subheader("Extracted Vehicle Information")
-        st.json(output)
-
-        # Optionally, display more vehicle images from the folder
-        img_dir = "/content/images"
-        image_paths = glob.glob(os.path.join(img_dir, "*.*"))
-        display_image_grid(image_paths)
+            # Show the results in a user-friendly format
+            st.subheader("Extracted Vehicle Information")
+            st.json(output)
 
 # Batch Images Upload
 elif upload_option == "Batch Images Upload":
@@ -143,10 +140,12 @@ elif upload_option == "Batch Images Upload":
             with open(f"/tmp/{file.name}", "wb") as f:
                 f.write(file.getbuffer())
 
-        # Process the batch and display the results in a DataFrame
-        batch_output = pipeline.batch(batch_input)
-        df = pd.DataFrame(batch_output)
-        st.dataframe(df)
+        # Add button to trigger batch information extraction
+        if st.button("Extract Vehicle Information from Batch"):
+            # Process the batch and display the results in a DataFrame
+            batch_output = pipeline.batch(batch_input)
+            df = pd.DataFrame(batch_output)
+            st.dataframe(df)
 
         # Show images in a grid
         image_paths = [f"/tmp/{file.name}" for file in batch_images]