def upload_img(): print("image upload") result = 'static/result' if not gfile.Exists(result): os.mkdir(result) shutil.rmtree(result) if request.method == 'POST' or request.method == 'GET': print(request.method) # check if the post request has the file part if 'file' not in request.files: print('No file part') return redirect(request.url) file = request.files['file'] print(file.filename) # if user does not select file, browser also # submit a empty part without filename if file.filename == '': print('No selected file') return redirect(request.url) if file and allowed_file(file.filename): filename = secure_filename(file.filename) file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename)) inputloc = os.path.join(app.config['UPLOAD_FOLDER'], filename) #outputloc = '/home/vinayak/todo-api/out' recommend(inputloc, extracted_features) os.remove(inputloc) #label = label1.replace("\n", "") image_path = "/result" image_list = [ os.path.join(image_path, file) for file in os.listdir(result) if not file.startswith('.') ] images = { 'image0': image_list[0], # 'image1':image_list[1], # 'image2':image_list[2], # 'image3':image_list[3], # 'image4':image_list[4], # 'image5':image_list[5], # 'image6':image_list[6], # 'image7':image_list[7], # 'image8':image_list[8] } return jsonify(images)
def main(): if request.method == 'POST': upload = 'static/Upload' if not gfile.Exists(upload): os.mkdir(upload) print( '_________________________upload directory_created________________________________' ) app.config["upload"] = upload result = 'static/result' if not gfile.Exists(result): os.mkdir(result) # print(request.method) confidence = request.form['conf'] confidence = int(confidence) mode = request.form['mod'] mode = int(mode) # check if the post request has the file part if 'query_img' not in request.files: print('No file part') return redirect(request.url) file = request.files['query_img'] print("Given File:- ", file.filename) # if user does not select file, browser also # submit a empty part without filename if file.filename == '': print('No selected file') return redirect(request.url) if file and file.filename.split('.')[-1] in ALLOWED_EXTENSIONS: filename = secure_filename(file.filename) file.save(os.path.join(app.config['upload'], file.filename)) inputloc = os.path.join(app.config['upload'], file.filename) success, rows = database.check_img(filename) if success == True: result_images, total_results_img = database.fetch_data( filename) final_result_images = result_images uploaded_img_path = os.path.join(app.config['upload'], filename)[7:] return render_template('index.html', query_path=uploaded_img_path, result_images=final_result_images, total_images=int(rows[1]), total_results=int(rows[0])) else: if mode == 1: _,total_results, total_images = search.recommend(inputloc, all_extracted_features_design, all_extracted_features_d_c,\ 'neighbor_all', filename, confidence, detection_graph, category_index) result_images, total_results_img = database.fetch_data( filename) final_result_images = result_images uploaded_img_path = os.path.join(app.config['upload'], filename)[7:] return render_template('index.html', query_path=uploaded_img_path, result_images=final_result_images, total_images=total_images, total_results=total_results) elif mode == 2: _,total_results, total_images = search.recommend_bracelet(inputloc, bracelate_extracted_features_design, bracelate_extracted_features_d_c,\ 'neighbor_bracelate', filename, confidence, bracelet_, category_index_Bracelet_D_C) result_images, total_results_img = database.fetch_data( filename) final_result_images = result_images uploaded_img_path = os.path.join(app.config['upload'], filename)[7:] return render_template('index.html', query_path=uploaded_img_path, result_images=final_result_images, total_images=total_images, total_results=total_results) elif mode == 3: _,total_results, total_images = search.recommend_earring(inputloc, earring_extracted_features_design,\ 'neighbor_earring', filename, confidence, earring_, category_index_Earring_D) result_images, total_results_img = database.fetch_data( filename) final_result_images = result_images uploaded_img_path = os.path.join(app.config['upload'], filename)[7:] return render_template('index.html', query_path=uploaded_img_path, result_images=final_result_images, total_images=total_images, total_results=total_results) elif mode == 4: _,total_results, total_images = search.recommend_necklace(inputloc, necklace_extracted_features_design,\ 'neighbor_necklace', filename, confidence, necklace_, category_index_Necklace_D) result_images, total_results_img = database.fetch_data( filename) final_result_images = result_images uploaded_img_path = os.path.join(app.config['upload'], filename)[7:] return render_template('index.html', query_path=uploaded_img_path, result_images=final_result_images, total_images=total_images, total_results=total_results) elif mode == 5: _,total_results, total_images = search.recommend_pendant(inputloc, pendant_extracted_features_design, pendant_extracted_features_d_c,\ 'neighbor_pendant', filename, confidence, pendant_, category_index_Pendant_D_C) result_images, total_results_img = database.fetch_data( filename) final_result_images = result_images uploaded_img_path = os.path.join(app.config['upload'], filename)[7:] return render_template('index.html', query_path=uploaded_img_path, result_images=final_result_images, total_images=total_images, total_results=total_results) elif mode == 6: _,total_results, total_images = search.recommend_ring(inputloc, rings_extracted_features_design, rings_extracted_features_d_c,\ rings_extracted_features_d_m, 'neighbor_ring', filename, confidence, ring_, category_index_Ring_D_C_M) result_images, total_results_img = database.fetch_data( filename) final_result_images = result_images uploaded_img_path = os.path.join(app.config['upload'], filename)[7:] return render_template('index.html', query_path=uploaded_img_path, result_images=final_result_images, total_images=total_images, total_results=total_results) else: return render_template('index.html')