Update app.py

app.py

@@ -75,12 +75,11 @@ def run_query(start_node, relation, end_node, limit):
     Runs an efficient query by filtering on text paths first,
     then joining by indexed IDs.
     
-    *** FIX ***: This now queries the `node.id` column (which contains the full path)
-    and the `relation.label` column, based on the raw query results.
+    This now queries the `node.id` column (which contains the full path)
+    and the `relation.label` column.
     """
     print(f"Running query: start='{start_node}', rel='{relation}', end='{end_node}'")
     
-    # --- FIX: Select the `id` column (which has the path) ---
     select_clause = """
         SELECT
             e.id AS edge_id,
@@ -106,26 +105,20 @@ def run_query(start_node, relation, end_node, limit):
     params = []
     
     try:
-        # --- FIX: Query `node.id` for nodes, `relation.label` for relations ---
         if start_node:
-            # We add a wildcard to the front to match the full http://... ID
-            # unless the user has already added their own wildcard.
             param_val = start_node if "%" in start_node else f"%{start_node}"
             where_conditions.append(f"e.start_id IN (SELECT id FROM node WHERE id LIKE ?)")
             params.append(param_val)
         
         if relation:
-            # Relation table uses `label` which is a direct match (e.g., /r/IsA)
             op = "LIKE" if "%" in relation else "="
             where_conditions.append(f"e.rel_id IN (SELECT id FROM relation WHERE label {op} ?)")
             params.append(relation)
 
         if end_node:
-            # We add a wildcard to the front to match the full http://... ID
             param_val = end_node if "%" in end_node else f"%{end_node}"
             where_conditions.append(f"e.end_id IN (SELECT id FROM node WHERE id LIKE ?)")
             params.append(param_val)
-        # --- End of Fix ---
 
         if not where_conditions:
             where_clause = " WHERE 1=1" # Get random edges if no filter
@@ -176,6 +169,73 @@ def run_raw_query(sql_query):
         traceback.print_exc()
         return pd.DataFrame(), f"**Query Failed!**\n\n`{e}`"
 
+def get_semantic_profile(word, lang='en'):
+    """
+    NEW FUNCTION: Gets a convenient profile of a single word,
+    querying for its most important semantic relations.
+    """
+    if not word:
+        return "Please enter a word."
+    
+    word = word.strip().lower().replace(' ', '_')
+    like_path = f"%/c/{lang}/{word}"
+    print(f"Getting semantic profile for: {like_path}")
+
+    # These are the relations we learned about from relations.py
+    relations_to_check = [
+        "/r/IsA", "/r/PartOf", "/r/HasA", "/r/UsedFor", "/r/CapableOf",
+        "/r/Causes", "/r/HasProperty", "/r/Synonym", "/r/Antonym", 
+        "/r/AtLocation", "/r/RelatedTo"
+    ]
+    
+    output_md = f"# Semantic Profile for '{word}'\n\n"
+    
+    try:
+        with get_db_connection() as conn:
+            for rel in relations_to_check:
+                output_md += f"## {rel}\n\n"
+                
+                # Query 1: Word as the start node
+                query1 = """
+                    SELECT en.id, en.label 
+                    FROM edge e 
+                    JOIN node s ON e.start_id = s.id 
+                    JOIN node en ON e.end_id = en.id 
+                    JOIN relation r ON e.rel_id = r.id 
+                    WHERE s.id LIKE ? AND r.label = ? 
+                    ORDER BY e.weight DESC LIMIT 7
+                """
+                cursor1 = conn.execute(query1, (like_path, rel))
+                results1 = cursor1.fetchall()
+                
+                # Query 2: Word as the end node
+                query2 = """
+                    SELECT s.id, s.label 
+                    FROM edge e 
+                    JOIN node s ON e.start_id = s.id 
+                    JOIN node en ON e.end_id = en.id 
+                    JOIN relation r ON e.rel_id = r.id 
+                    WHERE en.id LIKE ? AND r.label = ? 
+                    ORDER BY e.weight DESC LIMIT 7
+                """
+                cursor2 = conn.execute(query2, (like_path, rel))
+                results2 = cursor2.fetchall()
+                
+                if not results1 and not results2:
+                    output_md += "- (No results)\n\n"
+                else:
+                    for r_id, r_label in results1:
+                        output_md += f"- `{word}` {rel} `{r_label}` *({r_id})*\n"
+                    for r_id, r_label in results2:
+                         output_md += f"- `{r_label}` {rel} `{word}` *({r_id})*\n"
+                output_md += "\n"
+        
+        return output_md
+
+    except Exception as e:
+        print(f"Error in get_semantic_profile: {e}")
+        traceback.print_exc()
+        return f"**An error occurred while building the profile:**\n\n`{e}`"
 
 # --- 3. Build the Gradio UI ---
 
@@ -184,6 +244,15 @@ with gr.Blocks(title="ConceptNet SQLite Explorer") as demo:
     gr.Markdown(f"**Note:** Initial query might take a few seconds as the database warms up.")
     
     with gr.Tabs():
+        with gr.TabItem("Semantic Profile"):
+            gr.Markdown(
+                "**Get a simple semantic profile for a single English word.**\n"
+                "This will query for common relations like `/r/IsA`, `/r/HasA`, `/r/UsedFor`, etc."
+            )
+            word_input = gr.Textbox(label="Word", placeholder="dog")
+            semantic_btn = gr.Button("Get Profile", variant="primary")
+            semantic_output = gr.Markdown("Profile will appear here...")
+            
         with gr.TabItem("Query Builder"):
             gr.Markdown(
                 "**Run a query against the database.**\n"
@@ -191,7 +260,6 @@ with gr.Blocks(title="ConceptNet SQLite Explorer") as demo:
                 "Use the *path part* of the ID (e.g., `/c/en/dog` or `/r/Is%`)."
             )
             with gr.Row():
-                # --- FIX: Updated labels for clarity ---
                 start_input = gr.Textbox(label="Start Node Path", placeholder="/c/en/dog")
                 rel_input = gr.Textbox(label="Relation Path", placeholder="/r/IsA")
                 end_input = gr.Textbox(label="End Node Path", placeholder="/c/en/animal")
@@ -204,10 +272,9 @@ with gr.Blocks(title="ConceptNet SQLite Explorer") as demo:
         
         with gr.TabItem("Raw SQL Query"):
             gr.Markdown("**Danger Zone:** Run a raw `SELECT` query against the database. Use the Schema Explorer tab to see table/column names.")
-            # --- FIX: Updated placeholder to a useful, correct query ---
             raw_sql_input = gr.Textbox(
                 label="Raw SQL Query", 
-                placeholder="SELECT id, label, language, path FROM node WHERE id LIKE '%/c/en/dog' LIMIT 5",
+                placeholder="SELECT id, label, language FROM node WHERE id LIKE '%/c/en/dog' LIMIT 5",
                 lines=5
             )
             raw_query_btn = gr.Button("Run Raw SQL", variant="secondary")
@@ -220,6 +287,13 @@ with gr.Blocks(title="ConceptNet SQLite Explorer") as demo:
             schema_output = gr.Markdown("Schema will appear here...")
             
     # --- 4. Connect UI Elements to Functions ---
+    semantic_btn.click(
+        fn=get_semantic_profile,
+        inputs=[word_input],
+        outputs=[semantic_output],
+        api_name="get_semantic_profile"
+    )
+    
     query_btn.click(
         fn=run_query,
         inputs=[start_input, rel_input, end_input, limit_slider],
@@ -242,5 +316,4 @@ with gr.Blocks(title="ConceptNet SQLite Explorer") as demo:
     )
 
 if __name__ == "__main__":
-    # Removed experimental ssr_mode=False
     demo.launch(ssr_mode=False)