Update app.py

app.py

@@ -11,6 +11,7 @@ from diarization import diarize_segments
 from privacy import MedicalPrivacyProcessor
 from config import settings
 from typing import Tuple, Dict
+from pydub import AudioSegment
 
 # HuggingFace token'ını ayarla
 HF_TOKEN = os.getenv("HF_TOKEN", "")
@@ -80,35 +81,46 @@ def process_audio(
         Tuple[str, Dict]: (Transkripsiyon metni, JSON sonuç)
     """
     if not audio_path:
-        return "Lütfen bir ses dosyası yükleyin.", {}
+        return "Lütfen bir ses dosyası yükleyin.", {"error": "Dosya yüklenmedi"}
         
     try:
-        # Ses dosyasını yükle
-        audio = Audio.from_file(audio_path)
+        # Ses dosyasını WAV formatına çevir
+        base, ext = os.path.splitext(audio_path)
+        wav_path = base + ".wav"
+        audio = AudioSegment.from_file(audio_path)
         
         # Ses iyileştirme
         if do_enhance:
             audio = enhance_audio(audio)
+            audio.export(wav_path, format="wav")
         
         # Konuşmacı ayrımı
         if do_diarize:
-            diarization = diarize_speakers(audio)
+            diarization = diarize_speakers(wav_path)
         else:
             diarization = None
             
         # Transkripsiyon
-        result = transcribe_audio(audio, diarization)
+        result = transcribe_audio(wav_path, diarization)
         
         # Anonimleştirme
         if do_anonymize:
             result = anonymize_personal_info(result)
             
-        return result["text"], result
+        # JSON çıktısını hazırla
+        json_result = {
+            "transcript": result["text"],
+            "diarization": diarization,
+            "enhanced_audio": wav_path if do_enhance else None,
+            "anonymized": do_anonymize
+        }
+            
+        return result["text"], json_result
         
     except Exception as e:
         error_msg = f"İşlem sırasında bir hata oluştu: {str(e)}"
         print(error_msg)
-        return error_msg, {}
+        return error_msg, {"error": str(e)}
 
 # Ana arayüz
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", secondary_hue="blue"), css=css) as demo: