yasserrmd commited on
Commit
d5b8a83
1 Parent(s): 47f97bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -12
app.py CHANGED
@@ -92,23 +92,23 @@ def generate_journal_with_images(video_path, frame_interval=30):
92
 
93
  return journal_entries
94
 
95
- # Gradio interface for uploading video and generating journal with images
96
  def display_journal_with_images(video):
97
- journal_with_images = generate_journal_with_images(video, frame_interval=30)
98
-
99
- # Create the final display with text and images
100
- display_items = []
101
- for entry, image_path in journal_with_images:
102
- display_items.append((entry, image_path)) # Add both the journal entry and the image path
103
 
104
- return display_items
105
-
 
106
 
 
107
  with gr.Blocks() as iface:
108
- video_input = gr.Video(label="Upload Video", height=300)
109
- output_gallery = gr.Gallery(label="Generated Daily Journal with Images")
 
110
  run_button = gr.Button("Generate Journal")
111
 
112
- run_button.click(fn=display_journal_with_images, inputs=video_input, outputs=output_gallery)
 
 
113
 
114
  iface.launch()
 
92
 
93
  return journal_entries
94
 
95
+
96
  def display_journal_with_images(video):
97
+ journal_entries, image_paths = generate_journal_with_images(video, frame_interval=30)
 
 
 
 
 
98
 
99
+ # Return journal text and list of images separately
100
+ journal_text = "\n".join(journal_entries)
101
+ return journal_text, image_paths
102
 
103
+ # Define Gradio Blocks for custom display
104
  with gr.Blocks() as iface:
105
+ video_input = gr.Video(label="Upload Video", height=300)
106
+ journal_output = gr.Textbox(label="Generated Daily Journal", lines=10)
107
+ image_gallery = gr.Gallery(label="Annotated Frames").style(grid=[2], height="auto")
108
  run_button = gr.Button("Generate Journal")
109
 
110
+ run_button.click(fn=display_journal_with_images, inputs=video_input, outputs=[journal_output, image_gallery])
111
+
112
+ iface.launch()
113
 
114
  iface.launch()