Upload app.py

app.py

@@ -2,6 +2,7 @@
 """
 Educational AI Assistant for Arabic Lessons - Hugging Face Spaces Version
 Spécialement optimisée pour la reconnaissance vocale arabe
+Version avec gestion d'erreurs renforcée
 """
 
 import gradio as gr
@@ -167,14 +168,18 @@ def generate_questions(text, progress=None):
     if progress:
         progress(0.8, "Génération des questions...")
     
-    # Version simplifiée avec des questions génériques
-    questions = """1. ما هي الفكرة الرئيسية للدرس؟
+    try:
+        # Version simplifiée avec des questions génériques
+        questions = """1. ما هي الفكرة الرئيسية للدرس؟
 2. ما هي أهم المفاهيم التي تم شرحها في هذا الدرس؟
 3. كيف يمكن تطبيق المعلومات الواردة في هذا الدرس في الحياة العملية؟
 4. ما هي العلاقة بين المفاهيم المختلفة التي تم تقديمها؟
 5. لو طلب منك تلخيص هذا الدرس في ثلاث نقاط، ماذا ستكتب؟"""
-    
-    return questions
+        
+        return questions
+    except Exception as e:
+        print(f"Erreur dans generate_questions: {e}")
+        return f"Erreur lors de la génération des questions: {str(e)}"
 
 # Fonction pour générer une carte mentale
 def generate_mind_map(text, progress=None):
@@ -185,38 +190,42 @@ def generate_mind_map(text, progress=None):
     if progress:
         progress(0.9, "Création de la carte mentale...")
     
-    # Version ultra-simplifiée avec vérification des indices
-    lines = text.split('.')
-    
-    # Création d'une structure Markdown basique
-    markdown_map = "# خريطة ذهنية للدرس\n\n"
-    
-    # Titre principal - avec vérification que lines n'est pas vide
-    if lines and len(lines) > 0 and lines[0].strip():
-        main_idea = lines[0].strip()[:100] + ("..." if len(lines[0]) > 100 else "")
-        markdown_map += f"## الموضوع الرئيسي\n- {main_idea}\n\n"
-    else:
-        markdown_map += f"## الموضوع الرئيسي\n- (لم يتم العثور على موضوع رئيسي)\n\n"
-    
-    # Points principaux (limités à 3 pour simplicité)
-    markdown_map += "## النقاط الرئيسية\n\n"
-    
-    # Vérifier s'il y a des lignes supplémentaires
-    if len(lines) > 1:
-        count = 0
-        for line in lines[1:]:
-            if line and line.strip() and len(line.strip()) > 15:
-                point_text = line.strip()[:80] + ("..." if len(line) > 80 else "")
-                markdown_map += f"### نقطة {count+1}\n- {point_text}\n\n"
-                count += 1
-                if count >= 3:
-                    break
-    
-    # Si aucun point n'a été trouvé, ajouter un message
-    if "### نقطة" not in markdown_map:
-        markdown_map += "### ملاحظة\n- لم يتم العثور على نقاط رئيسية كافية في النص\n\n"
-    
-    return markdown_map
+    try:
+        # Version ultra-simplifiée avec vérification des indices
+        lines = text.split('.')
+        
+        # Création d'une structure Markdown basique
+        markdown_map = "# خريطة ذهنية للدرس\n\n"
+        
+        # Titre principal - avec vérification que lines n'est pas vide
+        if lines and len(lines) > 0 and lines[0].strip():
+            main_idea = lines[0].strip()[:100] + ("..." if len(lines[0]) > 100 else "")
+            markdown_map += f"## الموضوع الرئيسي\n- {main_idea}\n\n"
+        else:
+            markdown_map += f"## الموضوع الرئيسي\n- (لم يتم العثور على موضوع رئيسي)\n\n"
+        
+        # Points principaux (limités à 3 pour simplicité)
+        markdown_map += "## النقاط الرئيسية\n\n"
+        
+        # Vérifier s'il y a des lignes supplémentaires
+        if len(lines) > 1:
+            count = 0
+            for line in lines[1:]:
+                if line and line.strip() and len(line.strip()) > 15:
+                    point_text = line.strip()[:80] + ("..." if len(line) > 80 else "")
+                    markdown_map += f"### نقطة {count+1}\n- {point_text}\n\n"
+                    count += 1
+                    if count >= 3:
+                        break
+        
+        # Si aucun point n'a été trouvé, ajouter un message
+        if "### نقطة" not in markdown_map:
+            markdown_map += "### ملاحظة\n- لم يتم العثور على نقاط رئيسية كافية في النص\n\n"
+        
+        return markdown_map
+    except Exception as e:
+        print(f"Erreur dans generate_mind_map: {e}")
+        return f"Erreur lors de la création de la carte mentale: {str(e)}"
 
 # Fonction principale de traitement
 def process_lesson(file_obj, progress=gr.Progress()):
@@ -235,35 +244,55 @@ def process_lesson(file_obj, progress=gr.Progress()):
         progress(0.2, "Préparation du fichier...")
         
         # Traitement du fichier selon son type
-        if file_extension in ['.mp4', '.mov', '.avi', '.mkv']:
-            audio_path = extract_audio(file_path)
-            if not audio_path:
-                return "Échec de l'extraction audio.", "", "", ""
-        elif file_extension in ['.mp3', '.wav', '.ogg', '.flac', '.m4a']:
-            audio_path = file_path
-        else:
-            return f"Type de fichier non pris en charge: {file_extension}. Veuillez télécharger un fichier audio ou vidéo.", "", "", ""
+        try:
+            if file_extension in ['.mp4', '.mov', '.avi', '.mkv']:
+                audio_path = extract_audio(file_path)
+                if not audio_path:
+                    return "Échec de l'extraction audio.", "", "", ""
+            elif file_extension in ['.mp3', '.wav', '.ogg', '.flac', '.m4a']:
+                audio_path = file_path
+            else:
+                return f"Type de fichier non pris en charge: {file_extension}. Veuillez télécharger un fichier audio ou vidéo.", "", "", ""
+        except Exception as e:
+            print(f"Erreur lors de la préparation du fichier: {e}")
+            return f"Erreur lors de la préparation du fichier: {str(e)}", "", "", ""
         
         # 1. Transcription
-        transcription = transcribe_audio(audio_path, progress)
+        try:
+            transcription = transcribe_audio(audio_path, progress)
+        except Exception as e:
+            print(f"Erreur lors de la transcription: {e}")
+            transcription = f"La transcription a échoué: {str(e)}"
         
         # Libérer la mémoire après transcription
         clear_gpu_memory()
         
         # 2. Résumé (seulement si la transcription a réussi)
-        if isinstance(transcription, str) and "échoué" not in transcription.lower():
-            summary = summarize_text(transcription, progress)
-        else:
-            summary = "Impossible de générer un résumé car la transcription a échoué."
+        try:
+            if isinstance(transcription, str) and "échoué" not in transcription.lower():
+                summary = summarize_text(transcription, progress)
+            else:
+                summary = "Impossible de générer un résumé car la transcription a échoué."
+        except Exception as e:
+            print(f"Erreur lors du résumé: {e}")
+            summary = f"Le résumé a échoué: {str(e)}"
         
         # Libérer la mémoire après résumé
         clear_gpu_memory()
         
         # 3. Génération de questions (simplifiée)
-        questions = generate_questions(transcription, progress)
+        try:
+            questions = generate_questions(transcription, progress)
+        except Exception as e:
+            print(f"Erreur lors de la génération des questions: {e}")
+            questions = f"La génération des questions a échoué: {str(e)}"
         
         # 4. Génération de la carte mentale (simplifiée)
-        mind_map = generate_mind_map(summary, progress)
+        try:
+            mind_map = generate_mind_map(summary, progress)
+        except Exception as e:
+            print(f"Erreur lors de la génération de la carte mentale: {e}")
+            mind_map = f"La génération de la carte mentale a échoué: {str(e)}"
         
         progress(1.0, "Terminé!")
         return transcription, summary, questions, mind_map