Update app.py

app.py

@@ -45,7 +45,7 @@ def is_frame_different(frame1, frame2, threshold=0.9):
 def generate_journal_with_images(video_path, frame_interval=30):
     cap = cv2.VideoCapture(video_path)
     journal_entries = []
-    saved_images = []
+    image_paths = []
     frame_count = 0
     last_processed_frame = None
     output_folder = "detected_frames"
@@ -69,7 +69,7 @@ def generate_journal_with_images(video_path, frame_interval=30):
             # Save the annotated image
             frame_filename = os.path.join(output_folder, f"frame_{frame_count}.jpg")
             cv2.imwrite(frame_filename, annotated_frame[:, :, ::-1])  # Convert back to BGR for saving
-            saved_images.append(frame_filename)
+            image_paths.append(frame_filename)
             
             # Extract labels (class indices) and map them to class names
             detected_objects = [model.names[int(box.cls)] for box in results[0].boxes]  # Access the first result
@@ -82,7 +82,7 @@ def generate_journal_with_images(video_path, frame_interval=30):
             
             # Store the activities with their timestamp
             for activity, objects in activity_summary.items():
-                journal_entries.append((f"At {timestamp:.2f} seconds: {', '.join(objects[0])}", frame_filename))
+                journal_entries.append(f"At {timestamp:.2f} seconds: {', '.join(objects[0])}")
             
             last_processed_frame = frame  # Update the last processed frame
         
@@ -90,17 +90,21 @@ def generate_journal_with_images(video_path, frame_interval=30):
     
     cap.release()
     
-    return journal_entries
+    # Debug print to verify the return values
+    print(f"journal_entries: {journal_entries}")
+    print(f"image_paths: {image_paths}")
+    
+    return journal_entries, image_paths
 
 
 def display_journal_with_images(video):
     journal_entries, image_paths = generate_journal_with_images(video, frame_interval=30)
     
-    # Return journal text and list of images separately
+
     journal_text = "\n".join(journal_entries)
     return journal_text, image_paths
 
-# Define Gradio Blocks for custom display
+
 with gr.Blocks() as iface:
     video_input = gr.Video(label="Upload Video", height=300)  
     journal_output = gr.Textbox(label="Generated Daily Journal", lines=10)