Update app.py

app.py

@@ -7,6 +7,9 @@ import pandas as pd
 import re
 import tempfile
 import unicodedata  # Pour sanitizer le nom de fichier
+from openpyxl import Workbook
+from openpyxl.drawing.image import Image as OpenpyxlImage
+from openpyxl.styles import Font, Alignment
 
 # Initialize Groq client
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
@@ -104,7 +107,7 @@ def process_image_and_get_response(image):
     # Prompt optimisé pour structure, précision et inclusion des infos supplémentaires
     prompt = (
         "Analyse l'image et extrait tout le contenu. "
-        "D'abord, suggère un nom de fichier descriptif pour l'Excel basé sur le contenu principal de l'image (court, sans extension, ex: 'Tableau_Dossiers_2022'). "
+        "D'abord, suggère un nom de fichier descriptif et précis pour l'Excel basé sur le contenu principal de l'image (inclue des éléments clés comme dates, thèmes ou identifiants uniques ; court, sans extension, ex: 'Tableau_Dossiers_Dentistes_2022_2024'). "
         "Ensuite, extrait TOUT texte additionnel autour, au-dessus, en-dessous ou à côté du tableau (titres, en-têtes, notes, pieds de page, logos, etc.), en le recopiant mot pour mot, même si c'est dispersé. "
         "Si aucun texte additionnel, laisse vide. "
         "Enfin, extrait le tableau en entier, en recopiant TOUTES les lignes et colonnes à l'identique, y compris les lignes vides ou partielles si elles existent dans l'image. "
@@ -114,7 +117,7 @@ def process_image_and_get_response(image):
         "Remplis les cellules vides avec '' si nécessaire pour maintenir l'alignement, mais ne filtre pas les lignes vides si elles sont présentes. "
         "N'ajoute aucun texte explicatif dans le tableau. "
         "Structure ta réponse exactement comme suit :\n"
-        "Filename: [nom_suggéré]\n"
+        "Filename: [nom_suggéré_précis]\n"
         "Additional text: [tout le texte additionnel, séparé par \n si plusieurs lignes ; sinon vide]\n"
         "Table:\n"
         "[le tableau Markdown ici]"
@@ -135,8 +138,8 @@ def process_image_and_get_response(image):
                     ]
                 }
             ],
-            temperature=0.2,  # Réduit pour plus de précision et respect de la structure
-            max_completion_tokens=4096,  # Augmenté pour tableaux complexes
+            temperature=0.5,  # Réduit pour plus de précision et respect de la structure
+            max_completion_tokens=8096,  # Augmenté pour tableaux complexes
             top_p=1,
             stream=False,
             stop=None
@@ -150,19 +153,51 @@ def process_image_and_get_response(image):
         # Parse le tableau en DataFrame
         df = parse_markdown_table_to_df(table_text)
 
+        # Créer un fichier temporaire pour l'image originale
+        img_path = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg").name
+        image.save(img_path, format="JPEG")
+
         # Créer le fichier Excel avec nom personnalisé
         excel_path = f"/tmp/{filename if filename else 'tableau_extrait'}.xlsx"
-        with pd.ExcelWriter(excel_path, engine='openpyxl') as writer:
-            # Feuille pour le tableau
-            if not df.empty and "Erreur" not in df.columns:
-                df.to_excel(writer, sheet_name='Tableau_Extrait', index=False)
-            else:
-                pd.DataFrame({"Message": ["Erreur lors du parsing du tableau"]}).to_excel(writer, sheet_name='Tableau_Extrait', index=False)
-
-            # Feuille pour le texte additionnel, si présent
-            if additional_text:
-                additional_df = pd.DataFrame({"Texte Additionnel": additional_text.split('\n')})
-                additional_df.to_excel(writer, sheet_name='Infos_Supplementaires', index=False)
+        wb = Workbook()
+
+        # Feuille pour le tableau
+        ws_table = wb.active
+        ws_table.title = 'Tableau_Extrait'
+        if not df.empty and "Erreur" not in df.columns:
+            # Écrire les en-têtes en gras
+            for col_num, value in enumerate(df.columns.values, start=1):
+                cell = ws_table.cell(row=1, column=col_num, value=value)
+                cell.font = Font(bold=True)
+                cell.alignment = Alignment(wrap_text=True, vertical='top')
+
+            # Écrire les données avec wrap text
+            for row_num, row in enumerate(df.values, start=2):
+                for col_num, value in enumerate(row, start=1):
+                    cell = ws_table.cell(row=row_num, column=col_num, value=value)
+                    cell.alignment = Alignment(wrap_text=True, vertical='top')
+
+        else:
+            ws_table.cell(row=1, column=1, value="Erreur lors du parsing du tableau")
+
+        # Feuille pour le texte additionnel, si présent
+        if additional_text:
+            ws_additional = wb.create_sheet(title='Infos_Supplementaires')
+            lines = additional_text.split('\n')
+            for row_num, line in enumerate(lines, start=1):
+                cell = ws_additional.cell(row=row_num, column=1, value=line)
+                cell.alignment = Alignment(wrap_text=True, vertical='top')
+
+        # Feuille pour l'image originale
+        ws_image = wb.create_sheet(title='Image_Originale')
+        img = OpenpyxlImage(img_path)
+        ws_image.add_image(img, 'A1')
+
+        # Sauvegarder le workbook
+        wb.save(excel_path)
+
+        # Nettoyer le fichier image temporaire
+        os.unlink(img_path)
 
         return response, excel_path
 
@@ -175,10 +210,10 @@ iface = gr.Interface(
     inputs=gr.Image(type="pil", label="Uploader une image contenant un tableau"),
     outputs=[
         gr.Textbox(label="Réponse de l'IA (texte additionnel + tableau Markdown pour copier-coller)"),
-        gr.File(label="Télécharger le fichier Excel (avec tableau et infos supp.)")
+        gr.File(label="Télécharger le fichier Excel (avec tableau, infos supp. et image originale)")
     ],
     title="Extraction de Tableau depuis Image avec Groq et Export Excel",
-    description="Uploader une image avec un tableau. L'IA extrait le texte additionnel et le tableau, puis génère un Excel avec des feuilles séparées."
+    description="Uploader une image avec un tableau. L'IA extrait le texte additionnel et le tableau, puis génère un Excel avec des feuilles séparées, y compris l'image originale."
 )
 
 # Launch the interface