Merge pull request #3 from humandotlearning/feature/hf-spaces-mcp

app.py

@@ -0,0 +1,72 @@
+import gradio as gr
+from typing import List, Optional
+import os
+import sys
+
+# Ensure src is in python path
+sys.path.append(os.path.join(os.path.dirname(__file__), "src"))
+
+from npi_mcp_server.tools import search_providers, get_provider_by_npi
+from npi_mcp_server.schemas import ProviderSummary, ProviderDetail
+
+async def search_providers_tool(query: str, state: str = None, taxonomy: str = None) -> List[dict]:
+    """
+    Search for healthcare providers via the NPI API.
+
+    Args:
+        query: Name (first/last) or organization name.
+        state: Two-letter state code.
+        taxonomy: Taxonomy code or description.
+
+    Returns:
+        List of ProviderSummary objects as dictionaries.
+    """
+    results = await search_providers(query, state, taxonomy)
+    return [r.model_dump() for r in results]
+
+async def get_provider_by_npi_tool(npi: str) -> Optional[dict]:
+    """
+    Retrieve details for a specific provider by NPI.
+
+    Args:
+        npi: 10-digit NPI string.
+
+    Returns:
+        ProviderDetail object as dictionary or None if not found.
+    """
+    result = await get_provider_by_npi(npi)
+    if result:
+        return result.model_dump()
+    return None
+
+with gr.Blocks() as demo:
+    gr.Markdown("# NPI Registry MCP Server")
+    gr.Markdown("This interface exposes NPI Registry tools as an MCP server.")
+
+    with gr.Tab("Search Providers"):
+        with gr.Row():
+            query_input = gr.Textbox(label="Query", placeholder="Name or Organization")
+            state_input = gr.Textbox(label="State", placeholder="2-letter code (e.g., CA)")
+            taxonomy_input = gr.Textbox(label="Taxonomy", placeholder="Taxonomy code")
+        search_btn = gr.Button("Search")
+        search_output = gr.JSON(label="Results")
+
+        search_btn.click(
+            fn=search_providers_tool,
+            inputs=[query_input, state_input, taxonomy_input],
+            outputs=search_output
+        )
+
+    with gr.Tab("Get Provider by NPI"):
+        npi_input = gr.Textbox(label="NPI", placeholder="10-digit NPI")
+        get_btn = gr.Button("Get Details")
+        get_output = gr.JSON(label="Provider Details")
+
+        get_btn.click(
+            fn=get_provider_by_npi_tool,
+            inputs=[npi_input],
+            outputs=get_output
+        )
+
+if __name__ == "__main__":
+    demo.launch(mcp_server=True)

pyproject.toml

@@ -9,6 +9,7 @@ dependencies = [
     "httpx>=0.24.0",
     "pydantic>=2.0.0",
     "mcp>=1.0.0",
+    "gradio[mcp]>=6.0.1",
 ]
 
 [project.optional-dependencies]
@@ -29,3 +30,10 @@ packages = ["src/npi_mcp_server"]
 pythonpath = ["src"]
 testpaths = ["tests"]
 asyncio_mode = "auto"
+
+[dependency-groups]
+dev = [
+    "pytest>=9.0.1",
+    "pytest-asyncio>=1.3.0",
+    "pytest-mock>=3.15.1",
+]

requirements.txt

@@ -0,0 +1,2 @@
+gradio[mcp]
+.