bugs fixed

Former-commit-id: be62aca589877af3c96bd5a534feba24e88a46c5

.gitignore

@@ -5,3 +5,5 @@
 test.py
 test.srt
 test.txt
+log_*.csv
+log.csv

SRT.py

@@ -111,7 +111,7 @@ class SRT_script():
         
     
 
-    def set_translation(self, translate:str, id_range:tuple, model,vid_link=None):
+    def set_translation(self, translate:str, id_range:tuple, model, video_name, video_link=None):
         start_seg_id = id_range[0]
         end_seg_id = id_range[1]
         
@@ -159,12 +159,21 @@ class SRT_script():
             if len(lines) < (end_seg_id - start_seg_id + 1):
                 print("Failed Solving unmatched lines, Manually parse needed")
             
-            log_file = "log.csv"
-            log_exist = os.path.exists(log_file)
-            with open(log_file,"a") as log:
-                if not log_exist:
-                    log.write("range_of_text,iterations_solving,solved,file_length,video_link")
-                log.write(range+','+count+','+solved+','+len(self.segments)+','+vid_link)
+            
+            if video_link:
+                log_file = "./logs/log_link.csv"
+                log_exist = os.path.exists(log_file)
+                with open(log_file,"a") as log:
+                    if not log_exist:
+                        log.write("range_of_text,iterations_solving,solved,file_length,video_link" + "\n")
+                    log.write(str(id_range)+','+str(count)+','+str(solved)+','+str(len(self.segments))+','+video_link + "\n")
+            else:
+                log_file = "./logs/log_name.csv"
+                log_exist = os.path.exists(log_file)
+                with open(log_file,"a") as log:
+                    if not log_exist:
+                        log.write("range_of_text,iterations_solving,solved,file_length,video_name" + "\n")
+                    log.write(str(id_range)+','+str(count)+','+str(solved)+','+str(len(self.segments))+','+video_name + "\n")
             
             print(lines)
             #print(id_range)

pipeline.py

@@ -277,7 +277,7 @@ for sentence, range in tqdm(zip(script_arr, range_arr)):
             time.sleep(30)
             flag = True
     # add read-time output back and modify the post-processing by using one batch as an unit.
-    srt.set_translation(translate, range, model_name)
+    srt.set_translation(translate, range, model_name, VIDEO_NAME, args.link)
 
     # srt.realtime_bilingual_write_srt(f"{RESULT_PATH}/{VIDEO_NAME}/{VIDEO_NAME}_bi.srt",range, add_length,segidx)