Update app.py

app.py

@@ -2,93 +2,57 @@ import gradio as gr
 import easyocr
 from transformers import pipeline
 import numpy as np
+from PIL import Image
 from docx import Document
 from pptx import Presentation
 import os
-from PIL import Image
 
-# Initialisation du lecteur EasyOCR (anglais par défaut)
-reader = easyocr.Reader(['en','fr'], gpu=False)
+# OCR (fr + en)
+reader = easyocr.Reader(["fr", "en"], gpu=False)
 
-# Initialisation du pipeline de résumé (BART)
+# Résumé BART
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-def extraire_et_resumer(image):
-    """
-    Extraire le texte d'une image avec EasyOCR et le résumer avec BART
-    
-    Args:
-        image: image PIL ou tableau numpy
-    
-    Returns:
-        tuple: (texte_extrait, texte_resume)
-    """
+# ---------- IMAGE (OCR) ----------
+def process_image(image):
     try:
-        # Convertir l'image PIL en numpy si nécessaire
         if isinstance(image, Image.Image):
             image = np.array(image)
 
-        # OCR avec EasyOCR
-        results = reader.readtext(image)
-
-        # Combiner le texte extrait
-        texte_extrait = " ".join([result[1] for result in results])
-
-        if not texte_extrait.strip():
-            return "Aucun texte détecté dans l'image.", "Aucun texte à résumer."
-
-        # Vérifier le nombre de mots
-        nombre_mots = len(texte_extrait.split())
-
-        if nombre_mots < 30:
-            return texte_extrait, "Le texte est trop court pour être résumé (minimum 30 mots)."
-
-        # Paramètres du résumé
-        max_length = min(150, nombre_mots)
-        min_length = min(30, nombre_mots // 2)
-
-        # Résumé
-        resume = summarizer(
-            if not text or len(text.split()) < 30:
-            return "Texte trop court pour être résumé."
+        results = reader.readtext(image, paragraph=True)
+        text = " ".join([r[1] for r in results])
 
-    max_length = min(150, len(text.split()))
-    min_length = min(30, len(text.split()) // 2)
+        if not text.strip():
+            return "Aucun texte détecté.", "Résumé impossible."
 
-    summary = summarizer(
-        text,
-        max_length=max_length,
-        min_length=min_length,
-        do_sample=False
-    )
-
-    return summary[0]["summary_text"]
+        if len(text.split()) < 30:
+            return text, "Texte trop court pour être résumé."
 
+        wc = len(text.split())
+        summary = summarizer(
+            text,
+            max_length=min(150, wc),
+            min_length=min(30, wc // 2),
+            do_sample=False
         )
 
-        texte_resume = resume[0]["summary_text"]
-
-        return texte_extrait, texte_resume
+        return text, summary[0]["summary_text"]
 
     except Exception as e:
-        return f"Erreur : {str(e)}", "Impossible de générer le résumé."
-
-
+        return f"Erreur : {e}", "Erreur."
 
+# ---------- FICHIER ----------
 def extract_text_from_file(file_path):
     ext = os.path.splitext(file_path)[1].lower()
 
-    # TXT
     if ext == ".txt":
         with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
             return f.read()
 
-    # WORD
     elif ext == ".docx":
         doc = Document(file_path)
-        return "\n".join([p.text for p in doc.paragraphs])
+        return "\n".join(p.text for p in doc.paragraphs)
 
-    # POWERPOINT
     elif ext == ".pptx":
         prs = Presentation(file_path)
         text = []
@@ -101,44 +65,67 @@ def extract_text_from_file(file_path):
     else:
         return ""
 
-# Interface Gradio
-with gr.Blocks(title="OCR & Text Summarizer") as demo:
-    gr.Markdown("""
-    # 📝 OCR & Résumé de texte
-    Téléverse une image contenant du texte :
-    1. Le texte est extrait avec EasyOCR
-    2. Il est résumé avec une IA (BART)
+def process_file(file):
+    try:
+        text = extract_text_from_file(file.name)
+
+        if not text.strip():
+            return "Aucun texte détecté.", "Résumé impossible."
 
-    **Minimum : 30 mots pour le résumé**
+        if len(text.split()) < 30:
+            return text, "Texte trop court pour être résumé."
+
+        wc = len(text.split())
+        summary = summarizer(
+            text,
+            max_length=min(150, wc),
+            min_length=min(30, wc // 2),
+            do_sample=False
+        )
+
+        return text, summary[0]["summary_text"]
+
+    except Exception as e:
+        return f"Erreur : {e}", "Erreur."
+
+# ---------- INTERFACE ----------
+with gr.Blocks(title="OCR & Résumé (Image + Fichier)") as demo:
+    gr.Markdown("""
+    # 🧠 OCR & Résumé intelligent
+    - 🖼️ Images : OCR + résumé
+    - 📄 Fichiers : TXT / DOCX / PPTX
     """)
 
-    with gr.Row():
-        with gr.Column():
-            file_input = gr.File(
-                label="Téléverser un fichier (TXT, DOCX, PPTX)",
-                file_types=[".txt", ".docx", ".pptx"]
-)
+    with gr.Tabs():
 
+        # ----- TAB IMAGE -----
+        with gr.Tab("🖼️ Image (OCR)"):
+            img_input = gr.Image(type="pil", label="Téléverser une image")
+            img_btn = gr.Button("Extraire & Résumer", variant="primary")
+            img_text = gr.Textbox(label="Texte extrait", lines=10)
+            img_summary = gr.Textbox(label="Résumé", lines=5)
+
+            img_btn.click(
+                process_image,
+                inputs=img_input,
+                outputs=[img_text, img_summary]
             )
-            submit_btn = gr.Button("Extraire et résumer", variant="primary")
 
-        with gr.Column():
-            sortie_texte = gr.Textbox(
-                label="Texte extrait",
-                lines=10
+        # ----- TAB FILE -----
+        with gr.Tab("📄 Fichier"):
+            file_input = gr.File(
+                label="Téléverser un fichier",
+                file_types=[".txt", ".docx", ".pptx"]
             )
-            sortie_resume = gr.Textbox(
-                label="Résumé",
-                lines=5
+            file_btn = gr.Button("Extraire & Résumer", variant="primary")
+            file_text = gr.Textbox(label="Texte extrait", lines=10)
+            file_summary = gr.Textbox(label="Résumé", lines=5)
+
+            file_btn.click(
+                process_file,
+                inputs=file_input,
+                outputs=[file_text, file_summary]
             )
 
-    submit_btn.click(
-        fn=process_file,
-        inputs=file_input,
-        outputs=[sortie_texte, sortie_resume]
-    )
-
-
-# Lancer l'application
 if __name__ == "__main__":
     demo.launch()