Spaces:
Running
Running
| import gradio as gr | |
| import asyncio | |
| import sys | |
| import os | |
| from pydub import AudioSegment | |
| sys.path.append('voice_chat') | |
| from voice_chat import respond | |
| def process_audio(audio): | |
| if audio is None: | |
| return None | |
| if isinstance(audio, tuple): # Gradio returns a tuple (path, sampling_rate) | |
| audio_path = audio[0] | |
| else: | |
| audio_path = audio | |
| # Ensure the file exists | |
| if not os.path.exists(audio_path): | |
| raise FileNotFoundError(f"Audio file not found: {audio_path}") | |
| # Convert audio to wav if it's not already | |
| if not audio_path.endswith('.wav'): | |
| audio = AudioSegment.from_file(audio_path) | |
| wav_path = audio_path.rsplit('.', 1)[0] + '.wav' | |
| audio.export(wav_path, format="wav") | |
| audio_path = wav_path | |
| return audio_path | |
| async def voice_chat(audio): | |
| try: | |
| processed_audio = process_audio(audio) | |
| if processed_audio is None: | |
| return None | |
| response_audio = await respond(processed_audio) | |
| return response_audio | |
| except Exception as e: | |
| print(f"Error in voice_chat: {str(e)}") | |
| return None | |
| interface = gr.Interface( | |
| fn=lambda audio: asyncio.run(voice_chat(audio)), | |
| inputs=gr.Audio(type="filepath"), | |
| outputs=gr.Audio(), | |
| title="Voice Chat with AI Assistant", | |
| description="Upload an audio file or record using the microphone, and the AI will respond with voice.", | |
| examples=None, | |
| theme="default" | |
| ) | |
| if __name__ == "__main__": | |
| interface.launch(debug=True) |