init

app.py

@@ -1,7 +1,87 @@
 import gradio as gr
+from sentence_transformers import SentenceTransformer
+import weaviate
+from weaviate.classes.init import Auth
+from dotenv import load_dotenv
+import os
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Load environment variables
+load_dotenv()
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+# Initialize the model
+model = SentenceTransformer('all-MiniLM-L6-v2')
+
+# Initialize Weaviate client
+client = weaviate.connect_to_weaviate_cloud(
+    cluster_url=os.getenv("WEAVIATE_URL"),
+    auth_credentials=Auth.api_key(os.getenv("WEAVIATE_API_KEY"))
+)
+
+lux_collection = client.collections.get("LuxData")
+
+def search_images(query):
+    # Encode the query and search
+    query_vector = model.encode(query)
+    results = lux_collection.query.near_vector(
+        near_vector=query_vector.tolist(),
+        limit=20
+    )
+    
+    # Format results for display
+    formatted_results = []
+    for result in results.objects:
+        formatted_results.append({
+            "label": result.properties['label'],
+            "lux_url": result.properties['lux_url'],
+            "image_url": result.properties.get('image_url', "No image available")
+        })
+    
+    return formatted_results
+
+# Create Gradio interface
+def create_interface():
+    with gr.Blocks() as demo:
+        gr.Markdown("# Lux Image Search")
+        
+        with gr.Row():
+            query_input = gr.Textbox(label="Enter your search query")
+            search_button = gr.Button("Search")
+        
+        results_table = gr.HTML(label="Search Results")
+        
+        def on_search(query):
+            results = search_images(query)
+            # Create HTML table with images
+            html = """
+            <table style="width:100%; border-collapse: collapse;">
+                <tr>
+                    <th style="border: 1px solid #ddd; padding: 8px;">Label</th>
+                    <th style="border: 1px solid #ddd; padding: 8px;">Lux URL</th>
+                    <th style="border: 1px solid #ddd; padding: 8px;">Image</th>
+                </tr>
+            """
+            
+            for r in results:
+                image_cell = f'<img src="{r["image_url"]}" style="max-width:200px; max-height:200px;">' if r["image_url"] and r["image_url"] != "No image available" else "No image available"
+                html += f"""
+                <tr>
+                    <td style="border: 1px solid #ddd; padding: 8px;">{r["label"]}</td>
+                    <td style="border: 1px solid #ddd; padding: 8px;"><a href="{r["lux_url"]}" target="_blank">{r["lux_url"]}</a></td>
+                    <td style="border: 1px solid #ddd; padding: 8px;">{image_cell}</td>
+                </tr>
+                """
+            
+            html += "</table>"
+            return html
+        
+        search_button.click(
+            fn=on_search,
+            inputs=query_input,
+            outputs=results_table
+        )
+    
+    return demo
+
+if __name__ == "__main__":
+    demo = create_interface()
+    demo.launch()

requirements.txt

@@ -0,0 +1,4 @@
+weaviate-client
+python-dotenv
+tqdm
+srsly