Update app.py

app.py

@@ -199,15 +199,9 @@ def build_knowledge_graph(input_text):
             "knowledge_graph": None
         }
 
-# Create wrapper function for proper JSON formatting in UI
-def build_knowledge_graph_ui(input_text):
-    """Wrapper function that returns JSON string for UI display."""
-    result = build_knowledge_graph(input_text)
-    return json.dumps(result, indent=2, ensure_ascii=False)
-
 # Create Gradio interface
 demo = gr.Interface(
-    fn=build_knowledge_graph_ui,
+    fn=build_knowledge_graph,
     inputs=gr.Textbox(
         label="Text or URL Input",
         placeholder="Enter text to analyze or a web URL (e.g., https://example.com)",
@@ -235,17 +229,4 @@ demo = gr.Interface(
     theme=gr.themes.Soft()
 )
 
-# Register MCP tools
-demo.mcp.register_tool(
-    "build_knowledge_graph",
-    build_knowledge_graph,
-    "Build knowledge graph from text or URL",
-    {
-        "input_text": {
-            "type": "string",
-            "description": "Text content or URL to analyze and extract knowledge graph from"
-        }
-    }
-)
-
 demo.launch(mcp_server=True)