Redmind commited on
Commit
4a42972
·
verified ·
1 Parent(s): 84ab600

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -441,11 +441,13 @@ async def upload_document(
441
  values = (company_id,file_path, documentName, documentDescription, department, version,vectorDBflag, lastUpdated)
442
  cursor.execute(query, values)
443
  cnx.commit()
 
444
  cursor.close()
445
  cnx.close()
446
 
447
  logging.info(f"Document {documentName} uploaded successfully")
448
- return RedirectResponse(url="/knowledgebase", status_code=302)
 
449
  except mysql.connector.Error as err:
450
  logging.error(f"Database error: {err}")
451
  raise HTTPException(status_code=500, detail="Internal Server Error")
 
441
  values = (company_id,file_path, documentName, documentDescription, department, version,vectorDBflag, lastUpdated)
442
  cursor.execute(query, values)
443
  cnx.commit()
444
+ row_id=cursor.lastrowid
445
  cursor.close()
446
  cnx.close()
447
 
448
  logging.info(f"Document {documentName} uploaded successfully")
449
+ return JSONResponse(status_code=200, content={"message": "Data saved successfully", "row_id": row_id})
450
+ #return RedirectResponse(url="/knowledgebase", status_code=302)
451
  except mysql.connector.Error as err:
452
  logging.error(f"Database error: {err}")
453
  raise HTTPException(status_code=500, detail="Internal Server Error")