Update app.py

app.py

@@ -1,44 +1,43 @@
-import os
-import json
+import os, json
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# Load tool prompts
-with open("system_prompts.json", "r", encoding="utf-8") as f:
+# Load prompts
+with open("system_prompts.json", encoding="utf-8") as f:
     SYSTEM_PROMPTS = json.load(f)
-TOOLS = list(SYSTEM_PROMPTS.keys())
-
-# Set these environment variables in your HF Space or local env
-HF_TOKEN = os.environ["HF_TOKEN"]
-HF_MODEL = os.environ.get("HF_MODEL")
 
 # Initialize HF InferenceClient
-client = InferenceClient(model=HF_MODEL, token=HF_TOKEN)
+client = InferenceClient(model=os.getenv("HF_MODEL"), token=os.getenv("HF_TOKEN"))
 
+# Common runner
 def run_model(prompt: str, text: str) -> str:
-    """
-    Uses InferenceClient.chat.completions to call mnodel.
-    """
     resp = client.chat.completions.create(
-        messages=[
-            {"role": "system", "content": prompt},
-            {"role": "user", "content": text}
-        ],
-        max_tokens=512,
-        temperature=0.3
+        messages=[{"role":"system","content":prompt},{"role":"user","content":text}],
+        max_tokens=512, temperature=0.3
     )
     return resp.choices[0].message.content
 
-def ecoling_tool(tool: str, text: str) -> str:
-    return run_model(SYSTEM_PROMPTS[tool], text)
+# Dynamically create tool functions
+def make_tool_fn(prompt):
+    return lambda text: run_model(prompt, text)
+
+tool_fns = {
+    name.lower().replace(" ", "_"): make_tool_fn(prompt)
+    for name, prompt in SYSTEM_PROMPTS.items()
+}
 
-demo = gr.Interface(
-    fn=ecoling_tool,
-    inputs=[gr.Dropdown(TOOLS, label="Tool"), gr.Textbox(lines=8, label="Input Text")],
-    outputs=[gr.Textbox(label="Response")],
-    title="🌱 Ecolinguistic MCP Server",
-    description="UI + MCP server"
-)
+# Build Gradio interface
+demo = gr.Blocks()
+with demo:
+    gr.Markdown("# 🌱 Ecolinguistic MCP Server")
+    tabs = gr.Tabs()
+    for badge, fn in tool_fns.items():
+        with tabs:
+            with gr.Tab(badge.replace("_", " ").title()):
+                inp = gr.Textbox(lines=5, label="Text")
+                out = gr.Textbox(label="Response")
+                gr.Button("Run").click(fn, inp, out)
 
+# Expose as MCP server
 if __name__ == "__main__":
-    demo.launch(mcp_server=True,share=True)
+    demo.launch(mcp_server=True, share=True)