update file treatment

app.py

@@ -82,6 +82,8 @@ from transformers import pipeline
 import soundfile as sf
 from scipy.io import wavfile
 
+import re
+
 ds = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
 sample = ds[0]["audio"]
 
@@ -1105,8 +1107,8 @@ def chathmi3(message, history2):
         # real_content = response[-1:]
         # print("real_content", real_content)
         try:
-            temp = response.split("(sandbox:/")[1] # (sandbox:/sample-20230805-0807.wav)
-            file_name = temp.split(")")[0]
+            # temp = response.split("(sandbox:/")[1] # (sandbox:/sample-20230805-0807.wav)
+            file_name = CheckFileinResp(response)
             print("file_name:", file_name)
             history2 = history2 + [(None, (file_name,))]
             Filename_Chatbot = file_name
@@ -1123,6 +1125,14 @@ def chathmi3(message, history2):
     print("-" * 20)
     print("-" * 20)
 
+def CheckFileinResp(response):
+    try:
+        pattern = r'sample-\w+\.wav'
+        result = re.findall(pattern, response)
+        return result[-1]
+    except Exception as e:
+        print("No wav found:", e)
+
 def chathmi4(message, history2):
     global last_request
     global Filename_Chatbot
@@ -1147,8 +1157,8 @@ def chathmi4(message, history2):
         # real_content = response[-1:]
         # print("real_content", real_content)
         try:
-            temp = response.split("(sandbox:/")[1] # (sandbox:/sample-20230805-0807.wav)
-            file_name = temp.split(")")[0]
+            # temp = response.split("(sandbox:/")[1] # (sandbox:/sample-20230805-0807.wav)
+            file_name = CheckFileinResp(response)
             print("file_name:", file_name)
             history2 = history2 + [(None, (file_name,))]
             Filename_Chatbot = file_name