Update app.py

app.py

@@ -3,6 +3,9 @@ from transformers import pipeline
 import random
 import string
 import datetime
+import librosa
+import soundfile as sf
+import numpy as np
 
 # Daftar produk dan kategori
 categories = {
@@ -15,12 +18,32 @@ categories = {
 # Inisialisasi model pengenalan suara
 speech_recognizer = pipeline("automatic-speech-recognition", model="facebook/wav2vec2-base-960h")
 
+# Fungsi untuk memproses audio
+def preprocess_audio(audio):
+    if audio is None:
+        return None
+    try:
+        # audio[0] adalah sampling rate, audio[1] adalah data audio
+        sr, y = audio
+        # Resample ke 16kHz
+        y = librosa.resample(y.astype(np.float32), orig_sr=sr, target_sr=16000)
+        # Simpan sementara ke file WAV
+        temp_file = "temp_audio.wav"
+        sf.write(temp_file, y, 16000)
+        return temp_file
+    except Exception as e:
+        return f"Error memproses audio: {str(e)}"
+
 # Fungsi untuk mengonversi suara ke teks
 def speech_to_text(audio):
-    if audio is None:
-        return ""
-    result = speech_recognizer(audio)
-    return result["text"]
+    processed_audio = preprocess_audio(audio)
+    if not isinstance(processed_audio, str) or processed_audio.startswith("Error"):
+        return processed_audio if processed_audio else ""
+    try:
+        result = speech_recognizer(processed_audio)
+        return result["text"]
+    except Exception as e:
+        return f"Error pengenalan suara: {str(e)}"
 
 # Fungsi untuk membuat dokumen LaTeX
 def create_latex(data, filename):
@@ -88,12 +111,16 @@ PT. Contoh Perusahaan
 def buat_penawaran(nama_prospek, alamat_prospek, jenis_prospek, produk_dipilih, jumlah_produk, harga_produk, diskon, tanggal, syarat, audio_input):
     # Proses input suara jika ada
     if audio_input:
-        audio_text = speech_to_text(audio_input).lower()
-        if "nama" in audio_text and not nama_prospek:
-            nama_prospek = audio_text.split("nama")[-1].strip()
-        if "alamat" in audio_text and not alamat_prospek:
-            alamat_prospek = audio_text.split("alamat")[-1].strip()
-        # Logika serupa dapat ditambahkan untuk field lain
+        audio_text = speech_to_text(audio_input)
+        if isinstance(audio_text, str) and not audio_text.startswith("Error"):
+            audio_text = audio_text.lower()
+            if "nama" in audio_text and not nama_prospek:
+                nama_prospek = audio_text.split("nama")[-1].strip()
+            if "alamat" in audio_text and not alamat_prospek;
+
+                alamat_prospek = audio_text.split("alamat")[-1].strip()
+        else:
+            return None, f"Peringatan: Gagal memproses input suara - {audio_text}"
 
     data = {
         'nama_prospek': nama_prospek if nama_prospek else "Prospek Tanpa Nama",