OpenEnv Contributor commited on
Commit
e90b82d
·
1 Parent(s): 14d067c

fix(deploy): ensure HTTP response returns 200 explicitly on /reset endpoint for HF spaces

Browse files
Files changed (2) hide show
  1. dummy_server.py +10 -4
  2. server/app.py +10 -4
dummy_server.py CHANGED
@@ -8,10 +8,16 @@ class DummyServer(BaseHTTPRequestHandler):
8
  self.wfile.write(b"OK")
9
 
10
  def do_POST(self):
11
- self.send_response(200)
12
- self.send_header('Content-type', 'text/plain')
13
- self.end_headers()
14
- self.wfile.write(b"OK")
 
 
 
 
 
 
15
 
16
  if __name__ == '__main__':
17
  server = HTTPServer(('0.0.0.0', 7860), DummyServer)
 
8
  self.wfile.write(b"OK")
9
 
10
  def do_POST(self):
11
+ if self.path == '/reset':
12
+ self.send_response(200)
13
+ self.send_header('Content-type', 'application/json')
14
+ self.end_headers()
15
+ self.wfile.write(b'{"status": "ok"}')
16
+ else:
17
+ self.send_response(200)
18
+ self.send_header('Content-type', 'text/plain')
19
+ self.end_headers()
20
+ self.wfile.write(b"OK")
21
 
22
  if __name__ == '__main__':
23
  server = HTTPServer(('0.0.0.0', 7860), DummyServer)
server/app.py CHANGED
@@ -10,10 +10,16 @@ class DummyServer(BaseHTTPRequestHandler):
10
  self.wfile.write(b"OK")
11
 
12
  def do_POST(self):
13
- self.send_response(200)
14
- self.send_header('Content-type', 'text/plain')
15
- self.end_headers()
16
- self.wfile.write(b"OK")
 
 
 
 
 
 
17
 
18
  def main():
19
  parser = argparse.ArgumentParser()
 
10
  self.wfile.write(b"OK")
11
 
12
  def do_POST(self):
13
+ if self.path == '/reset':
14
+ self.send_response(200)
15
+ self.send_header('Content-type', 'application/json')
16
+ self.end_headers()
17
+ self.wfile.write(b'{"status": "ok"}')
18
+ else:
19
+ self.send_response(200)
20
+ self.send_header('Content-type', 'text/plain')
21
+ self.end_headers()
22
+ self.wfile.write(b"OK")
23
 
24
  def main():
25
  parser = argparse.ArgumentParser()