Update app/app_ui.py

app/app_ui.py

@@ -1,8 +1,45 @@
 import gradio as gr
-
+import numpy as np
+import whisper
+import torch
+import scipy.io.wavfile
+import cv2
+from app.safety_check import is_prompt_safe, check_nsfw_image, check_violence_image, check_url
 from app.gen_ai import generate_response
 from app.mlops_logger import log_prompt
-from app.safety_check import check_nsfw_image, check_violence_image, is_prompt_safe, check_url, transcribe_and_check
+
+# Load Whisper model
+asr_model = whisper.load_model("tiny", device="cuda" if torch.cuda.is_available() else "cpu")
+
+# Hàm chuyển giọng nói thành văn bản
+def transcribe_and_check(audio):
+    if audio is None:
+        return "❌ Không có dữ liệu âm thanh", "", ""
+
+    sr, data = audio
+
+    # Chuyển dữ liệu âm thanh sang dạng float32
+    audio_fp32 = whisper.pad_or_trim(data.astype(np.float32) / 32768.0)
+    
+    # Tạo spectrogram
+    mel = whisper.log_mel_spectrogram(audio_fp32).to(asr_model.device)
+
+    # Giải mã
+    result = asr_model.decode(mel)
+    text = result.text.strip()
+
+    if not text:
+        return "⚠️ Không nhận diện được giọng nói.", "", ""
+
+    # Kiểm tra prompt an toàn
+    is_safe, categories = is_prompt_safe(text)
+
+    if is_safe:
+        return "✅ Nội dung an toàn", text, ""
+    else:
+        cat_text = ", ".join(categories)
+        return f"🚨 Phát hiện nội dung không an toàn", text, f"❗ Các danh mục phát hiện: {cat_text}"
+
 
 # === Kiểm duyệt Prompt ===
 def handle_prompt(prompt):
@@ -10,11 +47,11 @@ def handle_prompt(prompt):
     if not safe:
         log_prompt(prompt, info, False, "")
         return f"🚨 Prompt không an toàn! Phát hiện: {', '.join(info)}", ""
-    
+
     response = generate_response(prompt)
     log_prompt(prompt, "OK", True, response)
     return "✅ Prompt an toàn", response
-    
+
 # === Giao diện ===
 with gr.Blocks(title="SAIFGuard - HỆ THỐNG KIỂM DUYỆT THÔNG MINH", css="""
 .yellow-btn {
@@ -23,7 +60,7 @@ with gr.Blocks(title="SAIFGuard - HỆ THỐNG KIỂM DUYỆT THÔNG MINH", css=
 }
 """) as demo:
     gr.Markdown("## 🛡️ SAIFGuard: HỆ THỐNG KIỂM DUYỆT THÔNG MINH")
-    
+
     with gr.Tab("📝 Kiểm duyệt Prompt"):
         with gr.Row():
             with gr.Column(scale=1):
@@ -33,7 +70,7 @@ with gr.Blocks(title="SAIFGuard - HỆ THỐNG KIỂM DUYỆT THÔNG MINH", css=
                 prompt_output = gr.Textbox(label="Kết quả GenAI")
                 prompt_button = gr.Button("Kiểm tra Prompt", elem_classes="yellow-btn")
         prompt_button.click(handle_prompt, inputs=prompt_input, outputs=[prompt_status, prompt_output])
-    
+
     with gr.Tab("🖼️ Kiểm duyệt Hình ảnh"):
         gr.Markdown("### 📷 Tải ảnh và kiểm tra từng tiêu chí")
 
@@ -43,7 +80,7 @@ with gr.Blocks(title="SAIFGuard - HỆ THỐNG KIỂM DUYỆT THÔNG MINH", css=
         with gr.Row():
             with gr.Column(scale=1):
                 nsfw_output = gr.Textbox(label="🔞 Kết quả kiểm duyệt ảnh nhạy cảm")
-                nsfw_button = gr.Button("Kiểm tra Ảnh Nhạy Cảm", elem_classes="yellow-btn")
+nsfw_button = gr.Button("Kiểm tra Ảnh Nhạy Cảm", elem_classes="yellow-btn")
             with gr.Column(scale=1):
                 violence_output = gr.Textbox(label="🧨 Kết quả kiểm duyệt ảnh bạo lực")
                 violence_button = gr.Button("Kiểm tra Ảnh Bạo Lực", elem_classes="yellow-btn")
@@ -59,8 +96,9 @@ with gr.Blocks(title="SAIFGuard - HỆ THỐNG KIỂM DUYỆT THÔNG MINH", css=
                 url_output = gr.Textbox(label="Kết quả kiểm duyệt URL")
                 url_button = gr.Button("Kiểm tra URL", elem_classes="yellow-btn")
         url_button.click(fn=check_url, inputs=url_input, outputs=url_output)
-    with gr.Tab("🤖 Giọng Nói"):
-        audio_input = gr.Audio( type="numpy", label="Thu âm giọng nói")
+
+    with gr.Tab("🎙️Giọng Nói"):
+        audio_input = gr.Audio(type="numpy", label="Thu âm giọng nói")
         btn2 = gr.Button("Chuyển đổi & Kiểm tra")
         stt, trans, gen2 = (
             gr.Textbox(label="Trạng thái kiểm duyệt"),