Update app.py

app.py

@@ -34,7 +34,22 @@ def create_retrieval_figure(res):
             image = cv2.imdecode(np.frombuffer(imgfile, np.uint8), 1)
         # image_resized = cv2.resize(image, (224, 224), interpolation=cv2.INTER_LINEAR)
         except Exception:
-            image = cv2.imdecode(np.array((224, 224, 3), np.uint8), 1)
+            image = np.ones((224, 224, 3), dtype=np.uint8) * 255
+            text = "file not found"
+            font = cv2.FONT_HERSHEY_SIMPLEX
+            font_scale = 0.7
+            font_thickness = 1
+            text_color = (0, 0, 0)  # Black color
+            
+            # Get the size of the text
+            (text_width, text_height), _ = cv2.getTextSize(text, font, font_scale, font_thickness)
+            
+            # Calculate the center position of the text
+            text_x = (image.shape[1] - text_width) // 2
+            text_y = (image.shape[0] + text_height) // 2
+            
+            # Put the text on the image
+            cv2.putText(image, text, (text_x, text_y), font, font_scale, text_color, font_thickness)
         ax = fig.add_subplot(rows, cols, i + 1)
         plt.axis('off')
         plt.imshow(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))