update upload/clear logic

app.py

@@ -87,8 +87,6 @@ def check_exist_before_upsert(index, video_path):
 def chat(message, history):
     image_input_path = None
 
-    # print(message['files'])
-
     video_name, video_input_path = None, None
     if len(message['files']) > 0:
         assert len(message['files']) == 1
@@ -105,6 +103,12 @@ def chat(message, history):
         history = []
 
     if video_name is not None:
+        global global_video_name
+        if global_video_name is not None:
+            history.append((None, "A video has already been uploaded. Please ask questions about the video or click `Clear` to work with another video."))
+            yield history
+            return
+
         # Check metadata
         history.append((None, f"✅ Video uploaded succesfully! Your video's title is {video_name}..."))
         yield history
@@ -151,14 +155,15 @@ def chat(message, history):
             embed.indexing(INDEX, MODEL_STACK, metadatas_path)
 
         # summarizing video 
-        video_summary = rag.summarize_video(metadatas_path)
-        with open(os.path.join(output_folder_path, "summary.txt"), "w") as f:
-            f.write(video_summary)
+        if not os.path.exists(os.path.join(output_folder_path, "summary.txt")):
+            video_summary = rag.summarize_video(metadatas_path)
+            with open(os.path.join(output_folder_path, "summary.txt"), "w") as f:
+                f.write(video_summary)
 
-        history.append((None, f"Video processing complete! You can now ask me questions about the video {video_name}!"))
+        history.append((None, f"Video processing complete! You can now ask me questions about the video {video_name}! Please remove the video from your input."))
         yield history        
 
-        global global_video_name 
+        # global global_video_name 
         global_video_name = video_name
     else:
         history.append((message, None))
@@ -169,7 +174,7 @@ def chat(message, history):
             yield history
             return
 
-        output_folder_path = f"{UPLOAD_FOLDER}/{video_name}"
+        output_folder_path = f"{UPLOAD_FOLDER}/{global_video_name}"
         metadatas_path = os.path.join(output_folder_path, 'metadatas.json') 
 
         video_summary = ''
@@ -206,8 +211,8 @@ def main():
                 clear = gr.Button("Clear")  # Clear button
                 # Clear chat history when clear button is clicked
                 clear.click(clear_chat, [], chatbot)
-                global video_name 
-                video_name = None
+                global global_video_name 
+                global_video_name = None
 
     demo.launch()
 

video_utils.py

@@ -39,9 +39,7 @@ def transcribe_video(path_to_extracted_audio_file, output_folder, whisper_model=
     results = whisper_model.transcribe(path_to_extracted_audio_file, **options)
 
     vtt = getSubs(results["segments"], "vtt")
-    # path to save generated transcript of video1
-    video_name = os.path.basename(path_to_video).replace('.mp4', '')
-    path_to_generated_transcript = os.path.join(output_folder, f'{video_name}.vtt')
+    path_to_generated_transcript = os.path.join(output_folder, 'transcipt.vtt')
     
     # write transcription to file
     with open(path_to_generated_transcript, 'w') as f: