added check of temp files

app.py

@@ -1,14 +1,18 @@
 import gradio as gr
 from textblob import TextBlob
-import utils
+from utils import *
 import tempfile
 
 
 def answer_video_question(query : str, url : str, file : bytes) -> dict:
     # Either `file` or `url` must be provided
+
+    output_path='/tmp/video'
+    if not os.path.exists(output_path):
+        os.mkdir(output_path)
     if file is not None:
         with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_vid:
-            temp_vid.write(file.read())
+            temp_vid.write(file)
             temp_video_path = temp_vid.name
 
         # Output frame folder
@@ -31,7 +35,7 @@ demo = gr.Interface(
     inputs=[
         gr.Textbox(placeholder="Enter Query about the movie", label="Query"),
         gr.Textbox(placeholder="Paste the URL of the movie", label="Movie URL (optional)"),
-        gr.File(label="Upload Movie File (optional)")
+        gr.File(label="Upload Movie File (optional)",type='binary')
     ],
     outputs=gr.JSON(),
     title="Text Sentiment Analysis",

check_image_extraction.py

@@ -0,0 +1,42 @@
+import gradio as gr
+import os
+
+# Get all images in the folder, sorted by filename
+image_dir = "/tmp/video/frames"
+image_paths = sorted([
+    os.path.join(image_dir, f)
+    for f in os.listdir(image_dir)
+    if f.endswith(".jpg")
+])
+
+def get_image(index):
+    # Make sure the index is within bounds
+    if 0 <= index < len(image_paths):
+        return image_paths[index], index
+    return None, index  # No change if out of bounds
+
+def next_image(current_index):
+    next_index = current_index + 1
+    return get_image(next_index)
+
+def previous_image(current_index):
+    prev_index = current_index - 1
+    return get_image(prev_index)
+
+with gr.Blocks() as demo:
+    gr.Markdown("### Browse Keyframes")
+    
+    image = gr.Image()
+    index_state = gr.State(0)  # Track current index
+    
+    with gr.Row():
+        prev_btn = gr.Button("Previous")
+        next_btn = gr.Button("Next")
+
+    next_btn.click(fn=next_image, inputs=index_state, outputs=[image, index_state])
+    prev_btn.click(fn=previous_image, inputs=index_state, outputs=[image, index_state])
+    
+    # Initialize with first image
+    demo.load(fn=get_image, inputs=index_state, outputs=[image, index_state])
+
+demo.launch()

utils.py

@@ -14,12 +14,12 @@ def download_video(url):
     """
 
     # instanciate output path
-    output_path='./cache'
+    output_path='/tmp'
     if not os.path.exists(output_path):
         os.mkdir(output_path)
 
     # get cookies
-    export_cookies_path = "./cache/exported_cookies.txt"
+    export_cookies_path = "/tmp/exported_cookies.txt"
     os.makedirs(os.path.dirname(export_cookies_path), exist_ok=True)
     try:
         ydl_opts_export_cookies = {
@@ -72,7 +72,7 @@ def download_video(url):
 
 
 
-def is_significantly_different(img1, img2, threshold=0.4):
+def is_significantly_different(img1, img2, threshold=0.1):
     """Check if two images are significantly different using SSIM.
     Args:
         img1 (numpy.ndarray): First image.
@@ -104,7 +104,7 @@ def extract_keyframes(video_path, diff_threshold=0.4):
         print("Failed to read video.")
         return
     
-    output_path='./cache/video/frames'
+    output_path='/tmp/video/frames'
     if not os.path.exists(output_path):
         os.mkdir(output_path)
 
@@ -115,14 +115,12 @@ def extract_keyframes(video_path, diff_threshold=0.4):
         frame_id += 1
 
         if is_significantly_different(prev_frame, frame, threshold=diff_threshold):
-            filename = os.path.join("./cache/video/frames/",f"keyframe_{saved_id:04d}.jpg")
+            filename = os.path.join("/tmp/video/frames/",f"keyframe_{saved_id:04d}.jpg")
             cv2.imwrite(filename, frame)
             prev_frame = frame
             saved_id += 1
+            print(f"frame{saved_id} saved")
 
     cap.release()
     print(f"Extracted {saved_id} key frames.")
     return "success"
-
-# Example usage
-extract_keyframes(video_path)