Update app.py

app.py

@@ -44,7 +44,7 @@ def is_frame_different(frame1, frame2, threshold=0.9):
 @spaces.GPU
 def generate_journal_with_images(video_path, frame_interval=30):
     cap = cv2.VideoCapture(video_path)
-    journal_entries = {}
+    journal_entries = []
     saved_images = []
     frame_count = 0
     last_processed_frame = None
@@ -82,9 +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():
-                if activity not in journal_entries:
-                    journal_entries[activity] = []
-                journal_entries[activity].append((f"At {timestamp:.2f} seconds: {', '.join(objects[0])}", frame_filename))
+                journal_entries.append((f"At {timestamp:.2f} seconds: {', '.join(objects[0])}", frame_filename))
             
             last_processed_frame = frame  # Update the last processed frame
         
@@ -92,26 +90,20 @@ def generate_journal_with_images(video_path, frame_interval=30):
     
     cap.release()
     
-    # Create a formatted journal output
-    formatted_journal = []
-    for activity, entries in journal_entries.items():
-        formatted_journal.append(f"**{activity}:**")
-        for entry, image_path in entries:
-            formatted_journal.append((entry, image_path))
-    
-    return formatted_journal
+    return journal_entries
 
 # Gradio interface for uploading video and generating journal with images
 def display_journal_with_images(video):
-    journal_with_images = generate_journal_with_images(video)
+    journal_with_images = generate_journal_with_images(video, frame_interval=30)
     
     # Create the final display with text and images
     display_items = []
     for entry, image_path in journal_with_images:
-        display_items.append((entry, image_path))
+        display_items.append((entry, image_path))  # Add both the journal entry and the image path
     
     return display_items
 
+
 with gr.Blocks() as iface:
     video_input = gr.Video(label="Upload Video", height=300)
     output_gallery = gr.Gallery(label="Generated Daily Journal with Images")