Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import cv2
-from ultralytics import YOLO, solutions
+from ultralytics import YOLO
 
 # Initialize the YOLO model
 model = YOLO("yolov8s.pt")
@@ -14,14 +14,6 @@ def process_video(video_path, analytics_type):
     output_filename = f"{analytics_type}_output.avi"
     out = cv2.VideoWriter(output_filename, cv2.VideoWriter_fourcc(*"MJPG"), fps, (w, h))
     
-    # Set up analytics based on the selected type
-    analytics = solutions.Analytics(
-        type=analytics_type,
-        writer=out,
-        im0_shape=(w, h),
-        view_img=False
-    )
-    
     clswise_count = {}
     frame_count = 0
 
@@ -41,16 +33,24 @@ def process_video(video_path, analytics_type):
                     else:
                         clswise_count[model.names[int(cls)]] = 1
 
-            # Update analytics based on type
+            # Perform simple analytics based on type
             if analytics_type == "line":
-                total_counts = sum(clswise_count.values())
-                analytics.update_line(frame_count, total_counts)
+                # Display the number of detections on each frame (for example)
+                cv2.putText(frame, f"Frame {frame_count}: Detections - {sum(clswise_count.values())}", 
+                            (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2, cv2.LINE_AA)
             elif analytics_type == "multiple_line":
-                analytics.update_multiple_lines(clswise_count, list(clswise_count.keys()), frame_count)
+                # Display classwise counts
+                for i, (cls_name, count) in enumerate(clswise_count.items()):
+                    cv2.putText(frame, f"{cls_name}: {count}", 
+                                (10, 30 + (i + 1) * 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2, cv2.LINE_AA)
             elif analytics_type == "pie":
-                analytics.update_pie(clswise_count)
+                # Placeholder for pie chart (can implement later with matplotlib and overlay on frame)
+                pass
             elif analytics_type == "area":
-                analytics.update_area(frame_count, clswise_count)
+                # Placeholder for area graph (implement as needed)
+                pass
+
+            out.write(frame)
 
             clswise_count = {}  # Reset for next frame
 
@@ -92,5 +92,5 @@ with gr.Blocks() as demo:
     # Define the output when the button is clicked
     submit_btn.click(gradio_app, inputs=[video_input, analytics_dropdown], outputs=output_video)
 
-# Launch the Gradio app
-demo.launch()
+# Launch the Gradio app with a public link
+demo.launch(share=True)