Adjoumani commited on
Commit
413559b
·
verified ·
1 Parent(s): 662bd54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +34 -20
app.py CHANGED
@@ -863,28 +863,42 @@ def handle_youtube_input():
863
  """Gestion des entrées YouTube"""
864
 
865
  youtube_url = st.text_input("URL YouTube")
866
-
867
  if youtube_url and st.button("Analyser"):
868
- if not re.match(r'^https?://(?:www\.)?youtube\.com/watch\?v=[\w-]+|^https?://youtu\.be/[\w-]+', youtube_url):
869
- st.error("URL YouTube invalide")
870
- else:
871
- video_processor = VideoProcessor()
872
- video_id = video_processor.extract_video_id(youtube_url)
873
- if video_id:
874
- st.video(youtube_url)
875
-
876
- with st.spinner("Récupération du contenu de la vidéo..."):
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
877
  # Essayer d'abord d'obtenir la transcription
878
- transcription = video_processor.get_youtube_transcription(video_id)
879
-
880
- if transcription:
881
- st.success("Transcription YouTube trouvée!")
882
- process_and_display_results(None, None, transcription)
883
- else:
884
- st.info("Pas de transcription disponible. Extraction de l'audio...")
885
- video_processor.load_cookies()
886
- audio_path = video_processor.download_youtube_audio(youtube_url)
887
- process_and_display_results(audio_path)
888
 
889
 
890
  def process_and_display_results(file_path=None, audio_bytes=None, transcription=None):
 
863
  """Gestion des entrées YouTube"""
864
 
865
  youtube_url = st.text_input("URL YouTube")
 
866
  if youtube_url and st.button("Analyser"):
867
+ video_processor = VideoProcessor()
868
+ video_id = video_processor.extract_video_id(youtube_url)
869
+
870
+ if video_id:
871
+ st.video(youtube_url)
872
+ with st.spinner("Traitement de la vidéo..."):
873
+ transcription = video_processor.get_youtube_transcription(video_id)
874
+ if transcription:
875
+ process_and_display_results(None, None, transcription)
876
+ else:
877
+ video_processor.load_cookies()
878
+ audio_path = video_processor.download_youtube_audio(youtube_url)
879
+ process_and_display_results(audio_path)
880
+
881
+ #if youtube_url and st.button("Analyser"):
882
+ # if not re.match(r'^https?://(?:www\.)?youtube\.com/watch\?v=[\w-]+|^https?://youtu\.be/[\w-]+', youtube_url):
883
+ # st.error("URL YouTube invalide")
884
+ # else:
885
+ # video_processor = VideoProcessor()
886
+ # video_id = video_processor.extract_video_id(youtube_url)
887
+ # if video_id:
888
+ # st.video(youtube_url)
889
+
890
+ # with st.spinner("Récupération du contenu de la vidéo..."):
891
  # Essayer d'abord d'obtenir la transcription
892
+ # transcription = video_processor.get_youtube_transcription(video_id)
893
+
894
+ # if transcription:
895
+ # st.success("Transcription YouTube trouvée!")
896
+ # process_and_display_results(None, None, transcription)
897
+ # else:
898
+ # st.info("Pas de transcription disponible. Extraction de l'audio...")
899
+ # video_processor.load_cookies()
900
+ # audio_path = video_processor.download_youtube_audio(youtube_url)
901
+ # process_and_display_results(audio_path)
902
 
903
 
904
  def process_and_display_results(file_path=None, audio_bytes=None, transcription=None):