frontend

HFSpace/app.py

@@ -0,0 +1,108 @@
+"""
+Hugging Face Spaces entry point.
+Deploy this file as app.py in your HF Space (SDK: gradio).
+"""
+import os
+import json
+import gradio as gr
+from app.models.analyzer import analyze_contract
+
+# ------------------------------------------------------------------ helpers --
+
+def _fmt_vulns(vulns: list) -> str:
+    if not vulns:
+        return "✅ No vulnerabilities detected."
+    icons = {"HIGH": "🚨", "MEDIUM": "⚠️", "LOW": "ℹ️"}
+    lines = []
+    for v in vulns:
+        icon = icons.get(v["severity"], "•")
+        name = v["name"].replace("_", " ").title()
+        lines.append(f"{icon} **{name}** [{v['severity']}]")
+        lines.append(f"   {v['description']}")
+        if v.get("line_numbers"):
+            lines.append(f"   Lines: {', '.join(map(str, v['line_numbers'][:8]))}")
+        if v.get("recommendation"):
+            lines.append(f"   💡 {v['recommendation']}")
+        lines.append("")
+    return "\n".join(lines)
+
+
+def audit(solidity_code: str):
+    if not solidity_code.strip():
+        return "⚠️ Please paste some Solidity code.", "{}", ""
+
+    result = analyze_contract(solidity_code)
+
+    # Summary block
+    risk_icons = {"CRITICAL": "🚨", "MEDIUM": "⚠️", "LOW": "ℹ️", "SAFE": "✅"}
+    icon = risk_icons.get(result["risk_level"], "•")
+    summary = (
+        f"## {icon} Risk Level: {result['risk_level']}\n\n"
+        f"- **Solidity Version:** {result['solidity_version']}\n"
+        f"- **Lines:** {result['total_lines']}\n"
+        f"- **Severity Score:** {result['severity_score']}\n"
+        f"- **Analysis type:** {result['analysis_type']}\n\n"
+        f"| HIGH | MEDIUM | LOW |\n|------|--------|-----|\n"
+        f"| {result['high_count']} | {result['medium_count']} | {result['low_count']} |"
+    )
+
+    vuln_text = _fmt_vulns(result["vulnerabilities"])
+    llm_text = result.get("llm_analysis") or "_LLM model not loaded – showing pattern analysis only._"
+
+    return summary, vuln_text, llm_text
+
+
+# ------------------------------------------------------------------- UI ------
+
+EXAMPLES = [
+    [
+        """pragma solidity ^0.4.24;
+contract VulnerableBank {
+    mapping(address => uint256) public balances;
+    function withdraw(uint256 _amount) public {
+        require(balances[msg.sender] >= _amount);
+        msg.sender.call.value(_amount)();
+        balances[msg.sender] -= _amount;
+    }
+}"""
+    ],
+    [
+        """// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.0;
+import "@openzeppelin/contracts/security/ReentrancyGuard.sol";
+import "@openzeppelin/contracts/access/Ownable.sol";
+contract SafeVault is ReentrancyGuard, Ownable {
+    mapping(address => uint256) private balances;
+    function withdraw(uint256 amount) external nonReentrant {
+        require(balances[msg.sender] >= amount);
+        balances[msg.sender] -= amount;
+        (bool ok,) = payable(msg.sender).call{value: amount}("");
+        require(ok);
+    }
+}"""
+    ],
+]
+
+with gr.Blocks(title="🛡️ Smart Contract Auditor", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🛡️ Smart Contract Security Auditor\nPaste Solidity code to detect security vulnerabilities using LLM + pattern analysis.")
+
+    with gr.Row():
+        with gr.Column():
+            code_input = gr.Code(label="Solidity Contract", language="javascript", lines=20)
+            analyze_btn = gr.Button("🔍 Analyze", variant="primary")
+
+        with gr.Column():
+            summary_out = gr.Markdown(label="Summary")
+            vuln_out    = gr.Markdown(label="Vulnerabilities")
+            llm_out     = gr.Markdown(label="LLM Analysis")
+
+    gr.Examples(examples=EXAMPLES, inputs=[code_input])
+
+    analyze_btn.click(
+        fn=audit,
+        inputs=[code_input],
+        outputs=[summary_out, vuln_out, llm_out],
+    )
+
+if __name__ == "__main__":
+    demo.launch()

HFSpace/requirements.txt

@@ -0,0 +1,11 @@
+gradio>=4.26.0
+torch>=2.2.0
+transformers>=4.40.0
+peft>=0.10.0
+accelerate>=0.30.0
+huggingface-hub>=0.22.0
+sentencepiece>=0.2.0
+fastapi>=0.110.0
+uvicorn
+pydantic>=2.0
+pydantic-settings>=2.0