Update app.py

app.py

@@ -185,24 +185,35 @@ from gradio_multimodalchatbot import MultimodalChatbot
 from gradio.data_classes import FileData
 import tempfile
 import soundfile as sf
+from gradio_client import Client
 def add_message(history, message):
    
     if message["text"] is not None:
-        sr,response_audio = genrate_speech(message["text"],'asg2024/vits-ar-sa-huba')
-        with tempfile.TemporaryFile() as temp_file:
-            filename='temp'+str(temp_file.name)+'.wav'
-            sf.write(filename,response_audio, sr,format='WAV')
-            user_msg1 = {"text": message["text"],
-                "files": [{"file":FileData(path=filename)}]}
+        # sr,response_audio = genrate_speech(message["text"],'asg2024/vits-ar-sa-huba')
+        # with tempfile.TemporaryFile() as temp_file:
+        #     filename='temp'+str(temp_file.name)+'.wav'
+        #     sf.write(filename,response_audio, sr,format='WAV')
+        #     user_msg1 = {"text": message["text"],
+        #         "files": [{"file":FileData(path=filename)}]}
         
         
-        txt_ai=get_answer_ai(message["text"] )
-        sr,response_audio = genrate_speech(txt_ai,'asg2024/vits-ar-sa-huba')
-        with tempfile.TemporaryFile() as temp_file:
-            filename='temp'+str(temp_file.name)+'.wav'
-            sf.write(filename,response_audio, sr,format='WAV')
-            bot_msg1 = {"text": txt_ai,
-                "files": [{"file":FileData(path=filename)}]}
+        # txt_ai=get_answer_ai(message["text"] )
+        # sr,response_audio = genrate_speech(txt_ai,'asg2024/vits-ar-sa-huba')
+        # with tempfile.TemporaryFile() as temp_file:
+        #     filename='temp'+str(temp_file.name)+'.wav'
+        #     sf.write(filename,response_audio, sr,format='WAV')
+        #     bot_msg1 = {"text": txt_ai,
+        #         "files": [{"file":FileData(path=filename)}]}
+        client = Client("asg2024/wasm-speeker-sa")
+        result = client.predict(
+                text="سلام عليكم",
+                model_choice="asg2024/vits-ar-sa-huba",
+                api_name="/process_audio"
+        )
+        user_msg1 = {"text": message["text"],
+                "files": [{"file":FileData(path=result[0])}]}
+        bot_msg1 = {"text": result[2],
+                "files": [{"file":FileData(path=result[1])}]}
         history.append([user_msg1, bot_msg1])
         
         # history.append((gr.Audio(response_audio,scale=1,streaming=True),None))