Update app.py

app.py

@@ -36,11 +36,13 @@ def process_document(image):
 def process_batch(images):
     if not images:
         return None, []
-    results = [process_document(Image.open(image.name).convert("RGB")) for image in images]
+    results = [process_document(Image.open(image).convert("RGB")) for image in images]
     return images, results
 
-description = "Démo Gradio pour Donut, une instance du modèle VisionEncoderDecoderModel affiné sur CORD (analyse de documents). Pour l'utiliser, téléchargez une ou plusieurs images et cliquez sur 'Submit', ou cliquez sur l'un des exemples pour les charger."
-article = "Cette application permet maintenant de traiter plusieurs images de tickets de caisse à la fois et affiche les images uploadées à côté des noms de fichiers pour une meilleure visibilité."
+description = "Démo Gradio pour Donut, une instance du modèle VisionEncoderDecoderModel affiné sur CORD (analyse de documents). Sélectionnez des exemples ou téléchargez vos propres images, puis cliquez sur 'Submit'."
+article = "Cette application permet de traiter plusieurs images de tickets de caisse à la fois. Vous pouvez sélectionner des exemples ou uploader vos propres images."
+
+example_images = ["example.jpg", "example_1.jpg", "example_2.jpg", "example_3.jpg", "example_4.jpg"]
 
 with gr.Blocks() as demo:
     gr.Markdown("# Reconnaissance des tickets de caisse en lot 🧾")
@@ -49,32 +51,46 @@ with gr.Blocks() as demo:
     with gr.Row():
         input_images = gr.File(file_count="multiple", type="file", label="Images à traiter")
     
+    with gr.Row():
+        example_gallery = gr.Gallery(example_images, label="Exemples (cliquez pour sélectionner)")
+        selected_examples = gr.State([])
+    
     with gr.Row():
         submit_btn = gr.Button("Submit")
         clear_btn = gr.Button("Clear")
     
     output_json = gr.JSON(label="Résultats")
     
-    gr.Examples(
-        examples=[
-            ["example.jpg"],
-            ["example_2.jpg"]
-        ],
-        inputs=input_images,
-        outputs=[input_images, output_json],
-        fn=process_batch,
-        cache_examples=True,
+    def update_selected_examples(evt: gr.SelectData, state):
+        if evt.index in state:
+            state.remove(evt.index)
+        else:
+            state.append(evt.index)
+        return state
+    
+    example_gallery.select(
+        update_selected_examples,
+        inputs=[selected_examples],
+        outputs=[selected_examples]
     )
     
+    def process_images(uploaded_images, selected_indices):
+        all_images = []
+        if uploaded_images:
+            all_images.extend(uploaded_images)
+        for idx in selected_indices:
+            all_images.append(example_images[idx])
+        return process_batch(all_images)
+    
     submit_btn.click(
-        process_batch,
-        inputs=input_images,
+        process_images,
+        inputs=[input_images, selected_examples],
         outputs=[input_images, output_json]
     )
     
     clear_btn.click(
-        lambda: (None, None),
-        outputs=[input_images, output_json]
+        lambda: (None, None, []),
+        outputs=[input_images, output_json, selected_examples]
     )
     
     gr.Markdown(article)