Update app.py

app.py

@@ -10,6 +10,7 @@ import logging
 import gradio as gr
 import tempfile
 import os
+import shutil
 
 class VideoRAGTool:
     def __init__(self, model_name: str = "openai/clip-vit-base-patch32"):
@@ -61,6 +62,9 @@ class VideoRAGTool:
             
         cap.release()
         
+        if not features_list:
+            raise ValueError("No frames were processed from the video")
+            
         features_array = np.vstack(features_list)
         self.frame_index = faiss.IndexFlatL2(features_array.shape[1])
         self.frame_index.add(features_array)
@@ -92,25 +96,34 @@ class VideoRAGApp:
         self.rag_tool = VideoRAGTool()
         self.current_video_path = None
         self.processed = False
+        self.temp_dir = tempfile.mkdtemp()
+
+    def __del__(self):
+        """Cleanup temporary files on deletion"""
+        if hasattr(self, 'temp_dir') and os.path.exists(self.temp_dir):
+            shutil.rmtree(self.temp_dir, ignore_errors=True)
 
     def process_video(self, video_file):
         """Process uploaded video and return status message"""
-        if video_file is None:
-            return "Please upload a video first."
-        
-        temp_dir = tempfile.mkdtemp()
-        temp_path = os.path.join(temp_dir, "uploaded_video.mp4")
-        
-        with open(temp_path, "wb") as f:
-            f.write(video_file)
-        
-        self.current_video_path = temp_path
-        
         try:
-            self.rag_tool.process_video(temp_path)
+            if video_file is None:
+                return "Please upload a video first."
+
+            # video_file is now a file path provided by Gradio
+            video_path = video_file.name
+            
+            # Create a copy in our temp directory
+            temp_video_path = os.path.join(self.temp_dir, "current_video.mp4")
+            shutil.copy2(video_path, temp_video_path)
+            
+            self.current_video_path = temp_video_path
+            
+            self.rag_tool.process_video(self.current_video_path)
             self.processed = True
             return "Video processed successfully! You can now ask questions about the video."
+            
         except Exception as e:
+            self.processed = False
             return f"Error processing video: {str(e)}"
 
     def query_video(self, query_text):