Update app.py

app.py

@@ -22,20 +22,17 @@ index = pinecone.Index(INDEX_NAME)
 bm25 = BM25Encoder()
 bm25.fit(metadata['productDisplayName'])
 
-# Function to display images in a grid layout
 def display_result(image_batch, match_batch):
     figures = []
     for img, title in zip(image_batch, match_batch):
-        # Ensure the image is in the correct format for encoding
+
         if img.mode != 'RGB':
             img = img.convert('RGB')
 
-        # Convert image to bytes and encode as base64
         b = BytesIO()
         img.save(b, format='PNG')
         img_str = b64encode(b.getvalue()).decode('utf-8')
 
-        # Create HTML figure element with the image title
         figures.append(f'''
             <figure style="margin: 0; padding: 0; text-align: left;">
               <figcaption style="font-weight: bold; margin:0;">{title}</figcaption>
@@ -43,7 +40,6 @@ def display_result(image_batch, match_batch):
             </figure>
         ''')
 
-    # Combine all figures into a single HTML string with reduced spacing
     html_content = f'''
     <div style="display: grid; grid-template-columns: repeat(4, 1fr); gap: 20px; align-items: start;">
         {''.join(figures)}
@@ -52,12 +48,10 @@ def display_result(image_batch, match_batch):
     return html_content
 
 
-# Function to scale vectors based on alpha for hybrid search
 def hybrid_scale(dense, sparse, alpha):
     if alpha < 0 or alpha > 1:
         raise ValueError("Alpha must be between 0 and 1")
     
-    # Scale sparse and dense vectors to create hybrid search vectors
     hsparse = {
         'indices': sparse['indices'],
         'values':  [v * (1 - alpha) for v in sparse['values']]
@@ -67,7 +61,6 @@ def hybrid_scale(dense, sparse, alpha):
     return hdense, hsparse
 
 
-# Function to process the input text and slider value, with error handling
 def process_input(query, slider_value):
     try:
         slider_value = float(slider_value)
@@ -90,29 +83,23 @@ def process_input(query, slider_value):
         return display_result(imgs, matches)
     
     except Exception as e:
-        # Handle exceptions and return a friendly error message
-        return f"<p style='color:red;'>Not found. Try another search: {str(e)}</p>"
+        return f"<p style='color:red;'>Not found. Try another search</p>"
 
 
-# Function to update the textbox value when a dropdown choice is selected
 def update_textbox(choice):
     return choice
 
 
 def text_process(search_string):
-    # Split the search string into words
     search_words = search_string.title().split()
 
-    # Create a regex pattern that matches all words in any order
     pattern = r"(?=.*\b" + r"\b)(?=.*\b".join(map(re.escape, search_words)) + r"\b)"
 
-    # Filter the master list to find items matching the pattern
     filtered_items = [item for item in item_list if re.search(pattern, item)]
 
     return gr.update(visible=True), gr.update(choices=filtered_items, value=filtered_items[0] if filtered_items else "")
 
 
-# Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Get Fashion Items Recommended Based On Your Search..\n"
     "## Recommender System implemented based Pinecone Vector Database with Dense & Sparse Embeddings and Hybrid Search..")
@@ -124,17 +111,12 @@ with gr.Blocks() as demo:
                              value= "Select an item from this list or start typing", allow_custom_value=True,  interactive=True, visible=False)
     slider = gr.Slider(minimum=0, maximum=1, step=0.1, value=0.5, label="Adjust the Slider to get better recommendations that suit what you are looking for..", interactive=True)
 
-    # Automatically update the text input when a dropdown selection is made
     dropdown.change(fn=update_textbox, inputs=dropdown, outputs=text_input)
 
-    # HTML output box to display images
     html_output = gr.HTML(label="Relevant Images")
 
     submit_btn.click(fn=text_process, inputs=[text_input], outputs=[dropdown, dropdown])
-
-    # Process and display images based on text input or slider changes
     text_input.change(fn=process_input, inputs=[text_input, slider], outputs=html_output)
-
     slider.change(fn=process_input, inputs=[text_input, slider], outputs=html_output)
 
 demo.launch(debug=True, share=True)