Update app.py

app.py

@@ -7,7 +7,11 @@ from pdf2image import convert_from_path
 from PIL import Image
 from torch.utils.data import DataLoader
 from tqdm import tqdm
+import os
 
+if not os.path.exists('/tmp/gradio'):
+    os.makedirs('/tmp/gradio')
+    
 from colpali_engine.models import ColQwen2, ColQwen2Processor
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
@@ -21,6 +25,9 @@ model = ColQwen2.from_pretrained(
 processor = ColQwen2Processor.from_pretrained("manu/colqwen2-v1.0-alpha")
 
 def search(query: str, ds, images, k):
+    if not ds:
+        return gr.Error("No documents have been indexed. Please upload and index documents first.")
+    
     k = min(k, len(ds))
     device = "cuda:0" if torch.cuda.is_available() else "cpu"
     if device != model.device: