Update app.py

app.py

@@ -67,7 +67,7 @@ d = {'GPT Commands': ['GOOGLE("question")', 'PYTHON(script.py)', 'MAKEFILE("cont
      }
 df = pd.DataFrame(data=d, dtype='string')
 commandTable = st.experimental_data_editor(df, use_container_width=True, num_rows='dynamic')
-st.markdown('### Upload/Download \n(note that this is a public server, uploaded files are visible to others)')
+st.markdown('### Upload/Download \n(note that this is a public server, uploaded files are visible to others and periodically deleted)')
 uploadFile = st.file_uploader('')
 cola, colb, colc = st.columns(3)
 with cola:
@@ -108,7 +108,7 @@ def askGPT(input):
     return resp
 
 def runCmd(flag):
-    if flag:
+    if flag and (not st.session_state['command'].startswith('rm '))):
         with st.spinner('Running command \'' + st.session_state['command'] + '\''):
             try:
                 p = subprocess.Popen(st.session_state['command'], shell=True, stderr=subprocess.STDOUT, stdout=subprocess.PIPE)
@@ -116,9 +116,10 @@ def runCmd(flag):
                 st.session_state['followupPrompt'] = 'Response: ' + p.communicate()[0].decode("utf-8")
             except subprocess.CalledProcessError as e:
                 st.session_state['followupPrompt'] = 'Response: ' + e.output.decode("utf-8")
+    elif st.session_state['command'].startswith('rm '):
+        st.session_state['followupPrompt'] = "Response: Rm command disabled on HuggingFace"
     else:
         st.session_state['followupPrompt'] = "Response: User rejected this command"
-        st.session_state['followup'], st.session_state['running'] = True, True
     st.session_state['followup'], st.session_state['running'] = True, True
     st.experimental_rerun()