Update app.py

app.py

@@ -302,7 +302,9 @@ def infer(prompt, neg, seed, rand, w, h, cfg, steps, scheduler_type,
         gallery_items = [(item[2], item[1].replace("|-|", "\n")) for item in history]
     
         processed_img, processed_gallery_items = process_image(img, gallery_items)
-    
+                if processed_img is None or processed_gallery_items is None:
+                print("Image processing failed, skipping history update.")
+                return None, history_gallery, history_tables
         latest_caption_table = make_html_table(caption_text)
         return processed_img, processed_gallery_items, latest_caption_table