Update app.py

app.py

@@ -1,33 +1,46 @@
 import gradio as gr
 from df.enhance import enhance, init_df, load_audio, save_audio
+from df.io import load_audio as load_audio_df # Name clash bachane ke liye
 import torch
+import numpy as np
+import tempfile
+import os
 
-# Model load karein (Sirf ek baar)
+# 1. Model Load karein
+# config_allow_defaults=True se minor errors ignore ho jate hain
 model, df_state, _ = init_df()
 
 def enhance_audio(audio_path):
     if audio_path is None:
         return None
     
-    # Audio load aur process karein
-    audio, _ = load_audio(audio_path, sr=df_state.sr())
-    
-    # Noise hatana (Inference)
-    enhanced_audio = enhance(model, df_state, audio)
-    
-    # Save output
-    output_path = "enhanced_voice.wav"
-    save_audio(output_path, enhanced_audio, df_state.sr())
-    
-    return output_path
+    try:
+        # 2. Audio Load karein
+        # df_state.sr() model ka required sample rate (48kHz) deta hai
+        audio, _ = load_audio_df(audio_path, sr=df_state.sr())
+        
+        # 3. Enhance (Noise Reduction)
+        # Model CPU par hi run karega kyunki init_df default wahi karta hai
+        enhanced_audio = enhance(model, df_state, audio)
+        
+        # 4. Output Save karein (Temp file banana safe rehta hai)
+        # Hugging Face par static naam ("enhanced.wav") kabhi kabhi permission error deta hai
+        output_file = tempfile.NamedTemporaryFile(suffix=".wav", delete=False).name
+        save_audio(output_file, enhanced_audio, df_state.sr())
+        
+        return output_file
+
+    except Exception as e:
+        print(f"Error aa gaya: {e}")
+        return None
 
-# Interface
+# Interface Setup
 iface = gr.Interface(
     fn=enhance_audio,
     inputs=gr.Audio(type="filepath", label="Upload Noisy Audio"),
-    outputs=gr.Audio(type="filepath", label="Clean Voice"),
-    title="AI Voice Enhancer (DeepFilterNet)",
-    description="CPU par chalne wala fast voice cleaner. Background noise hatayein."
+    outputs=gr.Audio(type="filepath", label="Clean Voice Output"),
+    title="AI Voice Cleaner (DeepFilterNet)",
+    description="Yeh tool background noise (pankha, traffic) hatata hai using CPU."
 )
 
 iface.launch()