Update LLMwithvoice.py

LLMwithvoice.py

@@ -33,8 +33,17 @@ def generate_speech(prompt, description):
 
     generation = model.generate(input_ids=input_ids, prompt_input_ids=prompt_input_ids).to(torch.float32)
     audio_arr = generation.cpu().numpy().squeeze()
-    audio_path = "/mnt/data/parler_tts_out.wav"
-    sf.write(audio_path, audio_arr, model.config.sampling_rate)
+
+    # Construct the audio path dynamically
+    audio_filename = "parler_tts_out.wav"
+    audio_path = f"/mnt/data/{audio_filename}"
+
+    try:
+        sf.write(audio_path, audio_arr, model.config.sampling_rate)
+    except Exception as e:
+        print(f"Error writing audio file: {e}")
+        # Handle the error, raise or log it, or provide an alternative approach
+
     return audio_path
 
 def chat_with_roberta(api_token, question, context):
@@ -51,10 +60,4 @@ def chat_with_roberta(api_token, question, context):
         try:
             return response['answer']
         except (IndexError, KeyError):
-            return f"Unexpected response structure: {response}"
-
-def gradio_interface(api_token, context, question):
-    answer = chat_with_roberta(api_token, question, context)
-    description = "A female speaker with a slightly low-pitched voice delivers her words quite expressively, in a very confined sounding environment with clear audio quality. She speaks very fast."
-    audio_path = generate_speech(answer, description)
-    return answer, audio_path
+            return f"Unexpected response structure: {response}"