Update streamlit_app.py

streamlit_app.py

@@ -14,8 +14,14 @@ if st.button("Explain"):
         with st.spinner("Generating explanation..."):
             tokenizer, model, device = load_model()
             prompt = build_prompt(code_input)
+            st.subheader("🔍 Prompt Sent to Model")
+            st.code(prompt)
+
             explanation = generate_explanation(prompt, tokenizer, model, device)
-            st.subheader("📝 Explanation")
+            st.subheader("📝 Raw Output from Model")
+            st.code(explanation)
+
+            st.subheader("✅ Final Explanation")
             st.write(explanation.split("Explanation:")[-1].strip())
     else:
         st.warning("Please paste some code to explain.")