Fix: Corrected detect_objects function call and return handling

app.py

@@ -176,30 +176,31 @@ def main():
             if st.button("🚀 Analyze Image", type="primary"):
                 with st.spinner("🔍 Detecting objects..."):
                     # Perform object detection
-                    results = detect_objects(image, confidence_threshold=confidence)
-
-                    if results and len(results['detections']) > 0:
-                        # Extract detected objects
-                        detected_objects = [det['class'] for det in results['detections']]
+                    annotated_img, detected_objects = detect_objects(image)
 
+                    if detected_objects and len(detected_objects) > 0:
                         # Display results
                         st.success(f"✅ Detected {len(detected_objects)} objects!")
 
                         # Show annotated image
                         st.markdown("### 🎯 Detection Results")
-                        if 'annotated_image' in results:
-                            st.image(results['annotated_image'], caption="Detected Objects", use_column_width=True)
+                        st.image(annotated_img, caption="Detected Objects", use_column_width=True)
 
                         # Show detection details
                         st.markdown("### 📋 Detected Objects")
-                        for i, detection in enumerate(results['detections']):
+                        # Count unique objects
+                        object_counts = {}
+                        for obj in detected_objects:
+                            object_counts[obj] = object_counts.get(obj, 0) + 1
+
+                        for i, (obj_name, count) in enumerate(object_counts.items()):
                             col_a, col_b, col_c = st.columns(3)
                             with col_a:
-                                st.metric("Object", detection['class'])
+                                st.metric("Object", obj_name)
                             with col_b:
-                                st.metric("Confidence", f"{detection['confidence']:.2%}")
+                                st.metric("Count", count)
                             with col_c:
-                                st.metric("Count", f"#{i+1}")
+                                st.metric("ID", f"#{i+1}")
 
                         # AI Explanation
                         if os.getenv("OPENAI_API_KEY"):
@@ -214,7 +215,7 @@ def main():
                             st.warning("🔑 Add OPENAI_API_KEY in settings for AI explanations")
 
                     else:
-                        st.warning("❌ No objects detected. Try adjusting the confidence threshold.")
+                        st.warning("❌ No objects detected in this image.")
 
     with col2:
         st.markdown("### 📊 Detection Statistics")