Update app.py

app.py

@@ -79,7 +79,7 @@ class ParrotletRetriever:
             docs.append({
                 "Rank": i + 1,
                 "Score": f"{match['score']:.4f}",
-                "Document": text
+                "Document": text,
                 "ID": match["concept_id"],
             })
 
@@ -94,19 +94,19 @@ retriever = ParrotletRetriever(MODEL_NAME, PINECONE_API_KEY, PINECONE_INDEX_NAME
 def retrieve_documents(query: str, top_k: int = 5):
     """Perform retrieval and return results."""
     if not query.strip():
-        return pd.DataFrame(), "⚠️ Please enter a valid query."
+        return pd.DataFrame(), "Please enter a valid query."
 
     try:
         results = retriever.search(query, top_k)
         if not results:
-            return pd.DataFrame(), "⚠️ No results found."
+            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)}"
 
 
 # =========================
@@ -119,11 +119,7 @@ SAMPLE_QUERIES = [
     "వాంతులు"
 ]
 
-with gr.Blocks(title="Parrotlet-E Retrieval (Pinecone)", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("""
-    # 🦜 Parrotlet-E: Indic Medical Entities Retrieval
-    Retrieve top-5 semantically similar medical entities from Pinecone using Parrotlet-E embeddings.
-    """)
+with gr.Blocks(title="Parrotlet-e Retrieval", theme=gr.themes.Soft()) as demo:
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -147,7 +143,7 @@ with gr.Blocks(title="Parrotlet-E Retrieval (Pinecone)", theme=gr.themes.Soft())
                 label="Top-K Retrieved"
             )
 
-            search_btn = gr.Button("🔍 Retrieve", variant="primary")
+            search_btn = gr.Button("retrieve", variant="primary")
 
         with gr.Column(scale=2):
             results_output = gr.Dataframe(