Update app.py

app.py

@@ -1,17 +1,11 @@
 import os
 import torch
-from datasets import load_dataset
-from qdrant_client import QdrantClient
-from qdrant_client.http import models
-from tqdm import tqdm
-from colpali_engine.models import ColQwen2, ColQwen2Processor
 import gradio as gr
 import requests
 from PIL import Image
 from io import BytesIO
-
-# Set up environment variable for faster downloads
-os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
+from qdrant_client import QdrantClient
+from transformers import ColQwen2, ColQwen2Processor
 
 # Initialize ColPali model and processor
 model_name = "vidore/colqwen2-v0.1"
@@ -30,8 +24,7 @@ QDRANT_API_KEY = os.getenv("QDRANT_API_KEY")
 qdrant_client = QdrantClient(url="https://davanstrien-qdrant-test.hf.space",
                              port=None, api_key=QDRANT_API_KEY, timeout=10)
 
-
-collection_name = "song_sheets"  # Replace with your actual collection name
+collection_name = "your_collection_name"  # Replace with your actual collection name
 
 def search_images_by_text(query_text, top_k=5):
     # Process and encode the text query
@@ -52,19 +45,20 @@ def search_images_by_text(query_text, top_k=5):
 
     return search_result
 
-def modify_iiif_url(url, width, height):
+def modify_iiif_url(url, size_percent):
+    # Modify the IIIF URL to use percentage scaling
     parts = url.split('/')
     size_index = -3
-    parts[size_index] = f"{width},{height}"
+    parts[size_index] = f"pct:{size_percent}"
     return '/'.join(parts)
 
-def search_and_display(query, top_k, width, height):
+def search_and_display(query, top_k, size_percent):
     results = search_images_by_text(query, top_k)
     images = []
     captions = []
 
     for result in results.points:
-        modified_url = modify_iiif_url(result.payload['image_url'], width, height)
+        modified_url = modify_iiif_url(result.payload['image_url'], size_percent)
         response = requests.get(modified_url)
         img = Image.open(BytesIO(response.content)).convert("RGB")
         images.append(img)
@@ -78,15 +72,14 @@ iface = gr.Interface(
     inputs=[
         gr.Textbox(label="Search Query"),
         gr.Slider(minimum=1, maximum=20, step=1, label="Number of Results", value=5),
-        gr.Slider(minimum=100, maximum=1000, step=50, label="Image Width", value=300),
-        gr.Slider(minimum=100, maximum=1000, step=50, label="Image Height", value=300)
+        gr.Slider(minimum=1, maximum=100, step=1, label="Image Size (%)", value=100)
     ],
     outputs=[
         gr.Gallery(label="Search Results", show_label=False, columns=5, height="auto"),
         gr.JSON(label="Captions")
     ],
-    title="Image Search with IIIF Resizing",
-    description="Enter a text query to search for images. You can adjust the number of results and the size of the returned images."
+    title="Image Search with IIIF Percentage Resizing",
+    description="Enter a text query to search for images. You can adjust the number of results and the size of the returned images as a percentage of the original size."
 )
 
 # Launch the Gradio interface