Update app.py

app.py

@@ -1,10 +1,9 @@
 import gradio as gr
 from sentence_transformers import SentenceTransformer
 import torch
-
 # ----------------------------- Load Models on Demand -----------------------------
 _loaded_models = {}
-
+tab_state = "single"
 def load_model(model_name):
     if model_name not in _loaded_models:
         _loaded_models[model_name] = SentenceTransformer(model_name)
@@ -30,7 +29,7 @@ def find_similar_documents(query, documents, model_name):
     sorted_indices = torch.argsort(similarities, descending=True)
     results = []
     for i, idx in enumerate(sorted_indices):
-        score = similarities[0][idx].item()
+        score = similarities[idx].item()
         doc = doc_list[idx]
         results.append(f"{i+1}. Score: {score:.4f}\n   Document: {doc}")
     
@@ -55,6 +54,7 @@ with gr.Blocks(
     with gr.Tabs():
         # ---------------- Single Model Tab ----------------
         with gr.Tab("🔹 Single Model Search"):
+            tab_state="single"
             with gr.Row():
                 with gr.Column(scale=1):
                     model_selector = gr.Dropdown(
@@ -77,10 +77,31 @@ with gr.Blocks(
 
             search_btn.click(find_similar_documents, [query_input, docs_input, model_selector], result_box)
             query_input.submit(find_similar_documents, [query_input, docs_input, model_selector], result_box)
-
+            examples = [
+                [
+                    "Which planet is known as the Red Planet?",
+                    "Venus is often called Earth's twin because of its similar size.\nMars, known for its reddish hue, is called the Red Planet.\nJupiter, the largest planet, has a red spot.\nSaturn has iconic rings."
+                ],
+                [
+                    "What causes seasons on Earth?",
+                    "The tilt of Earth's axis causes different sunlight distribution.\nThe moon affects tides but not seasons.\nEarth's orbit has minimal effect on seasons.\nRotation causes day and night."
+                ],
+                [
+                    "What gas do plants release during photosynthesis?",
+                    "Plants use sunlight to convert CO₂ into glucose and release oxygen.\nAnimals inhale oxygen and exhale CO₂.\nPhotosynthesis occurs mainly in leaves."
+                ]
+            ]
+            
+            
+            gr.Examples(
+                examples=examples,
+                inputs=[query_input, docs_input],
+                label="Try Examples"
+            )
         # ---------------- Comparison Tab ----------------
         with gr.Tab("⚖️ Model Comparison"):
             with gr.Row():
+                tab_state="comparsion"
                 with gr.Column(scale=1):
                     tarka_selector = gr.Dropdown(
                         label="Tarka Model",
@@ -115,7 +136,27 @@ with gr.Blocks(
                         with gr.Column():
                             open_label = gr.Markdown(visible=False)
                             open_output = gr.Markdown(visible=False)
-
+            examples = [
+                [
+                    "Which planet is known as the Red Planet?",
+                    "Venus is often called Earth's twin because of its similar size.\nMars, known for its reddish hue, is called the Red Planet.\nJupiter, the largest planet, has a red spot.\nSaturn has iconic rings."
+                ],
+                [
+                    "What causes seasons on Earth?",
+                    "The tilt of Earth's axis causes different sunlight distribution.\nThe moon affects tides but not seasons.\nEarth's orbit has minimal effect on seasons.\nRotation causes day and night."
+                ],
+                [
+                    "What gas do plants release during photosynthesis?",
+                    "Plants use sunlight to convert CO₂ into glucose and release oxygen.\nAnimals inhale oxygen and exhale CO₂.\nPhotosynthesis occurs mainly in leaves."
+                ]
+            ]
+            
+            
+            gr.Examples(
+                examples=examples,
+                inputs=[cmp_query, cmp_docs],
+                label="Try Examples"
+            )
             def run_comparison(query, docs, tarka_model, open_model):
                 tarka_res, open_res = compare_models(query, docs, tarka_model, open_model)
                 return (
@@ -139,28 +180,9 @@ with gr.Blocks(
             )
 
     # ---------------- Example Section ----------------
-    gr.Markdown("## 🧪 Try Examples")
-
-    examples = [
-        [
-            "Which planet is known as the Red Planet?",
-            "Venus is often called Earth's twin because of its similar size.\nMars, known for its reddish hue, is called the Red Planet.\nJupiter, the largest planet, has a red spot.\nSaturn has iconic rings."
-        ],
-        [
-            "What causes seasons on Earth?",
-            "The tilt of Earth's axis causes different sunlight distribution.\nThe moon affects tides but not seasons.\nEarth's orbit has minimal effect on seasons.\nRotation causes day and night."
-        ],
-        [
-            "What gas do plants release during photosynthesis?",
-            "Plants use sunlight to convert CO₂ into glucose and release oxygen.\nAnimals inhale oxygen and exhale CO₂.\nPhotosynthesis occurs mainly in leaves."
-        ]
-    ]
-
-    gr.Examples(
-        examples=examples,
-        inputs=[query_input, docs_input],
-        label="Try Example"
-    )
+
+    
+      
 
 # Launch the app
-demo.launch()
+demo.launch()