Update app.py

app.py

@@ -79,7 +79,9 @@ class ParrotletRetriever:
         return docs
 
 
-
+# =========================
+# Instantiate Retriever
+# =========================
 MODEL_NAME = "ekacare/parrotlet-e"
 retriever = ParrotletRetriever(MODEL_NAME)
 
@@ -95,62 +97,102 @@ def retrieve_documents(query: str, top_k: int = 5):
             return pd.DataFrame(), "No results found."
 
         df = pd.DataFrame(results)
-        status = f"Retrieved top {len(results)} documents."
+        status = f"✅ Retrieved top {len(results)} documents."
         return df, status
 
     except Exception as e:
-        return pd.DataFrame(), f"Error: {str(e)}"
+        return pd.DataFrame(), f"⚠️ Error: {str(e)}"
 
 
 # =========================
-# Gradio Interface
+# Gradio Interface (VERTICAL)
 # =========================
 SAMPLE_QUERIES = [
-    "dm t2",
+    "takhne me dard",
+    "ghotyalu dard",
+    "ghera aana",
+    "vayiru vali",
     "छाती में दर्द",
     "talenovu",
     "వాంతులు"
+    "ಕಾಮಲೆ", # jaundice
+    "பேசுவது சிரமம்", # Dysphasia 
+    "Peshab Kartaana Jalan", # Scalding pain on urination
+    "Kurunnal",
+    "sunn hua",
+    "moochithinaral",
+    "মাথাব্যথা"
 ]
 
 with gr.Blocks(title="Parrotlet-e Retrieval", theme=gr.themes.Soft()) as demo:
 
-    with gr.Row():
-        with gr.Column(scale=1):
-            query_input = gr.Textbox(
-                label="Enter a medical term",
-                placeholder="Type your query here...",
-                lines=1,
-            )
-
-            examples = gr.Examples(
-                examples=SAMPLE_QUERIES,
-                inputs=query_input,
-                label="Example Queries"
-            )
-            
-            top_k_input = gr.Number(
-                label="K",
-                value=5,
-                precision=0,
-                interactive=True
-            )
-
-            search_btn = gr.Button("retrieve", variant="primary")
-
-        with gr.Column(scale=2):
-            results_output = gr.Dataframe(
-                headers=["Rank", "Score", "Document", "ID"],
-                datatype=["number", "str", "str", "str"],
-                interactive=False,
-                wrap=True
-            )
-            status_box = gr.Textbox(label="Status", interactive=False)
+    # ---- Header ----
+    gr.Markdown(
+        """
+        <h1 style='text-align:center; color:#2b4162;'>🦜 Parrotlet-e Retrieval</h1>
+        <p style='text-align:center; font-size:16px; color:#555;'>
+        Multilingual Medical Embedding Search powered by <b>EkaCare's Parrotlet-e</b> model
+        </p>
+        <hr>
+        """,
+    )
+
+    # ---- Model Description ----
+    gr.Markdown(
+        """
+        ### 🧠 Model Description
+        **Parrotlet-e: Indic Medical Entity Embedding Model**  
+        A multilingual embedding model designed to represent Indian medical terminology across multiple scripts and languages.
+
+        - 🔗 **Model on Hugging Face:** [ekacare/parrotlet-e](https://huggingface.co/ekacare/parrotlet-e)  
+        - 📊 **Benchmarked on:** [Eka-IndicMTEB](https://huggingface.co/datasets/ekacare/Eka-IndicMTEB)  
+        - 📰 Read more on our [blog](https://info.eka.care/services/introducing-parrotlet-e-and-eka-indicmteb-bridging-indias-multilingual-healthcare-gap)
+
+        ---
+        """,
+    )
+
+    # ---- Input Section ----
+    with gr.Group():
+        query_input = gr.Textbox(
+            label="Enter a medical term (not sentences in any language)",
+            placeholder="Type your query here...",
+            lines=1,
+        )
+
+        examples = gr.Examples(
+            examples=SAMPLE_QUERIES,
+            inputs=query_input,
+            label="💡 Example Queries"
+        )
+
+        top_k_input = gr.Number(
+            label="Number of results (K)",
+            value=5,
+            precision=0,
+            interactive=True
+        )
 
+        search_btn = gr.Button("🔍 retrieve", variant="primary")
+
+    # ---- Output Section ----
+    with gr.Group():
+        results_output = gr.Dataframe(
+            headers=["Rank", "Score", "Term", "ID"],
+            datatype=["number", "str", "str", "str"],
+            interactive=False,
+            wrap=True,
+            label="📄 Search Results"
+        )
+        status_box = gr.Textbox(label="Status", interactive=False)
+
+    # ---- Function Binding ----
     search_btn.click(
         fn=retrieve_documents,
         inputs=[query_input, top_k_input],
         outputs=[results_output, status_box],
     )
 
+
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)