Update app.py

app.py

@@ -3,10 +3,6 @@ from open_image_models import LicensePlateDetector
 from PIL import Image
 import cv2
 import numpy as np
-from rich.console import Console
-
-# Set up the rich console for better terminal output
-console = Console()
 
 # Define the available models
 PlateDetectorModel = ['yolo-v9-t-640-license-plate-end2end', 
@@ -15,11 +11,11 @@ PlateDetectorModel = ['yolo-v9-t-640-license-plate-end2end',
                       'yolo-v9-t-256-license-plate-end2end']
 
 # Streamlit interface
-st.title("🚗 License Plate Detection with Open Image Models 🚓")
-st.write("Select a model and upload an image to perform license plate detection.")
+st.title("🦀 Open Image Models: Pre-trained Models for Object Detection")
+st.write("Leverage fast and efficient pre-trained ONNX models for various object detection tasks, starting with license plate detection.")
 st.markdown("---")
 
-# Model selection dropdown
+# Model selection dropdown (specific to license plate detection in this example)
 selected_model = st.selectbox("🔍 Select a License Plate Detection Model", PlateDetectorModel)
 
 # File uploader for images
@@ -43,14 +39,18 @@ if uploaded_file is not None:
     # Perform license plate detection
     detections = lp_detector.predict(image_cv2)
 
-    # Display the detected plates using `rich` for colorful output in the console
-    console.print(f"[bold green]Detections: [/bold green] {detections}")
-
     # Streamlit display for detections
     if detections:
         st.success(f"✅ {len(detections)} License Plates Detected!")
-        for i, detection in enumerate(detections):
-            st.write(f"**Plate {i+1}:** {detection}")
+        
+        # Use an expander to show details in a more organized way
+        with st.expander("See detected plates details"):
+            for i, detection in enumerate(detections):
+                st.markdown(f"""
+                **Plate {i+1}:**  
+                - **Bounding Box:** {detection['bbox']}  
+                - **Confidence:** {detection['confidence']:.2f}
+                """)
     else:
         st.warning("⚠️ No license plates detected!")
 
@@ -70,5 +70,9 @@ st.markdown("""
         color: white;
         border-radius: 8px;
     }
+    .stImage img {
+        border-radius: 10px;
+        padding: 10px;
+    }
 </style>
 """, unsafe_allow_html=True)