WebashalarForML commited on
Commit
241e962
·
verified ·
1 Parent(s): ccb3d80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -258,25 +258,27 @@ def create_app():
258
  thread.start()
259
  return "OK", 200
260
 
261
- @flask_app.route("/upload", methods=["POST","GET"])
262
  def upload():
263
-
264
  if request.method == 'POST':
265
- file = request.files['file']
266
  if not file:
267
  return "No file uploaded", 400
268
  if file and file.filename.endswith('.db'):
269
- db_path = os.path.join(app.config['UPLOAD_FOLDER'], 'uploaded.db')
 
270
  file.save(db_path)
271
-
272
  # Convert the file path to an absolute path and reinitialize the agent_app
273
  abs_file_path = os.path.abspath(db_path)
274
  global agent_app
275
  agent_app = create_agent_app(abs_file_path)
276
- print("agent app",agent_app)
277
- print("[INFO_PRINT]: Database file '{file.filename}' uploaded and loaded.")
 
278
  socketio.emit("log", {"message": f"[INFO]: Database file '{file.filename}' uploaded and loaded."})
279
  return redirect(url_for("index")) # Go back to index page
 
280
  return render_template("upload.html")
281
 
282
  return flask_app, socketio
 
258
  thread.start()
259
  return "OK", 200
260
 
261
+ @flask_app.route("/upload", methods=["POST", "GET"])
262
  def upload():
 
263
  if request.method == 'POST':
264
+ file = request.files.get('file')
265
  if not file:
266
  return "No file uploaded", 400
267
  if file and file.filename.endswith('.db'):
268
+ # Use flask_app.config instead of app.config
269
+ db_path = os.path.join(flask_app.config['UPLOAD_FOLDER'], 'uploaded.db')
270
  file.save(db_path)
271
+
272
  # Convert the file path to an absolute path and reinitialize the agent_app
273
  abs_file_path = os.path.abspath(db_path)
274
  global agent_app
275
  agent_app = create_agent_app(abs_file_path)
276
+
277
+ # Use an f-string properly to log the uploaded file name
278
+ print(f"[INFO_PRINT]: Database file '{file.filename}' uploaded and loaded.")
279
  socketio.emit("log", {"message": f"[INFO]: Database file '{file.filename}' uploaded and loaded."})
280
  return redirect(url_for("index")) # Go back to index page
281
+ # For GET requests, simply render the upload form.
282
  return render_template("upload.html")
283
 
284
  return flask_app, socketio