Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -418,14 +418,17 @@ class WhisperBase(ABC):
                 title_line += f'\n{tabbed_space}Language:\t{info["lang"]} (probability {info["lang_prob"]}%)'
                 if params.is_translate: title_line += f'\n{tabbed_space}Translation:\t{info["transcription"]} (Handled by OpenAI Whisper)'
                 if translate_output: title_line += f'\n{tabbed_space}Translation:\t{info["translation"]} (Handled by Facebook NLLB)'
-                title_line += "#NEWLINE#"
-
+                
+                html_code = ""
                 for temp_file in files_to_download.items():
-                    print("test:")
-                    temp1 = str(temp_file[0].split("_")[0])
-                    temp2 = str((temp_file[1])["path"])
-                    print("temp1: "+temp1)
-                    print("temp2: "+temp2)
+                    dl_file = str(temp_file[0].split("_")[0])
+                    dl_type = str(temp_file[0].split("_")[1])
+                    dl_link = str((temp_file[1])["path"])
+                    if dl_file == file_name:
+                        if html_code != "": html_code+=" "
+                        html_code += "<a href=''"+dl_link+"'>"+dl_type+"</a>"
+
+                title_line += "#TAB#"+html_code+"#NEWLINE#"
                 
                 total_result += title_line+(((info["subtitle"].rstrip("\n")).replace("\t","#TAB#")).replace("\n","#NEWLINE#"))
                 total_time += info["time_for_task"]