Update app.py

app.py

@@ -8,6 +8,23 @@ from genai_chat_ai import AI,create_chat_session
 API_URL = "https://api-inference.huggingface.co/models/asg2024/vits-ar-sa"
 api_key = os.environ.get("Id_mode_vits") 
 headers = {"Authorization": f"Bearer {api_key}"}
+from df.enhance import enhance, init_df, load_audio
+
+model_enhance, df_state, _ = init_df()
+def remove_noise_enhance(audio_bytes):
+    # save_audio("full_generation.wav", wav, 48000)
+  
+
+    # Load default model
+
+
+    # Download and open some audio file. You use your audio files here
+
+    audio, sr = load_audio(audio_bytes,df_state.sr())
+    # Denoise the audio
+    enhanced = enhance(model_enhance, df_state, audio)
+    return df_state.sr(),enhanced 
+
 
 def query(text):
   payload={"inputs": text}
@@ -30,6 +47,9 @@ def reverse_audio(text):
     text_answer=get_answer_ai(text)
     
     data = query(text_answer)
+    data_enhanc = remove_noise_enhance(data)
+
+    
     return data,text_answer#(16000, np.flipud(data))