Update app.py

app.py

@@ -300,6 +300,32 @@ class VideoQA:
         # Use moviepy to cut both video and audio
         video_clip = VideoFileClip(input_file).subclip(start_time, end_time)
         video_clip.write_videofile(output_file, codec='libx264', audio_codec='aac', temp_audiofile='temp-audio.m4a', remove_temp=True)
+        
+    def sentence_timestamp(self,data):
+      result_sentences = []
+
+      current_sentence = ""
+      current_start_timestamp = ""
+
+      for entry in data:
+          text = entry['text']
+          start_timestamp = entry['start_timestamp']
+          end_timestamp = entry['end_timestamp']
+
+          # If the current sentence is empty, update start timestamp
+          if not current_sentence:
+              current_start_timestamp = start_timestamp
+
+          # Concatenate sentences until a sentence ends with a full stop
+          current_sentence += " " + text
+          if text.endswith('.'):
+              result_sentences.append({
+                  'start_timestamp': current_start_timestamp,
+                  'end_timestamp': end_timestamp,
+                  'text': current_sentence.strip()
+              })
+              current_sentence = ""
+      return result_sentences
 
     def main(self,input_video_path,question):
 
@@ -309,7 +335,8 @@ class VideoQA:
 
       subrip_text  = self.results_to_sentence_srt(subtitle)
       result = self.extract_timestamps_and_text(subrip_text)
-      start_time,end_time = self.start_end_timestamp(result,answer)
+      sent = self.sentence_timestamp(result)  
+      start_time,end_time = self.start_end_timestamp(sent,answer)
       output_video_path = 'output_video.mp4'
 
       self.cut_video(input_video_path, output_video_path, start_time, end_time)