Add files via upload

Author: SentryCoderDev

app.py

@@ -1,17 +1,17 @@
-from flask import Flask, render_template, Response, request, jsonify
+from fastapi import FastAPI, WebSocket, Request
+from fastapi.responses import HTMLResponse, StreamingResponse
+from fastapi.templating import Jinja2Templates
 import cv2
 import pytesseract
-import time
 import threading
 import serial
+import asyncio
 
-app = Flask(__name__)
+app = FastAPI()
+templates = Jinja2Templates(directory="templates")
 
 # All Cameras
-outside_garage_cam = cv2.VideoCapture(0)  
-inside_garage_cam = cv2.VideoCapture(1)   
-entrance_cam = cv2.VideoCapture(2)        
-
+cameras = [cv2.VideoCapture(i) for i in range(3)]
 ser = serial.Serial('/dev/ttyUSB0', 115200)  # Arduino's serial port
 result_message = ""
 
@@ -24,8 +24,11 @@
 # Pytesseract configuration
 pytesseract.pytesseract.tesseract_cmd = '/usr/bin/tesseract'  # Path to Tesseract OCR executable
 
-def gen_frames(cam):
+def gen_frames(cam_index):
+    cam = cameras[cam_index]
     while True:
+        if not cam.isOpened():
+            cam.open(cam_index)
         success, frame = cam.read()
         if not success:
             break
@@ -35,9 +38,9 @@ def gen_frames(cam):
             yield (b'--frame\r\n'
                    b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
 
-def garage_control():
+async def garage_control():
     while True:
-        ret, frame = outside_garage_cam.read()
+        ret, frame = cameras[0].read()
         if ret:
             gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
             # Perform OCR using Pytesseract
@@ -46,68 +49,59 @@ def garage_control():
             # Check if the target plate number is recognized
             if "your_target_plate" in text:
                 ser.write(b'OPEN_GARAGE\n')
-                print("Opening garage for plate 'your_target_plate'")
-        time.sleep(1)
-
-def audio_detection():
-    while True:
-        if detect_doorbell():
-            print("ringtone detected!")
-            # Start camera streaming at home entrance
-            # Notify and view the camera as needed
-        time.sleep(1)
+                print("Garage opens for plate 'your_target_plate'")
+        await asyncio.sleep(1)
 
-def detect_doorbell():
-    return False
-
-def read_from_arduino():
+async def read_from_arduino():
     global result_message
     while True:
         if ser.in_waiting > 0:
             line = ser.readline().decode('utf-8').strip()
             if line.startswith("RESULT:"):
                 result_message = line.split("RESULT:")[1]
+        await asyncio.sleep(0.1)
+
+@app.get("/", response_class=HTMLResponse)
+async def index(request: Request):
+    return templates.TemplateResponse("index.html", {"request": request})
+
+@app.get("/video_feed/{cam_id}")
+async def video_feed(cam_id: int):
+    if 0 <= cam_id < len(cameras):
+        return StreamingResponse(gen_frames(cam_id), media_type="multipart/x-mixed-replace; boundary=frame")
+    return "Invalid Camera ID", 404
 
-@app.route('/video_feed/<int:cam_id>')
-def video_feed(cam_id):
-    if cam_id == 0:
-        return Response(gen_frames(outside_garage_cam), mimetype='multipart/x-mixed-replace; boundary=frame')
-    elif cam_id == 1:
-        return Response(gen_frames(inside_garage_cam), mimetype='multipart/x-mixed-replace; boundary=frame')
-    elif cam_id == 2:
-        return Response(gen_frames(entrance_cam), mimetype='multipart/x-mixed-replace; boundary=frame')
-    else:
-        return "Invalid Camera ID", 404
-
-@app.route('/open_door', methods=['POST'])
-def open_door():
+@app.post("/open_door")
+async def open_door():
     ser.write(b'OPEN_DOOR\n')
-    return "Door Opened", 200
-
-@app.route('/validate_password', methods=['POST'])
-def validate_password():
-    data = request.json
-    user = data.get('user')
-    password = data.get('password')
-    
+    return {"message": "Door Opened"}
+
+@app.post("/validate_password")
+async def validate_password(user: str, password: str):
     if user in user_passwords and user_passwords[user] == password:
         ser.write(b'VALID_PASSWORD\n')
-        return "Password Validated", 200
-    else:
-        return "Invalid Password", 401
+        return {"message": "Password Validated"}
+    return {"message": "Invalid Password"}, 401
 
-@app.route('/result')
-def result():
+@app.get("/result")
+async def result():
     global result_message
-    return jsonify(result=result_message)
+    return {"result": result_message}
+
+@app.websocket("/ws")
+async def websocket_endpoint(websocket: WebSocket):
+    await websocket.accept()
+    while True:
+        data = await websocket.receive_text()
+        if data == "result":
+            await websocket.send_text(result_message)
+
+if __name__ == "__main__":
+    threading.Thread(target=asyncio.run, args=(garage_control(),)).start()
+    threading.Thread(target=asyncio.run, args=(read_from_arduino(),)).start()
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)
+
 
-@app.route('/')
-def index():
-    return render_template('index.html')
 
-if __name__ == '__main__':
-    threading.Thread(target=garage_control).start()
-    threading.Thread(target=audio_detection).start()
-    threading.Thread(target=read_from_arduino).start()
-    app.run(host='0.0.0.0', port=5000, debug=True)
 

ra.jpg

@@ -125,65 +125,49 @@
     <header class="header">
         <h1>H3RU Home Control System</h1>
     </header>
-
     <div class="content">
-        <!-- Camera Feed Left -->
         <div class="camera-feed">
             <h3>Outside Garage Camera</h3>
             <div class="camera-box">
                 <img src="{{ url_for('video_feed', cam_id=0) }}" alt="Outside Garage Camera">
             </div>
         </div>
-
-        <!-- Logo in the Center -->
         <div class="logo-container">
-            <img src="file:///C:/Users/emohi/Desktop/ra.jpg" alt="H3RU Logo">
+            <img src="file:///C:/Users/emohi/Desktop/H3RU-Home_Control_System-main/ra.jpg" alt="H3RU Logo">
             <p>To everyone who can't finish what they started<br>- H3RU</p>
         </div>
-
-        <!-- Camera Feed Right -->
         <div class="camera-feed">
             <h3>Inside Garage Camera</h3>
             <div class="camera-box">
                 <img src="{{ url_for('video_feed', cam_id=1) }}" alt="Inside Garage Camera">
             </div>
         </div>
     </div>
-
-    <!-- Button Below the Logo -->
     <div class="centered-button">
         <button id="open-door">Open Door</button>
     </div>
-
-    <!-- Third Camera Feed Below the Button -->
     <div class="camera-feed">
         <h3>Entrance Camera</h3>
         <div class="camera-box">
             <img src="{{ url_for('video_feed', cam_id=2) }}" alt="Entrance Camera">
         </div>
     </div>
-
     <div id="result-message"></div>
-
     <script>
+        const ws = new WebSocket("ws://localhost:8000/ws");
         document.getElementById('open-door').addEventListener('click', function () {
             fetch('/open_door', { method: 'POST' })
-                .then(response => response.text())
-                .then(data => {
-                    document.getElementById('result-message').innerText = data;
-                });
-        });
-
-        function fetchResult() {
-            fetch('/result')
                 .then(response => response.json())
                 .then(data => {
-                    document.getElementById('result-message').innerText = data.result;
+                    document.getElementById('result-message').innerText = data.message;
                 });
-        }
-
-        setInterval(fetchResult, 1000); // Check results every second
+        });
+        ws.onmessage = function (event) {
+            document.getElementById('result-message').innerText = event.data;
+        };
+        setInterval(() => ws.send("result"), 1000);
     </script>
 </body>
 
 </html>
+