Update app.py

app.py

@@ -39,12 +39,11 @@ if 'history' not in st.session_state:
 def run_command(command):
     try:
         result = subprocess.run(command, shell=True, capture_output=True, text=True)
-        if result.stdout:
-            st.markdown(f'<div class="output">{result.stdout}</div>', unsafe_allow_html=True)
-        if result.stderr:
-            st.markdown(f'<div class="error">{result.stderr}</div>', unsafe_allow_html=True)
+        st.session_state.output = result.stdout
+        st.session_state.error = result.stderr
     except Exception as e:
-        st.error(f"An error occurred: {e}")
+        st.session_state.error = str(e)
+        st.session_state.output = None
 
 # Layout with three columns
 col1, col2, col3 = st.columns([1, 2, 1])
@@ -56,7 +55,7 @@ with col1:
         if st.button(f"Run: {cmd}", key=cmd):
             run_command(cmd)
 
-# Command input and output in the second column
+# Command input in the second column
 with col2:
     command = st.text_area("Enter terminal command", placeholder="e.g., ls, pwd, echo 'Hello World'", height=100)
     if st.button("Run Command"):
@@ -68,6 +67,15 @@ with col2:
         else:
             st.warning("Please enter a command to run.")
 
+# Output display in the third column
+with col3:
+    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)
 uploaded_file = st.file_uploader("Upload a file to process with command:", type=["txt", "csv"])
 if uploaded_file: