Update app.py

app.py

@@ -39,24 +39,20 @@ if 'history' not in st.session_state:
 def run_command(command):
     try:
         result = subprocess.run(command, shell=True, capture_output=True, text=True)
-        st.session_state.output = result.stdout
-        st.session_state.error = result.stderr
+        if result.stdout:
+            st.subheader("Output:")
+            st.markdown(f'<div class="output">{result.stdout}</div>', unsafe_allow_html=True)
+        if result.stderr:
+            st.subheader("Error:")
+            st.markdown(f'<div class="error">{result.stderr}</div>', unsafe_allow_html=True)
     except Exception as e:
-        st.session_state.error = str(e)
-        st.session_state.output = None
+        st.error(f"An error occurred: {e}")
 
-# Layout with three columns
-col1, col2, col3 = st.columns([1, 2, 1])
+# Layout with columns
+col1, col2 = st.columns([2, 1])
 
-# Command history display in the first column
+# Command input and button in the first column
 with col1:
-    st.subheader("Command History")
-    for cmd in reversed(st.session_state.history):
-        if st.button(f"Run: {cmd}", key=cmd):
-            run_command(cmd)
-
-# 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"):
         if command:
@@ -67,13 +63,12 @@ with col2:
         else:
             st.warning("Please enter a command to run.")
 
-# Output display in the third column
-with col3:
-    st.subheader("Output:")
-    if 'output' in st.session_state and st.session_state.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.markdown(f'<div class="error">{st.session_state.error}</div>', unsafe_allow_html=True)
+# Command history display in the second column
+with col2:
+    st.subheader("Command History")
+    for cmd in reversed(st.session_state.history):
+        if st.button(f"Run: {cmd}", key=cmd):
+            run_command(cmd)
 
 # Optional file upload (if needed)
 uploaded_file = st.file_uploader("Upload a file to process with command:", type=["txt", "csv"])