Stereo0001 commited on
Commit
6b1a6c1
·
verified ·
1 Parent(s): 99b8699

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -101,11 +101,15 @@ def execute_jupyter_agent(
101
  user_input, files, message_history, request: gr.Request
102
  ):
103
 
 
 
 
 
104
  if request.session_hash not in SANDBOXES:
105
  SANDBOXES[request.session_hash] = Sandbox.create(timeout=SANDBOX_TIMEOUT)
106
  sbx = SANDBOXES[request.session_hash]
107
  if f'{user_input[0]}'!='zhen:':
108
- return 'something erro!'
109
  else:
110
  user_input=user_input[2:]
111
  save_dir = os.path.join(TMP_DIR, request.session_hash)
 
101
  user_input, files, message_history, request: gr.Request
102
  ):
103
 
104
+ print(f'{user_input[0]}')
105
+ if not user_input.startswith("zhen:"):
106
+ return
107
+ user_input = user_input[5:].lstrip()
108
  if request.session_hash not in SANDBOXES:
109
  SANDBOXES[request.session_hash] = Sandbox.create(timeout=SANDBOX_TIMEOUT)
110
  sbx = SANDBOXES[request.session_hash]
111
  if f'{user_input[0]}'!='zhen:':
112
+ break
113
  else:
114
  user_input=user_input[2:]
115
  save_dir = os.path.join(TMP_DIR, request.session_hash)