Update app.py

app.py

@@ -1,24 +1,26 @@
 import streamlit as st
-from model_utils import load_model, generate_code
+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
 
-st.title("🧠 Binary Clone Generator using GenAI")
+st.title("🛡️ Deceptive SoC Clone Generator (GEN AI)")
 
-binary_input = st.text_area("Enter Binary Input (from primary SoC)", height=200)
+binary_input = st.text_area("📥 Enter Binary from Primary SoC", height=150)
 
 if binary_input:
-    st.subheader("🔍 Converting Binary to Text")
+    st.subheader("🔍 Step 1: Decode Original Binary")
     original_text = binary_to_text(binary_input)
-    st.code(original_text, language='text')
+    st.code(original_text)
 
-    if st.button("🛠 Generate Replica for Secondary SoC"):
-        st.write("⏳ Loading model and generating...")
-        tokenizer, model = load_model()
-        gen_code = generate_code(tokenizer, model, 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("🎯 Generated Code/Text")
-        st.code(gen_code, language='python')
+        st.subheader("🎭 Step 2: Deceptive Code Output")
+        st.code(deceptive_code, language='c')
 
-        st.subheader("📦 Re-converted to Binary for SoC")
-        generated_binary = text_to_binary(gen_code)
-        st.code(generated_binary, language='text')
+        st.subheader("📦 Step 3: Reconvert to Binary (for Secondary SoC)")
+        deceptive_binary = text_to_binary(deceptive_code)
+        st.code(deceptive_binary)