Update app.py

app.py

@@ -771,7 +771,7 @@ def enhance_main():
           
         ### **Comment utiliser l'application ?**  
         1. **Documents** 📄 :  
-           - **Formats supportés** : `.pdf`, `.docx`, `.pptx`  
+           - **Formats supportés** : `.pdf`, `.docx`, `.pptx`, `.txt`
            - Chargez un document via le bouton **"Télécharger un fichier"**.  
            - ⚠️ **Remarque** : Les documents contenant plus de **10 pages** peuvent entraîner des résultats imprécis en raison des limitations des modèles d'IA.  
              
@@ -862,20 +862,29 @@ def handle_video_input():
 def handle_youtube_input():
     """Gestion des entrées YouTube"""
     youtube_url = st.text_input("URL YouTube")
-    if youtube_url and st.button("Analyser"):
-        video_processor = VideoProcessor()
-        video_id = video_processor.extract_video_id(youtube_url)
-        
-        if video_id:
-            st.video(youtube_url)
-            with st.spinner("Traitement de la vidéo..."):
-                transcription = video_processor.get_youtube_transcription(video_id)
-                if transcription:
-                    process_and_display_results(None, None, transcription)
-                else:
-                    video_processor.load_cookies()
-                    audio_path = video_processor.download_youtube_audio(youtube_url)
-                    process_and_display_results(audio_path)
+
+    if youtube_url:
+            if not re.match(r'^https?://(?:www\.)?youtube\.com/watch\?v=[\w-]+|^https?://youtu\.be/[\w-]+', youtube_url):
+                st.error("URL YouTube invalide")
+            else:
+                video_processor = VideoProcessor()
+                video_id = video_processor.extract_video_id(youtube_url)
+                if video_id:
+                    st.video(youtube_url)
+
+                    with st.spinner("Récupération du contenu de la vidéo..."):
+                        # Essayer d'abord d'obtenir la transcription
+                        transcription = video_processor.get_youtube_transcription(video_id)
+
+                        if transcription:
+                            st.success("Transcription YouTube trouvée!")
+                            process_and_display_results(None, None, transcription)
+                        else:
+                            st.info("Pas de transcription disponible. Extraction de l'audio...")
+                            video_processor.load_cookies()
+                            audio_path = video_processor.download_youtube_audio(youtube_url)
+                            process_and_display_results(audio_path)
+
 
 def process_and_display_results(file_path=None, audio_bytes=None, transcription=None):
     """Traitement et affichage des résultats"""
@@ -1036,12 +1045,8 @@ def process_document():
             text = process_document_with_password(file, password, doc_processor)
             if text:
                 summary = doc_processor.summarize_text(text)
-                # Génération et téléchargement des documents
-                #generate_and_download_documents(summary)
                 
-                # Option d'envoi par email
-                handle_email_sending(summary)
-                #display_summary_and_downloads(summary)
+                display_summary_and_downloads(summary)
         except ValueError as e:
             st.error(str(e))
 
@@ -1087,8 +1092,60 @@ def process_web():
         except ValueError as e:
             st.error(str(e))
 
-
-
+def display_summary_and_downloads(summary: str):
+    """Affichage du résumé et options de téléchargement"""
+    st.markdown("### 📝 Résumé et Analyse")
+    st.markdown(summary)
+    
+    timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
+    
+    # Génération PDF
+    pdf_filename = f"resume_{timestamp}.pdf"
+    pdf_path = PDFGenerator.create_pdf(summary, pdf_filename)
+    
+    # Génération DOCX
+    docx_filename = f"resume_{timestamp}.docx"
+    docx_path = generate_docx(summary, docx_filename)
+    
+    # Boutons de téléchargement
+    col1, col2 = st.columns(2)
+    with col1:
+        with open(pdf_path, "rb") as pdf_file:
+            st.download_button(
+                "📥 Télécharger PDF",
+                pdf_file,
+                file_name=pdf_filename,
+                mime="application/pdf"
+            )
+    
+    with col2:
+        with open(docx_path, "rb") as docx_file:
+            st.download_button(
+                "📥 Télécharger DOCX",
+                docx_file,
+                file_name=docx_filename,
+                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+            )
+    
+    # Option d'envoi par email
+    st.markdown("### 📧 Recevoir le résumé par email")
+    recipient_email = st.text_input("Entrez votre adresse email:")
+    
+    if st.button("Envoyer par email"):
+        if not is_valid_email(recipient_email):
+            st.error("Veuillez entrer une adresse email valide.")
+        else:
+            with st.spinner("Envoi de l'email en cours..."):
+                email_sender = EmailSender(Config.SENDER_EMAIL, Config.SENDER_PASSWORD)
+                if email_sender.send_email(
+                    recipient_email,
+                    "Résumé de votre contenu",
+                    "Veuillez trouver ci-joint le résumé de votre contenu.",
+                    pdf_path
+                ):
+                    st.success("Email envoyé avec succès!")
+                else:
+                    st.error("Échec de l'envoi de l'email.")
 
 def save_audio_bytes(audio_bytes: bytes) -> str:
     """Sauvegarde les bytes audio dans un fichier temporaire"""