Update app.py

app.py

@@ -69,11 +69,10 @@ with col2:
 
 # Output display in the third column
 with col3:
+     st.subheader("Output:")
     if 'output' in st.session_state and st.session_state.output:
-        st.subheader("Output:")
         st.markdown(f'<div class="output">{st.session_state.output}</div>', unsafe_allow_html=True)
     if 'error' in st.session_state and st.session_state.error:
-        st.subheader("Error:")
         st.markdown(f'<div class="error">{st.session_state.error}</div>', unsafe_allow_html=True)
 
 # Optional file upload (if needed)