Update app.py

app.py

@@ -1,20 +1,16 @@
 from flask import Flask, render_template, request, jsonify
 import cv2
 import numpy as np
+import base64
 from flask_socketio import SocketIO
 
 app = Flask(__name__)
 socketio = SocketIO(app)
 
-# Function to perform object detection
-def perform_detection(image):
-    # Convert the frame to grayscale
+# Function to perform grayscale conversion
+def convert_to_grayscale(image):
     gray_frame = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-
-    # Perform your object detection on the grayscale frame
-    # Replace this with your object detection code
-    # For demonstration, we'll just return a dummy result
-    return {"objects_detected": ["car", "person"], "frame": gray_frame}
+    return gray_frame
 
 # Route to render index.html
 @app.route('/')
@@ -30,14 +26,21 @@ def detect():
     # Read the image
     image = cv2.imdecode(np.fromstring(image_file.read(), np.uint8), cv2.IMREAD_COLOR)
     
-    # Perform detection
-    detection_results = perform_detection(image)
+    # Perform grayscale conversion
+    gray_image = convert_to_grayscale(image)
+    
+    # Encode grayscale image to base64
+    retval, buffer = cv2.imencode('.jpg', gray_image)
+    gray_image_str = base64.b64encode(buffer).decode('utf-8')
     
-    # Emit detection results using socket
-    socketio.emit('detection_results', detection_results)
+    # Emit grayscale image using socket
+    socketio.emit('grayscale_image', {'image': gray_image_str})
     
     # Return a response to the client
-    return 'Detection complete'
+    return 'Grayscale conversion complete'
+
+if __name__ == '__main__':
+    socketio.run(app, debug=True)