Deploy Panini 1.0 Demo

app.py

@@ -12,55 +12,56 @@ print(f"✅ Boot Complete in {boot_time:.4f}s")
 # 2. Key Processors
 def omni_search(query):
     if not query or len(query) < 2:
-        return "", "", "", ""
+        return "", ""
         
     start = time.time()
     results = model.query_omni(query)
     duration = time.time() - start
     
-    # Format HTML Cards
+    html = ""
     
     # 🌍 Roots
-    html_roots = ""
     if results["roots"]:
-        html_roots += "<div class='card glass-panel'>"
-        html_roots += "<h3>🌍 Ancestry</h3>"
+        html += "<div class='card glass-panel'>"
+        html += "<h3>🌍 Ancestry</h3>"
+        html += "<div class='item'>The concept <b>" + query.upper() + "</b> connects to:</div>"
         for r in results["roots"]:
-            html_roots += f"<div class='item'>{r}</div>"
-        html_roots += "</div>"
+            html += f"<div class='item'>🔗 {r}</div>"
+        html += "</div>"
     
     # ⚖️ Law
-    html_law = ""
     if results["law"]:
-        html_law += "<div class='card glass-panel'>"
-        html_law += "<h3>⚖️ Sovereign Laws</h3>"
+        html += "<div class='card glass-panel'>"
+        html += "<h3>⚖️ Sovereign Laws</h3>"
         for item in results["law"]:
-            html_law += f"<div class='item'><b>{item['source']}</b><br>{item['content']}</div>"
-        html_law += "</div>"
+            html += f"<div class='item'>{item['content']}</div>"
+        html += "</div>"
 
     # 💰 Economy
-    html_econ = ""
     if results["econ"]:
-        html_econ += "<div class='card glass-panel'>"
-        html_econ += "<h3>💰 Artha (Economy)</h3>"
+        html += "<div class='card glass-panel'>"
+        html += "<h3>💰 Artha (Economy)</h3>"
         for item in results["econ"]:
-             html_econ += f"<div class='item'><b>{item['source']}</b><br>{item['content']}</div>"
-        html_econ += "</div>"
+             html += f"<div class='item'><b>{item['source']}</b><br>{item['content']}</div>"
+        html += "</div>"
         
     # 🗺️ Geography / Ayurveda
-    html_geo = ""
     if results["geo"] or results["med"]:
-        html_geo += "<div class='card glass-panel'>"
-        html_geo += "<h3>🗺️ Desha & Ayurveda</h3>"
+        html += "<div class='card glass-panel'>"
+        html += "<h3>🗺️ Desha & Ayurveda</h3>"
         for item in results["geo"]:
-            html_geo += f"<div class='item'><b>📍 {item['content']}</b></div>"
+            html += f"<div class='item'><b>📍 {item['content']}</b></div>"
         for item in results["med"]:
-            html_geo += f"<div class='item'><b>🌿 {item['content']}</b></div>"
-        html_geo += "</div>"
+            html += f"<div class='item'><b>🌿 {item['content']}</b></div>"
+        html += "</div>"
         
+    # If nothing found
+    if not html:
+        html = "<div style='text-align:center; padding: 20px; color: #888;'>Scanning Smriti... No direct matches found. Try a broader term.</div>"
+
     stats = f"⚡ Inference: {duration*1000:.1f}ms | 🧠 Active Nodes: {len(model.smriti)}"
     
-    return html_roots, html_law, html_econ, html_geo, stats
+    return html, stats
 
 # 3. The Sovereign Design (CSS) - Light Theme
 custom_css = """
@@ -83,10 +84,11 @@ textarea, input {
     border: 1px solid #e5e7eb !important;
     color: #000000 !important;
     font-family: 'Outfit', sans-serif !important;
-    font-size: 20px !important;
+    font-size: 24px !important;
     text-align: left !important;
-    border-radius: 8px !important;
-    box-shadow: 0 1px 2px 0 rgba(0, 0, 0, 0.05);
+    padding: 15px !important;
+    border-radius: 12px !important;
+    box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.05);
 }
 
 /* Cards */
@@ -94,40 +96,39 @@ textarea, input {
     background: #ffffff;
     border: 1px solid #e5e7eb;
     border-radius: 12px;
-    padding: 20px;
-    margin-bottom: 15px;
-    box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1);
+    padding: 24px;
+    margin-bottom: 20px;
+    box-shadow: 0 10px 15px -3px rgba(0, 0, 0, 0.05);
     transition: all 0.2s ease;
-}
-
-.card:hover {
-    border-color: #ff5722;
-    transform: translateY(-2px);
+    max-width: 800px;
+    margin-left: auto;
+    margin-right: auto;
 }
 
 .card h3 {
     color: #ea580c;
     border-bottom: 2px solid #f3f4f6;
-    padding-bottom: 8px;
-    margin-bottom: 12px;
-    font-size: 1.1em;
+    padding-bottom: 12px;
+    margin-bottom: 16px;
+    font-size: 1.25em;
     font-weight: 600;
+    letter-spacing: -0.02em;
 }
 
 .item {
-    margin-bottom: 10px;
+    margin-bottom: 12px;
     font-size: 16px;
-    line-height: 1.5;
-    color: #374151;
+    line-height: 1.6;
+    color: #4b5563;
 }
 
-b { color: #0369a1; }
+b { color: #0369a1; font-weight: 600; }
 
 #stats-bar {
     text-align: center;
-    color: #6b7280;
-    font-size: 13px;
-    margin-top: 30px;
+    color: #9ca3af;
+    font-size: 12px;
+    margin-top: 40px;
 }
 """
 
@@ -138,9 +139,9 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     gr.Markdown("Type a concept (e.g. 'RIGHTS', 'MOTHER', 'GST'). The model searches Law, Roots, and Economy in real-time.")
     
     # The Omni-Input (Light Mode)
-    inp = gr.Textbox(placeholder="Type to search Sovereign Knowledge...", label="", autofocus=True)
+    inp = gr.Textbox(placeholder="Search Knowledge Base...", label="", autofocus=True)
     
-    # Predefined Questions (Capabilities Demo)
+    # Predefined Questions
     gr.Examples(
         examples=[
             "Constitution Preamble",
@@ -154,18 +155,11 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
         label="Try these examples:"
     )
     
-    # The HUD (Heads Up Display)
-    with gr.Row():
-        out_roots = gr.HTML()
-        out_law = gr.HTML()
-    
-    with gr.Row():
-        out_econ = gr.HTML()
-        out_geo = gr.HTML()
-        
+    # Single Stream Output
+    out_stream = gr.HTML()
     stats = gr.Markdown(elem_id="stats-bar")
 
     # Real-time Trigger
-    inp.change(fn=omni_search, inputs=inp, outputs=[out_roots, out_law, out_econ, out_geo, stats])
+    inp.change(fn=omni_search, inputs=inp, outputs=[out_stream, stats])
 
 demo.launch()

meru_os/panini_model/panini_model.py

@@ -107,8 +107,35 @@ class PaniniLLM(nn.Module):
         if query_upper in self.universal_roots:
             results["roots"] = self.universal_roots[query_upper]
             
-        # 2. Linear Scan of Smriti (Knowledge)
-        # Limit results per category for speed
+    def query_omni(self, query):
+        """
+        The All-Seeing Eye. Searches EVERYTHING.
+        Returns a dict of results by category.
+        """
+        query_parts = query.lower().strip().split()
+        if not query_parts: return {}
+        
+        query_upper = query.upper().strip()
+        
+        results = {
+            "roots": [],
+            "law": [],
+            "econ": [],
+            "geo": [],
+            "med": []
+        }
+        
+        # 1. Check Roots (Exact Match preferred, or partial)
+        if query_upper in self.universal_roots:
+            results["roots"] = self.universal_roots[query_upper]
+        else:
+            # Try partial root match
+            for k, v in self.universal_roots.items():
+                if query_upper in k:
+                    results["roots"] = v
+                    break
+            
+        # 2. Key-Value Scan of Smriti (Knowledge)
         limits = {"constitution": 2, "ipc": 2, "gst": 3, "mandi": 3, "pin": 3, "ayur": 2}
         counts = {k: 0 for k in limits}
         
@@ -116,42 +143,46 @@ class PaniniLLM(nn.Module):
             source = item.get("_source", "unknown")
             category = self._categorize(source)
             
-            # Simple Text Match
             matches = False
             content = ""
             
-            # PIN Code match (Exact or City)
-            if source == "pin_codes":
-                if query_upper in str(item.get("pincode", "")) or \
-                   query_upper in item.get("officeName", "").upper() or \
-                   query_upper in item.get("districtName", "").upper():
-                    matches = True
-                    content = f"{item.get('officeName')} ({item.get('pincode')})"
-            
-            # Law/Text match
-            else:
-                for v in item.values():
-                    if isinstance(v, str) and query_lower in v.lower():
-                        matches = True
-                        # Find the relevant snippet
-                        content = str(v)[:150] + "..."
-                        break
+            # Helper to stringify item for search
+            # We construct a "Searchable Text" blob from values
+            blob = " ".join([str(v).lower() for v in item.values() if isinstance(v, (str, int))])
             
+            # SEARCH LOGIC: ALL Keywords must be present
+            if all(part in blob for part in query_parts):
+                matches = True
+                
+                # Format Content based on Source
+                if source == "pin_codes":
+                    content = f"{item.get('officeName')} ({item.get('pincode')}) - {item.get('districtName')}"
+                elif source == "constitution":
+                    article = item.get("Article", "") or item.get("title", "")
+                    text = item.get("Text", "") or str(item)
+                    content = f"<b>{article}</b><br>{text[:300]}..."
+                elif source == "ipc":
+                    section = item.get("Section", "")
+                    desc = item.get("Description", "") or item.get("Offense", "")
+                    content = f"<b>Section {section}</b><br>{desc[:300]}..."
+                else:
+                    # Generic fallback
+                    content = str(blob)[:200] + "..."
+
             if matches:
-                # Add to results if under limit
-                # Tag mapping
-                limit_key = source.split("_")[0] # 'constitution', 'ipc'
-                if limit_key not in counts: limit_key = "pin" # default
+                limit_key = source.split("_")[0] 
+                if limit_key not in counts: limit_key = "pin"
                 
-                if counts.get(limit_key, 0) < 5:
+                if counts.get(limit_key, 0) < limits.get(limit_key, 5):
                     results[category].append({
                         "source": source.upper(), 
-                        "content": content,
-                        "full": item
+                        "content": content
                     })
                     if limit_key in counts: counts[limit_key] += 1
                     
         return results
+                    
+        return results
 
     def _categorize(self, source):
         if source in ["constitution", "ipc"]: return "law"