mcma_malware / app.py
zeltera's picture
Upload app.py
26c1f56 verified
raw
history blame contribute delete
833 Bytes
import gradio as gr
import tempfile
from inference.analyze import analyze
from yara.generate import generate_yara
def analyze_file(file):
if file is None:
return None, None
path = file.name
prompt = f"Suspicious file uploaded: {path}"
result = analyze(prompt)
yara = generate_yara(result)
return result, yara
with gr.Blocks() as demo:
gr.Markdown("# 🛡️ MCMA – Malware Static Analyzer")
file = gr.File(label="Drag & drop malware sample (static analysis only)")
json_out = gr.JSON(label="Analysis Result")
# CHANGED: language="yara" -> language="c"
yara_out = gr.Code(label="Generated YARA Rule", language="c")
btn = gr.Button("Analyze")
btn.click(analyze_file, inputs=file, outputs=[json_out, yara_out])
demo.launch()