Update app.py

app.py

@@ -1,34 +1,15 @@
 import gradio as gr
 import subprocess
-from fastapi import FastAPI
-from pydantic import BaseModel
 
-# Create FastAPI backend
-fastapi_app = FastAPI(title="MCP Linux Command Runner")
+def run_command(cmd: str) -> str:
+    """
+    Run a Linux command and return stdout or stderr.
 
-class Command(BaseModel):
-    cmd: str
-
-@fastapi_app.post("/run")
-def run_command(command: Command):
-    try:
-        result = subprocess.run(
-            command.cmd,
-            shell=True,
-            capture_output=True,
-            text=True,
-            timeout=10
-        )
-        return {
-            "stdout": result.stdout,
-            "stderr": result.stderr,
-            "returncode": result.returncode
-        }
-    except Exception as e:
-        return {"error": str(e)}
-
-# Gradio UI
-def run_cmd_ui(cmd):
+    Args:
+        cmd (str): Linux command
+    Returns:
+        str: Command output (stdout or stderr)
+    """
     try:
         result = subprocess.run(
             cmd,
@@ -37,17 +18,20 @@ def run_cmd_ui(cmd):
             text=True,
             timeout=10
         )
-        return result.stdout if result.stdout else result.stderr
+        output = result.stdout if result.stdout else result.stderr
+        return output.strip()
     except Exception as e:
-        return str(e)
+        return f"Error: {str(e)}"
 
+# Gradio interface
 demo = gr.Interface(
-    fn=run_cmd_ui,
+    fn=run_command,
     inputs=gr.Textbox(label="Enter Linux Command"),
     outputs=gr.Textbox(label="Output", lines=15),
-    title="Linux MCP Runner",
-    description="Run Linux commands inside Hugging Face Space (sandboxed)"
+    title="MCP Linux Shell",
+    description="Run Linux commands inside a Hugging Face Space (MCP-compatible)"
 )
 
-# Mount FastAPI inside Gradio app (so HF initializes Gradio correctly)
-demo = gr.mount_gradio_app(fastapi_app, demo, path="/")
+if __name__ == "__main__":
+    # 🚨 Key line: enable MCP server
+    demo.launch(mcp_server=True)