Update app.py

app.py

@@ -56,8 +56,23 @@ def search_by_image(user_image):
     user_vector = features.cpu().numpy().flatten().reshape(1, -1)
     return get_recommendations_from_vector(user_vector)
 
+def search_by_text_only(prompt):
+    """Tab 2: Fast Text-to-Image Retrieval (No Image Generation)"""
+    if not prompt:
+        return None, "", None, "", None, ""
+        
+    # CLIP embeds the text directly into the same space as the images
+    inputs = processor(text=[prompt], return_tensors="pt", padding=True).to(device)
+    with torch.no_grad():
+        features = clip_model.get_text_features(**inputs)
+        if not isinstance(features, torch.Tensor):
+            features = features.pooler_output if hasattr(features, 'pooler_output') else features[0]
+            
+    user_vector = features.cpu().numpy().flatten().reshape(1, -1)
+    return get_recommendations_from_vector(user_vector)
+
 def generate_and_recommend(prompt):
-    """Tab 2: Generative AI Concept Search"""
+    """Tab 3: Generative AI Concept Search"""
     if not prompt:
         return None, None, "", None, "", None, ""
         
@@ -76,7 +91,6 @@ def generate_and_recommend(prompt):
 
 # --- 4. Gradio User Interface (Premium UI) ---
 
-# Using a sleek, monochrome theme for an architectural, high-end feel
 custom_theme = gr.themes.Monochrome(
     primary_hue="neutral",
     secondary_hue="neutral",
@@ -93,13 +107,13 @@ with gr.Blocks(theme=custom_theme) as demo:
     
     with gr.Tabs():
         
-        # --- TAB 1: Classic Search (Safe for the assignment requirements) ---
-        with gr.TabItem("🔍 Search by Image (Catalog Match)"):
+        # --- TAB 1: Classic Search by Image ---
+        with gr.TabItem("🖼️ Search by Image"):
             gr.Markdown("Upload an inspiration photo to instantly discover visually and stylistically similar rooms from our curated catalog.")
             with gr.Row():
                 with gr.Column(scale=1):
                     image_input = gr.Image(label="Upload Inspiration", type="pil")
-                    img_submit_btn = gr.Button("Find Matches", variant="primary")
+                    img_submit_btn = gr.Button("Find Matches (Instant)", variant="primary")
                     
                 with gr.Column(scale=2):
                     with gr.Row():
@@ -114,18 +128,42 @@ with gr.Blocks(theme=custom_theme) as demo:
                             img_score3 = gr.Textbox(label="Confidence", interactive=False)
                             
             img_submit_btn.click(
-                fn=search_by_image,
-                inputs=[image_input],
+                fn=search_by_image, inputs=[image_input],
                 outputs=[img_rec1, img_score1, img_rec2, img_score2, img_rec3, img_score3]
             )
 
-        # --- TAB 2: Advanced GenAI Search (The Startup Pitch feature) ---
-        with gr.TabItem("✨ AI Dream Room (Generative Concept)"):
-            gr.Markdown("Describe your ideal space. Our Generative AI will draft a concept, and our vision engine will find the closest real-world equivalents.")
+        # --- TAB 2: Fast Text Search (CLIP Multi-modal) ---
+        with gr.TabItem("🔍 Fast Text Search"):
+            gr.Markdown("Describe a room in text. Our multi-modal vision engine will instantly search the catalog for matching designs.")
+            with gr.Row():
+                with gr.Column(scale=1):
+                    fast_text_input = gr.Textbox(label="Search Query", placeholder="e.g., A minimalist industrial bedroom with exposed brick...", lines=3)
+                    fast_txt_submit_btn = gr.Button("Search Catalog (Instant)", variant="primary")
+                    
+                with gr.Column(scale=2):
+                    with gr.Row():
+                        with gr.Column():
+                            ft_rec1 = gr.Image(label="Top Match")
+                            ft_score1 = gr.Textbox(label="Confidence", interactive=False)
+                        with gr.Column():
+                            ft_rec2 = gr.Image(label="2nd Match")
+                            ft_score2 = gr.Textbox(label="Confidence", interactive=False)
+                        with gr.Column():
+                            ft_rec3 = gr.Image(label="3rd Match")
+                            ft_score3 = gr.Textbox(label="Confidence", interactive=False)
+
+            fast_txt_submit_btn.click(
+                fn=search_by_text_only, inputs=[fast_text_input],
+                outputs=[ft_rec1, ft_score1, ft_rec2, ft_score2, ft_rec3, ft_score3]
+            )
+
+        # --- TAB 3: Advanced GenAI Search ---
+        with gr.TabItem("✨ AI Design Studio"):
+            gr.Markdown("Describe your ideal space. Our Generative AI will draft a concept from scratch, and then find the closest real-world equivalents. <br> ⏳ *Note: Image generation takes ~3-5 minutes on free servers.*")
             with gr.Row():
                 with gr.Column(scale=1):
-                    text_input = gr.Textbox(label="Concept Description", placeholder="e.g., A minimalist industrial bedroom with exposed brick...", lines=3)
-                    txt_submit_btn = gr.Button("Generate & Match", variant="primary")
+                    text_input = gr.Textbox(label="Concept Description", placeholder="e.g., A cozy modern living room with large windows...", lines=3)
+                    txt_submit_btn = gr.Button("Generate Concept & Match", variant="primary")
                     gen_output = gr.Image(label="AI Drafted Concept", type="pil")
                     
                 with gr.Column(scale=2):
@@ -141,8 +179,7 @@ with gr.Blocks(theme=custom_theme) as demo:
                             txt_score3 = gr.Textbox(label="Confidence", interactive=False)
 
             txt_submit_btn.click(
-                fn=generate_and_recommend,
-                inputs=[text_input],
+                fn=generate_and_recommend, inputs=[text_input],
                 outputs=[gen_output, txt_rec1, txt_score1, txt_rec2, txt_score2, txt_rec3, txt_score3]
             )