Update translate.py

translate.py

@@ -28,13 +28,13 @@ class Translate:
         return audio_path
 
     def gender_detection(self):
-        # gender_model_url = "https://salman11223-gender-detection.hf.space/--replicas/wml9f/"
-        # gender_client = Client(gender_model_url)
-        # gender = gender_client.predict(
-        #     'audio.wav', api_name="/predict"
-        # )
-        # print(gender)
-        return "male"
+        gender_model_url = "https://salman11223-gender-detection.hf.space/--replicas/wml9f/"
+        gender_client = Client(gender_model_url)
+        gender = gender_client.predict(
+            'audio.wav', api_name="/predict"
+        )
+        print(gender)
+        return gender
 
 
 
@@ -208,3 +208,12 @@ class Translate:
         # Write the video with translated audio to the output file
         video.write_videofile(output_video_path, codec="libx264", audio_codec="aac")
 
+        # Clean up temporary files
+        self.cleanup_temp_files()
+
+    def cleanup_temp_files(self):
+        temp_files = ["audio.wav", "t.txt", "transcript.srt","translated_audio.wav","translated_file.srt"] + [f"translated_audio_{i}.wav" for i in range(1, 100)]  # Adjust the range accordingly
+        for file in temp_files:
+            if os.path.exists(file):
+                os.remove(file)
+                print(f"Deleted {file}")