mitulagr2 commited on
Commit
713c080
1 Parent(s): 7ae7aad

fix upload

Browse files
Files changed (1) hide show
  1. app/main.py +4 -9
app/main.py CHANGED
@@ -54,22 +54,17 @@ async def process_input(text: str):
54
  return StreamingResponse(astreamer(generator), media_type='text/event-stream')
55
 
56
  async def parse_body(request: Request):
57
- data = await request.body()
58
- print("TYPE: ")
59
- print(type(data))
60
- print("DATA: ")
61
- print(data)
62
  return data
63
 
64
  @app.post("/upload")
65
- def upload(data = Depends(parse_body)):
66
- if data.get('bytes', None) and data.get('filename', None):
67
  try:
68
- filename = data['filename']
69
  print("Filename: " + filename)
70
  path = f"{files_dir}/{filename}"
71
  with open(path, "wb") as f:
72
- f.write(data['bytes'])
73
  session_assistant.ingest(files_dir)
74
  pathlib.Path(path).unlink()
75
  except Exception as e:
 
54
  return StreamingResponse(astreamer(generator), media_type='text/event-stream')
55
 
56
  async def parse_body(request: Request):
57
+ data: bytes = await request.body()
 
 
 
 
58
  return data
59
 
60
  @app.post("/upload")
61
+ def upload(filename: str, data: bytes = Depends(parse_body)):
62
+ if data:
63
  try:
 
64
  print("Filename: " + filename)
65
  path = f"{files_dir}/{filename}"
66
  with open(path, "wb") as f:
67
+ f.write(data)
68
  session_assistant.ingest(files_dir)
69
  pathlib.Path(path).unlink()
70
  except Exception as e: