Create app.py

app.py

@@ -0,0 +1,38 @@
+import gradio as gr
+import requests
+
+# --- Inference Logic ---
+def detect_language(code_snippet, token):
+    headers = {"Authorization": f"Bearer {token}"} if token else {}
+    response = requests.post(
+        "https://api-inference.huggingface.co/models/huggingface/CodeBERTa-language-id",
+        headers=headers,
+        json={"inputs": code_snippet}
+    )
+    if response.status_code == 200:
+        prediction = response.json()[0]['label']
+        return f"Detected Language: {prediction}"
+    else:
+        return f"Error: {response.text}"
+
+# --- Gradio Interface ---
+with gr.Blocks(title="SecuPromptMCP - Language Detection") as demo:
+    with gr.Row():
+        gr.Markdown("# 🔐 SecuPromptMCP")
+    gr.Markdown(
+        "This tool uses the `huggingface/CodeBERTa-language-id` model to detect the programming language from code snippets. "
+        "Sign in with your Hugging Face account to use the Inference API. Perfect for pentesters, devs, and agents on the edge."
+    )
+
+    with gr.Row():
+        with gr.Column():
+            code_input = gr.Code(label="Paste your code snippet here")
+            token_input = gr.Textbox(label="Hugging Face Access Token (optional)", type="password", placeholder="hf_...")
+            detect_btn = gr.Button("Detect Language")
+        with gr.Column():
+            output = gr.Textbox(label="Result", lines=3)
+
+    detect_btn.click(fn=detect_language, inputs=[code_input, token_input], outputs=output)
+
+# --- Launch App ---
+demo.launch()