Update app.py

app.py

@@ -25,7 +25,7 @@ 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.")
+        return None, "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"
@@ -46,11 +46,11 @@ def search(query: str, ds, images, k):
     for idx in top_k_indices:
         results.append((images[idx], f"Page {idx}"))
 
-    return results
+    return results, ""
 
 def index(files, ds):
     if not files:
-        return gr.Error("No files uploaded. Please upload PDF files to index."), ds, []
+        return "No files uploaded. Please upload PDF files to index.", ds, []
     
     print("Converting files")
     images = convert_files(files)
@@ -117,8 +117,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     convert_button.click(index, inputs=[file, embeds], outputs=[message, embeds, imgs])
     search_button = gr.Button("🔍 Search", variant="primary")
     output_gallery = gr.Gallery(label="Retrieved Documents", height=600, show_label=True)
+    error_output = gr.Textbox(label="Error Message")
 
-    search_button.click(search, inputs=[query, embeds, imgs, k], outputs=[output_gallery])
+    search_button.click(search, inputs=[query, embeds, imgs, k], outputs=[output_gallery, error_output])
 
 if __name__ == "__main__":
     demo.queue(max_size=10).launch(debug=True)