U =i@s UddlZddlZddlZddlZddlZddlZddlmZddlm Z mZddlmZ ddl m Z ddl mZddlZddlZddlZddlZddlmZmZmZmZddlmZdd lmZdd lmZmZm Z m!Z!m"Z"e#d Z$e d dZ%e d dZ&e'dddZ(dZ)dZ*ee)e*ZdZ+dZ,e-Z.e-e/e0d<dZ1dZ2dZ3ej4e2dde5ej6j7dZ8eddZ9eZ:GdddZ;e;Zej?e0d <ga@e>eAe0d!<eZBd"d#ZCeCe/e/d$d%d&ZDeEe'd'd(d)ZFdej?eGeEdBd+d,d-ZHej?d.d/d0ZIdej?e>eJe/eJfdBej?d1d2d3ZKej?eAd4d5d6ZLe>ej?e>eAd7d8d9ZMe/e/eAd:d;d<ZNe>ej?e>eAd7d=d>ZOej?e/e/e'd?d@dAZPGdBdCdCZQeQZRdddddDdDdEZSdFdGZTdHdIZUe:VdJe!dKdLdMZWe:VdNe!dKdOdPZXe>ej?dQdRdSZYdTdUZZe/eJeAe'fdVdWdXZ[e:\dYdZd[Z]e:\d\d]d^Z^e:_d_e d`dadbZ`e:_dcedddfe/dedfdgZae:\dhedddfe/dedidjZbdkZcdlZddmZednZfe:j\doedpdqdrZge:j\dsedpdtduZhe:j\dvedpdwdxZie:j\dyedpedddfe/dedzd{Zje:\d|edddfe/ded}d~ZkddZledddfe/deddZmedddfe/deddZne:_d_e d`ddbZ`e:_dcedddfe/deddgZae:\dhedddfe/deddjZbeodkrddlpZpepjqe:ddddS)N)ThreadPoolExecutor)datedatetime)time)BytesIO)Lock) HTMLResponseRedirectResponseResponseStreamingResponse)Image) create_client)CookieFastAPIRequest WebSocketWebSocketDisconnectz Asia/JakartareturncCs&tt}t|ko tkSS)z)Cek apakah sekarang dalam jam operasional)rnowTIMEZONErOPERATION_START OPERATION_END)rr)/home/afrizal/public_html/fastapi/main.pyis_operation_hours!srz(https://vtuamuvlfnouzekjhtsk.supabase.coZ.sb_publishable_0LJO9qBDgV29zXMPaS44Iw_3d4GP9uwzadmin@example.comZadmin123active_sessions facesg?T)exist_okz#haarcascade_frontalface_default.xml) max_workersc@s\eZdZdZddZddZeeefdddZe j d Bdd d Z e d Bdd d Z d S) FrameBufferzE Simpan frame terbaru di memory. Thread-safe pakai Lock. cCs*d|_d|_g|_g|_t|_d|_dSN)frame display_framer face_cropsrlock jpeg_bytesselfrrr__init__Ks zFrameBuffer.__init__c Cs4|j$||_||_||_||_||_W5QRXdSr%)r)r&r'r r(r*)r,r&r'r r(r*rrrupdateSs zFrameBuffer.updaterc Cs8|j(t|jdd|jDfW5QRSQRXdS)z*Ambil snapshot face_crops + faces saat inicSsg|] }|qSr)copy).0crrr ^sz0FrameBuffer.get_latest_crops..N)r)listr r(r+rrrget_latest_crops[szFrameBuffer.get_latest_cropsNc Cs6|j&|jdk r|jndW5QRSQRXdS)z!Ambil snapshot raw frame saat iniN)r)r&r/r+rrrget_latest_frame`szFrameBuffer.get_latest_framec Cs$|j|jW5QRSQRXdSr%)r)r*r+rrrget_jpegeszFrameBuffer.get_jpeg)__name__ __module__ __qualname____doc__r-r.tupler3r4npndarrayr5bytesr6rrrrr$Es r$KNOWN_ENCODINGS KNOWN_USERSc Cs4ztdddd}g}g}|jD]}|ds:q,tjt |d}tj |sht d|q,t |}|jdkr|dd}|D]&}||||d |d d qq,t|a|aW5QRXt d t|d ttdd|DdWn2tk r.}zt d|W5d}~XYnXdS)z-Load semua enrolled faces dari DB + file .npyuserszid, user_name, face_file is_enrolledT face_fileu⚠️ File tidak ditemukan: id user_name)user_idrGu ✅ Loaded z encodings, css|]}|dVqdS)rGNr)r0urrr sz!load_all_faces..z useru❌ load_all_faces error: N)supabasetableselecteqexecutedataospathjoin FACES_DIRexistsprintr<loadndimreshapeappend faces_lockr?r@lenset Exception)rAZ new_encodingsZ new_usersrIrRZencserrrload_all_facesus>       &r`)namercCstdd|S)Nz [^a-z0-9_]_)resubstriplower)rarrrnormalize_namesrg) image_bytesrcCs4ztt|WdStk r.YdSXdS)NTF)r openrverifyr^)rhrrris_valid_images rkU)r&qualityrcCs(td|tj|g\}}|r$|SdS)Nz.jpg)cv2ZimencodeZIMWRITE_JPEG_QUALITYtobytes)r&rmsuccessencodedrrrencode_to_jpegsrr)r&c Cst|tj}tj|ddd}g}|D]~\}}}}td|}td||} td||} t|jd|||} t|jd|||} | || | | | fq&||fS)zK Deteksi wajah + crop. Ini CEPAT (~5-15ms), aman di main loop. g?)Z scaleFactorZ minNeighborsg?rrD) rncvtColorZCOLOR_BGR2GRAY face_cascadeZdetectMultiScaleintmaxminshaperZ) r&Zgrayr r(xywhpadx1y1Zx2y2rrr detect_facess r)r&labelsrc Cs|}t|D]\}\}}}}|r>|t|kr>||\} } nd\} } t|||f||||f| d| rt|| |t|ddftjd| dq|S)zg Draw bounding box. Kalau labels None, box kuning polos. labels: list of (text, color_bgr) ))rrr"rg333333?)r/ enumerater\rnZ rectangleZputTextrwZFONT_HERSHEY_SIMPLEX) r&r rdisplayirzr{r|r}textcolorrrr draw_boxess"" r)face_bgrrc Cst6ttdkr&ddiW5QRStt}tt}W5QRXt|tj}t |}t|dkrlddiS|d}t t ||}t t |}t||}|tkr||} d| d| dt|ddSddiS) uI Identify 1 wajah. BLOCKING — panggil via asyncio.to_thread(). rstatusunknown recognizedrHrGrs)rrHradistance)r[r\r?r3r@rnrt COLOR_BGR2RGBface_recognitionface_encodingsZ face_distancer<arrayrvZargminfloat THRESHOLDround) rZencodings_snapshotZusers_snapshotface_rgb encodingsrZ distancesZbest_idx best_distanceuserrrridentify_face_syncs*    r)r(rcCs$g}|D]}t|}||q|S)uA Recognize semua wajah. BLOCKING — jalankan di thread. )rrZ)r(resultsZcropresultrrrrun_recognition_syncs  r)rHrGrc Cst}t}z~tddd|d|  }|j rTd| dWStd ||| | dd d | | dd WStk r}z$td |d t|d WYSd}~XYnXdS)ud Record attendance ke DB. BLOCKING — jalankan di thread. Cek duplikat di DB langsung. attendancerFrH checkin_dateZalready_checked_in)rrz%H:%M:%S)rHrGrZ checkin_timeZ checked_in)rrru❌ Attendance error: errorrmessageN)rtodayrrrrKrLrMrN isoformatrOrPinsertstrftimer^rVstr)rHrGrrexistingr_rrrrecord_attendance_syncs:    rcCs8t|}|D]&}|ddkr t|d|d|d<q |S)zj Gabung: recognition + attendance dalam 1 blocking function. Panggil via asyncio.to_thread(). rrrHrar)rr)r(rrrrrdo_recognition_and_attendanceBs  r) face_croprHrGrc Cst|tj}t|}t|dkreZdZddZedddZedddZedd d Zd S) ConnectionManagercCs g|_dSr%)active_connectionsr+rrrr-szConnectionManager.__init__ websocketcs4|IdH|j|tdt|jddS)Nu✅ WS connected (total: ))acceptrrZrVr\r,rrrrconnects zConnectionManager.connectcCs0||jkr,|j|tdt|jddS)Nu❌ WS disconnected (total: r)rremoverVr\rrrr disconnects  zConnectionManager.disconnect)rc s\g}|jD]8}z||IdHWq tk r@||Yq Xq |D]}||qHdSr%)r send_textr^rZr)r,rZdeadwsrrrbroadcast_texts z ConnectionManager.broadcast_textN) r7r8r9r-rrrrrrrrrrsrF)moderHrG enroll_countcapture_requested processingcCs4dtd<dtd<dtd<dtd<dtd<dtd<dS) NrrHrGrrFrr)SESSIONrrrr reset_sessions rcsXt}tdIdHt}||kr|r,dnd}t|IdHtd||}qdS)z Background task yang jalan terus untuk broadcast schedule changes. Cek waktu tiap 30 detik, broadcast saat transisi. NitsWorkingTimenotWorkingTimeu📢 Schedule broadcast: )rasynciosleepmanagerrrV)Z last_stateZ current_staterrrrschedule_broadcasters rz /ws/streamrc st|IdHzzr|IdH}|ddkr6q|ddkrd|kr|d}|sXqtj|tjd}t|tj }|dkrqt |\}}t ||}t |}t |||||tddkrtdrtd stt|ddkrd |kr|d } | d kr2trd nd } || IdHtd| q| dkr^tttddiIdHq| dkrtddkrdtd<tdqqWntk rYnXW5t|XdS)Ntypewebsocket.disconnectwebsocket.receiver>)ZdtyperENROLLrrrisWorkingTime?rr"📞 ESP32 query working time → stopcaptureTu!📸 Capture requested dari ESP32)rrrreceiver<Z frombufferZuint8rnZimdecodeZ IMREAD_COLORrrrrbufferr.rr create_taskprocess_enroll_capturererrrVrrjsondumpsr) rrrawZ img_arrayr&r r(rjpegrresponserrr ws_streamsV         rz/ws/cmdc s|t|IdHzXz:|IdH}|ddkr6qN|ddkrd|kr|d}td|d|dkrttt ddiIdHq|dkrt rd nd }| |IdHtd |q|d krtt d dddIdHtdq|dkr(tt d dddIdHtdq|dkrt srtt d ddddIdHtdt dIdHt\}}t|dkr| t ddddIdHt std tt d dddIdHq| t d!d"d#IdHt t||qzt|}Wntjk r0YqYnX|d }|d$kr|d%pTd&}|s~| t d'd(d#IdHqtd)|s| t d'd*d#IdHqzt t|IdH\} } WnLtk r} z,| t d't| d#IdHWYqW5d} ~ XYnX| d+} | rtjt| } tj | rt!"| }|j#d,kr`|$d,d-}t|t%kr| t d'd|d.t|d/d#IdHqtd0t&d1<| d2t&d3<|t&d4<dt&d5<tt d6|dt%d7IdHtd8|dq|d9kr$t&d1d0krLd:t&d;<td<q|d=krttt ddiIdHqWnt'k rfYnXW5t|XdS)>Nrrrru 📨 CMD: 'rrrrrrZ ForceStartcommandesp32ZSTARTrtargetactionu&📞 ESP32 ForceStart command receivedZ ForceStopSTOPu%📞 ESP32 ForceStop command receivedZ recognizeZ TEMP_STARTr)rrrtimeoutu"⏳ Waiting for stream to start...g@rrecognize_resultZno_facezTidak ada wajah terdeteksi)rrru1🛑 Stopping temporary stream (no face detected)Zrecognize_statusz Memproses...rrZ start_enrollnamarrzNama tidak boleh kosongz^[a-zA-Z0-9_ ]+$zNama tidak validrCrDrEz' sudah ter-enroll (z encodings)rrrFrHrGrZenroll_started)rracounttotalu🔵 Enroll started: 'rTru%📸 Capture requested dari dashboardZ stop_enroll)(rrrrrerVrrrrrrrrrr4r\rrun_recognize_taskloadsJSONDecodeErrorgetrcmatch to_threadget_or_create_user_syncr^rrQrRrSrTisfiler<rWrXrY ENROLL_TARGETrr)rrrrr r(rPcmdrrcreatedr_rCZnpy_pathrrrrws_cmds.                              r)r(c slztt|IdH}g}|D].}|ddkr@||ddfq|dqt}|dk rt|||}t|}tj|t_ |t_ W5QRXt t d|dIdHtd d d |Dtst t d d ddIdHWn~tk rf}z^td|t t ddt|dIdHtsVt t d d ddIdHW5d}~XYnXdS)zARecognition + attendance di background thread, broadcast hasilnyaNrrra)rrr)ZUnknown)rrrr)rr u🎯 Recognize done: cSsg|]}|ddqS)rar)r)r0rrrrr2 sz&run_recognize_task..rrrru❌ Recognition error: rzRecognition error: r)rrrrZrr5rrrr)r'r*rrrrrVrr^r) r r(rrrr&Zlabeledrr_rrrrsR      rc sdtd<dtd<zƐzft\}}t|dkrdttddt|dd Id HWWzd St t |d td td Id H}|sttddd Id HWW,d Stdd7<td}t d|dt ttd|t td dId H|t krztd }t t tId Httd|d|ddId HttddiId Ht d|dWnXtk r}z8t d|ttddt|d Id HW5d }~XYnXW5dtd<Xd S)z%Proses 1 enroll capture di backgroundTrFrrDZenroll_warningz Ditemukan z wajah. Harus tepat 1.rNrrHrGzGagal enroll. Coba lagi.ru📸 Enroll progress: /Zenroll_progress)rrrraZ enroll_donezEnroll 'z ' selesai!)rrarrru✅ Enroll selesai: 'ru❌ Enroll error: rzEnroll error: )rrr4r\rrrrrrrrVrrr`r^r)rbr(rprrr_rrrr(st           r)rGrcCshtddd|}|jr0|jddfStd|ddd}|jsZtd|jddfS) NrA*rGrF)rGrBrCzGagal membuat userT)rKrLrMrNrOrPr RuntimeError)rGrresrrrrxs  rz/streamcCsdd}t|ddS)Ncss*t}|rd|dVtdqdS)Ns%--frame Content-Type: image/jpeg s g?)rr6rr)rrrr generatorszstream_video..generatorz)multipart/x-mixed-replace; boundary=frame) media_type)r )rrrr stream_videos rz/imagecCsJt}|rt|ddStjdddd}t}||dt|ddS)Nz image/jpegcontentrZRGB)i@)rr)rZJPEG)rr6r r newrrgetvalue)rZblankbufrrr get_images  rz /auth/login)requestcs|IdH}|dtkrl|dtkrltd}t|tt ddidd}|j d|d d d d |Stt d dddddS)Nemailpassword rrpapplication/jsonr session_tokenTlaxQ)httponlysamesitemax_agerInvalid credentialsrr status_coder) rr ADMIN_EMAILADMIN_PASSWORDsecrets token_hexraddr r set_cookie)rbodytokenresprrrlogins*   r z /auth/logout)defaultr cs(|rt|tddd}|d|S)N/login/urlrr rdiscardr Z delete_cookie)r rrrrlogouts    r)z /auth/checkcs,|r|tkrddiSttddidddS)N authenticatedTFrr rrr rrr"rrr check_auths  r,u$ Rekap Absensi Bulanan

📊 Rekap Absensi Bulanan

⚙️ Admin
Nama Bulan Total Absen
Memuat data...
💡 Klik pada nama untuk melihat detail absensi
u! Detail Absensi
← Kembali

📋 Detail Absensi

-
-
-
# Tanggal Hari Check-in
Memuat data...
u Admin Login un Admin Dashboard
📊 Rekap Absensi 🎯 Face Recognition 👥 User Management

Face Recognition

🔒 Admin Only
Status
Idle
WebSocket
Disconnected
Enrolled
0
📹 Menunggu stream dari ESP32...
Status
Connection Disconnected
Mode —
Stream Menunggu
—
0 / 10 0%
Hasil Recognition
—

📸 Enroll Baru

Masukkan nama untuk enrollment wajah

r)Zresponse_classcstS)zHalaman rekap bulanan) html_rekaprrrr get_rekap sr.z/detailcstS)zHalaman detail absensi per nama) html_detailrrrr get_detail# sr0r#cCstS)zHalaman login - PUBLIC) html_loginrrrr login_page) sr2z /dashboardcCs|r |tkrtdddStS)zAdmin Dashboard - PROTECTEDr#r$r%)rr html_dashboardr"rrr dashboard/ s  r4z/userscCs |r |tkrtdddStdS)zUser Management - PROTECTEDr#r$r%uFv User Management
📊 Rekap Absensi 🎯 Face Recognition 👥 User Management

User Management

🔒 Admin Only
🔍
0 user
Nama ↕ Face File Status ↕ Dibuat ↕ Aksi
⏳

Memuat data...
)rr rr"rrr users_page? s  r5cCs tdS)zGenerate unique session tokenr )rrrrrrgenerate_token sr6cCs|r|tkrdSdS)z Check apakah session token validTFrr"rrrget_admin_from_cookie s r8cCs|r |tkrdSdS)z.Dependency: redirect ke login jika belum loginNTr7r"rrr require_admin s r9cs|IdH}|dd}|dd}|tkrv|tkrvt}t|ttddddd }|j d |d d d d|SttddddddSdS)z Login adminNr rr rpzLogin successfulrr rr Trr)keyvaluerrrrrrr) rrrrr6rrr rr)rrr r rrrrrr  s.   cs2|r|tkrt|tddd}|jdd|S)z Logout adminr#r$r%r )r:r')r rrrrr) s     cs,|r|tkrddiSttddidddS)z$Check status login (dipakai oleh JS)r*TFrr rr+r"rrrr,s  __main__z0.0.0.0i@)hostport)rl)N)rrrrQrcrrconcurrent.futuresrrrZdt_timeior threadingrrnrZnumpyr<ZpytzZfastapi.responsesrr r r ZPILr rKr ZfastapirrrrrtimezonerrrboolrZ SUPABASE_URLZ SUPABASE_KEYrrr]rr__annotations__rrTrmakedirsZCascadeClassifierrPZ haarcascadesruexecutorappr$rr?r3r=r@dictr[r`rgr>rkrvrrrr;rrrrrrrrrrrrrrrrrrrrpostr r)r,r-r/r1r3r.r0r2r4r5r6r8r9r7uvicornrunrrrrs              %, " '.  NQ9P   ~8