import streamlit as st
from model_utils import load_model, generate_deceptive_code
from binary_utils import binary_to_text, text_to_binary
from deception_engine import inject_deception
from model_utils import load_model, generate_deceptive_code


st.title("🛡️ Deceptive SoC Clone Generator (GEN AI)")

binary_input = st.text_area("📥 Enter Binary from Primary SoC", height=150)

if binary_input:
    st.subheader("🔍 Step 1: Decode Original Binary")
    original_text = binary_to_text(binary_input)
    st.code(original_text)

    if st.button("🧠 Generate Deceptive Clone"):
        with st.spinner("Loading model and crafting deception..."):
            tokenizer, model = load_model()
            raw_generated = generate_deceptive_code(tokenizer, model, original_text)
            deceptive_code = inject_deception(raw_generated)

        st.subheader("🎭 Step 2: Deceptive Code Output")
        st.code(deceptive_code, language='c')

        st.subheader("📦 Step 3: Reconvert to Binary (for Secondary SoC)")
        deceptive_binary = text_to_binary(deceptive_code)
        st.code(deceptive_binary)