Update main.py

app/main.py

@@ -54,17 +54,18 @@ async def process_input(text: str):
     return StreamingResponse(astreamer(generator), media_type='text/event-stream')
 
 async def parse_body(request: Request):
-    data: bytes = await request.body()
+    data = await request.body()
     return data
 
 @app.post("/upload")
-def upload(filename: str, data: bytes = Depends(parse_body)):
-    print("Filename: " + filename)
-    if data:
+def upload(data = Depends(parse_body)):
+    if data.get('bytes', None) and data.get('filename', None):
         try:
+            filename = data['filename']
+            print("Filename: " + filename)
             path = f"{files_dir}/{filename}"
             with open(path, "wb") as f:
-                f.write(data)
+                f.write(data['bytes'])
             session_assistant.ingest(files_dir)
             pathlib.Path(path).unlink()
         except Exception as e: