PriyankaSatish commited on
Commit
b29f4bf
·
verified ·
1 Parent(s): 0a9d475

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -18
app.py CHANGED
@@ -262,7 +262,7 @@ with col1:
262
 
263
  st.title("Insightly Video")
264
  stream_url = st.text_input("Enter the live stream URL (YouTube, Twitch, etc.):")
265
- keyword = st.text_input("Enter a keyword to filter the frames (optional):")
266
  extract_frames_button = st.button("Extract Frames")
267
  uploaded_video = st.file_uploader("Or upload a video file (MP4):", type=["mp4"])
268
  prompt1 = "keyword is " + st.text_input("Enter a keyword for analysis:")
@@ -296,7 +296,7 @@ with col1:
296
  else:
297
  st.write("Please upload an MP3 file to get started.")
298
 
299
- if extract_frames_button and stream_url and keyword:
300
  # Execute FFmpeg command to extract frames
301
 
302
  # Check if URL is provided
@@ -341,17 +341,6 @@ with col1:
341
  extracted_text = extract_text_from_base64_frame(frame_base64)
342
  frame_texts[idx] = extracted_text
343
 
344
- if not keyword or keyword.lower() in extracted_text.lower():
345
- col1, col2 = st.columns([3, 2])
346
- with col1:
347
- frame_bytes = base64.b64decode(frame_base64)
348
- st.image(Image.open(BytesIO(frame_bytes)), caption=f'Frame {idx + 1}', use_column_width=True)
349
- with col2:
350
- st.write(f"Extracted Text: {extracted_text}")
351
- if keyword:
352
- st.write(f"Displaying frames containing the keyword '{keyword}'.")
353
- else:
354
- st.write("Displaying all extracted frames.")
355
  # Use Streamlit columns for side-by-side display (1 column for image, 1 for text)
356
  # col1, col2 = st.columns([3, 2])
357
  # with col1:
@@ -427,11 +416,6 @@ with col1:
427
  else:
428
  st.write("Failed to generate overall description.")
429
 
430
- if keyword:
431
- st.write(f"Displaying frames containing the keyword '{keyword}'.")
432
- else:
433
- st.write("Displaying all extracted frames.")
434
-
435
  elif uploaded_video is not None and extract_frames_button:
436
  with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmpfile:
437
  tmpfile.write(uploaded_video.getvalue())
 
262
 
263
  st.title("Insightly Video")
264
  stream_url = st.text_input("Enter the live stream URL (YouTube, Twitch, etc.):")
265
+ #keyword = st.text_input("Enter a keyword to filter the frames (optional):")
266
  extract_frames_button = st.button("Extract Frames")
267
  uploaded_video = st.file_uploader("Or upload a video file (MP4):", type=["mp4"])
268
  prompt1 = "keyword is " + st.text_input("Enter a keyword for analysis:")
 
296
  else:
297
  st.write("Please upload an MP3 file to get started.")
298
 
299
+ if extract_frames_button and stream_url:
300
  # Execute FFmpeg command to extract frames
301
 
302
  # Check if URL is provided
 
341
  extracted_text = extract_text_from_base64_frame(frame_base64)
342
  frame_texts[idx] = extracted_text
343
 
 
 
 
 
 
 
 
 
 
 
 
344
  # Use Streamlit columns for side-by-side display (1 column for image, 1 for text)
345
  # col1, col2 = st.columns([3, 2])
346
  # with col1:
 
416
  else:
417
  st.write("Failed to generate overall description.")
418
 
 
 
 
 
 
419
  elif uploaded_video is not None and extract_frames_button:
420
  with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmpfile:
421
  tmpfile.write(uploaded_video.getvalue())