Update audio_predictions.py

updated audio transcription code

audio_predictions.py

@@ -20,12 +20,44 @@ class AudioTranslation:
             return converted_file_path
         return file_path
 
+    # def transcribe_audio(self, file_path):
+    #     url = "https://stt.umuganda.digital/transcribe/"
+    #     converted_file_path = self.convert_to_mp3(file_path)
+    #     with open(converted_file_path, 'rb') as file:
+    #         files = {'file': (file_path, file, 'audio/mpeg')}
+    #         print('transcribing audio')
+    #         try:
+    #             response = requests.post(url, files=files)
+    #             response.raise_for_status()
+    #             transcription = response.json()
+    #             # Remove punctuation
+    #             translator = str.maketrans('', '', string.punctuation)
+    #             cleaned_text = transcription['text'].translate(translator)
+    #             print('cleaned text')
+    #             print(cleaned_text)
+    #             return cleaned_text
+    #         except requests.exceptions.HTTPError as err:
+    #             print(f"HTTP error occurred: {err}")
+    #         except Exception as err:
+    #             print(f"An error occurred: {err}")
+    #     return None
     def transcribe_audio(self, file_path):
         url = "https://stt.umuganda.digital/transcribe/"
-        converted_file_path = self.convert_to_mp3(file_path)
+        
+        # Check if the file is an MP3; if not, convert it
+        if not file_path.lower().endswith('.mp3'):
+            print(f"Converting file to MP3: {file_path}")
+            converted_file_path = self.convert_to_mp3(file_path)
+        else:
+            converted_file_path = file_path
+        
+        if not os.path.exists(converted_file_path) or os.path.getsize(converted_file_path) == 0:
+            print(f"File does not exist or is empty: {converted_file_path}")
+            return None
+
         with open(converted_file_path, 'rb') as file:
-            files = {'file': (file_path, file, 'audio/mpeg')}
-            print('transcribing audio')
+            files = {'file': (os.path.basename(converted_file_path), file, 'audio/mpeg')}
+            print('Transcribing audio...')
             try:
                 response = requests.post(url, files=files)
                 response.raise_for_status()
@@ -33,14 +65,13 @@ class AudioTranslation:
                 # Remove punctuation
                 translator = str.maketrans('', '', string.punctuation)
                 cleaned_text = transcription['text'].translate(translator)
-                print('cleaned text')
-                print(cleaned_text)
+                print('Cleaned text:', cleaned_text)
                 return cleaned_text
-            except requests.exceptions.HTTPError as err:
-                print(f"HTTP error occurred: {err}")
-            except Exception as err:
-                print(f"An error occurred: {err}")
-        return None
+            except requests.exceptions.RequestException as e:
+                print(f"Request error occurred: {e}")
+                if hasattr(e, 'response') and e.response:
+                    print(f"Server response: {e.response.text}")
+                return None
 
     def get_translation(self, src, tgt, text):
         url = f"https://nmt-api.umuganda.digital/api/v1/translate/?src={src}&tgt={tgt}&text={text}"