Update translate.py

translate.py

@@ -25,8 +25,8 @@ class translation:
             self.lan_code='en'
         elif original_language =='German':
             self.lan_code='de'
-        elif original_language =='French':
-            self.lan_code='fr'
+        elif original_language =='Italian':
+            self.lan_code='it'
         elif original_language =='Spanish':
             self.lan_code='es'
         else:
@@ -37,8 +37,8 @@ class translation:
             self.tran_code='en'
         elif target_language =='German':
             self.tran_code='de'
-        elif target_language =='French':
-            self.tran_code='fr'
+        elif target_language =='Italian':
+            self.tran_code='it'
         elif target_language =='Spanish':
             self.tran_code='es'
         else:
@@ -51,6 +51,8 @@ class translation:
           audio.write_audiofile(audio_path)
           print("Audio extracted successfully!")
           return audio_path
+
+
   def transcribe_audio(self,audio_path):
       aai.settings.api_key = "c29eb650444a4ae4be6a787ebb15d5e2"
       config = aai.TranscriptionConfig(language_code=self.lan_code)
@@ -101,24 +103,18 @@ class translation:
     transcript_text = self.transcribe_audio(audio_path)
     translated_text = self.translate_text(transcript_text)
     translated_audio_path = self.generate_audio(translated_text)
-
-
-
-      
       #Generate video
-    # %cd /content/Wav2Lip
+    %cd /content/Wav2Lip
 
       #This is the detection box padding, if you see it doesnt sit quite right, just adjust the values a bit. Usually the bottom one is the biggest issue
-    # pad_top =  0
-    # pad_bottom =  15
-    # pad_left =  0
-    # pad_right =  0
-    # rescaleFactor =  1
-    # video_path_fix = f"'../{video_path}'"
-    # audio_path_fix = f"'../{translated_audio_path}'"
-    # !python inference.py --checkpoint_path 'checkpoints/wav2lip_gan.pth' --face $video_path_fix --audio $audio_path_fix --pads $pad_top $pad_bottom $pad_left $pad_right --resize_factor $rescaleFactor --nosmooth --outfile '/content/output_video.mp4'
-
-
-
-
+    pad_top =  0
+    pad_bottom =  15
+    pad_left =  0
+    pad_right =  0
+    rescaleFactor =  1
+    video_path_fix = f"'../{self.video_path}'"
+    audio_path_fix = f"'../{translated_audio_path}'"
+    !python inference.py --checkpoint_path 'checkpoints/wav2lip_gan.pth' --face $video_path_fix --audio $audio_path_fix --pads $pad_top $pad_bottom $pad_left $pad_right --resize_factor $rescaleFactor --nosmooth --outfile '/content/output_video.mp4'
+    return '/content/output_video.mp4'  # Return the path to the translated video file
+    # return '/content/output_video.mp4', open('/content/output_video.mp4', 'rb')  # Return the path and file object of the translated video file