Update app.py

app.py

@@ -1,38 +1,33 @@
 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."
-    )
+import black
+import isort
+
+def format_code(code: str, apply_black=True, apply_isort=True) -> str:
+    try:
+        if apply_black:
+            code = black.format_str(code, mode=black.Mode())
+        if apply_isort:
+            code = isort.code(code)
+        return code
+    except Exception as e:
+        return f"⚠️ Error during formatting: {str(e)}"
+
+with gr.Blocks(title="🧰 Python Configurator") as demo:
+    gr.Markdown("# 🧰 Python Configurator\nFormat, lint, and inspect your Python configuration files easily.")
 
     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)
+        with gr.Column(scale=1):
+            input_code = gr.Code(label="Paste your Python config", language="python", lines=20)
+
+            with gr.Row():
+                use_black = gr.Checkbox(label="Apply Black Formatter", value=True)
+                use_isort = gr.Checkbox(label="Apply isort", value=True)
+
+            submit_btn = gr.Button("Format Code")
+
+        with gr.Column(scale=1):
+            output_code = gr.Code(label="Formatted Output", language="python", lines=20)
 
-    detect_btn.click(fn=detect_language, inputs=[code_input, token_input], outputs=output)
+    submit_btn.click(fn=format_code, inputs=[input_code, use_black, use_isort], outputs=output_code)
 
-# --- Launch App ---
 demo.launch()