Spaces:
Build error
Build error
File size: 1,825 Bytes
6609086 7c70a38 6609086 7c70a38 daae4bc 6609086 daae4bc 6609086 daae4bc 6609086 c0fd3bb daae4bc c0fd3bb 6609086 daae4bc 6609086 daae4bc c0fd3bb daae4bc 7c70a38 7e3eb15 4450bb2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 |
import os
import cv2
import dlib
import numpy as np
import tensorflow as tf
import mediapipe as mp
from flask import Flask, render_template, Response
# Initialize Flask App
app = Flask(__name__)
# Load Face Detector (Dlib)
detector = dlib.get_frontal_face_detector()
# Load Pretrained Model for Skin Analysis (Placeholder)
model_path = "skin_model.h5"
if os.path.exists(model_path):
skin_model = tf.keras.models.load_model(model_path)
else:
skin_model = None
# OpenCV Video Capture
cap = cv2.VideoCapture(0)
# Function to Analyze Skin
def analyze_skin(frame):
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
faces = detector(gray)
for face in faces:
x, y, w, h = face.left(), face.top(), face.width(), face.height()
face_crop = frame[y:y+h, x:x+w]
if skin_model:
face_crop = cv2.resize(face_crop, (224, 224)) / 255.0
prediction = skin_model.predict(np.expand_dims(face_crop, axis=0))
return f"Skin Condition Score: {prediction[0][0]:.2f}"
return "No face detected"
# Video Stream Function
def generate_frames():
while True:
success, frame = cap.read()
if not success:
break
else:
skin_result = analyze_skin(frame)
cv2.putText(frame, skin_result, (50, 50), cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0, 255, 0), 2)
ret, buffer = cv2.imencode('.jpg', frame)
frame = buffer.tobytes()
yield (b'--frame\r\n'
b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
@app.route('/')
def index():
return render_template('index.html')
@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)
|