Update app.py

app.py

@@ -1,32 +1,84 @@
-from flask import Flask, Response, render_template
+# Import necessary modules
+from flask import Flask, render_template, Response, request, jsonify, redirect, url_for
+from aiortc import RTCPeerConnection, RTCSessionDescription
 import cv2
-import base64
+import json
+import uuid
+import asyncio
+import logging
+import time
 
-app = Flask(__name__)
-camera = cv2.VideoCapture(0)
+# Create a Flask app instance
+app = Flask(__name__, static_url_path='/static')
 
+# Set to keep track of RTCPeerConnection instances
+pcs = set()
+
+# Function to generate video frames from the camera
 def generate_frames():
+    camera = cv2.VideoCapture(0)
     while True:
+        start_time = time.time()
         success, frame = camera.read()
         if not success:
             break
         else:
-            # Preprocess the frame here (e.g., resize, color conversion)
-            processed_frame = frame  # Placeholder for preprocessing
-
-            _, buffer = cv2.imencode('.jpg', processed_frame)
-            frame_bytes = base64.b64encode(buffer.tobytes()).decode('utf-8')
-
-            yield frame_bytes
+            ret, buffer = cv2.imencode('.jpg', frame)
+            frame = buffer.tobytes()
+            # Concatenate frame and yield for streaming
+            yield (b'--frame\r\n'
+                   b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n') 
+            elapsed_time = time.time() - start_time
+            logging.debug(f"Frame generation time: {elapsed_time} seconds")
 
+# Route to render the HTML template
 @app.route('/')
 def index():
     return render_template('index.html')
+    # return redirect(url_for('video_feed')) #to render live stream directly
+
+# Asynchronous function to handle offer exchange
+async def offer_async():
+    params = await request.json
+    offer = RTCSessionDescription(sdp=params["sdp"], type=params["type"])
+
+    # Create an RTCPeerConnection instance
+    pc = RTCPeerConnection()
+
+    # Generate a unique ID for the RTCPeerConnection
+    pc_id = "PeerConnection(%s)" % uuid.uuid4()
+    pc_id = pc_id[:8]
+
+    # Create a data channel named "chat"
+    # pc.createDataChannel("chat")
+
+    # Create and set the local description
+    await pc.createOffer(offer)
+    await pc.setLocalDescription(offer)
+
+    # Prepare the response data with local SDP and type
+    response_data = {"sdp": pc.localDescription.sdp, "type": pc.localDescription.type}
+
+    return jsonify(response_data)
+
+# Wrapper function for running the asynchronous offer function
+def offer():
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    
+    future = asyncio.run_coroutine_threadsafe(offer_async(), loop)
+    return future.result()
+
+# Route to handle the offer request
+@app.route('/offer', methods=['POST'])
+def offer_route():
+    return offer()
 
-@app.route('/capture_frame', methods=['GET', 'POST'])
-def capture_frame():
-    frame = next(generate_frames())
-    return frame
+# Route to stream video frames
+@app.route('/video_feed')
+def video_feed():
+    return Response(generate_frames(), mimetype='multipart/x-mixed-replace; boundary=frame')
 
-if __name__ == '__main__':
-    app.run(debug=True)
+# Run the Flask app
+if __name__ == "__main__":
+    app.run(debug=True, host='0.0.0.0')