Spaces:

SayedAi
/

lpd-2

Runtime error

lpd-2 / app.py

coder-black-mamba

initial commit

dd5050e about 2 months ago

No virus

3.25 kB

	from ultralytics import YOLO
	import easyocr
	import cv2
	from flask import Flask, render_template, request, redirect, url_for,jsonify
	import os
	import time

	app = Flask(__name__)

	# Set upload folder
	UPLOAD_FOLDER = 'static/uploads'
	app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER




	@app.route('/', methods=['GET', 'POST'])
	def upload_file():
	if request.method == 'POST':
	# Get the uploaded file
	uploaded_file = request.files['image']
	# Check if a file is selected
	if uploaded_file.filename != '':
	# Save the file
	filename = uploaded_file.filename
	filepath = os.path.join(app.config['UPLOAD_FOLDER'], "input.pdf")
	uploaded_file.save(filepath)
	# Redirect to success page (you can modify this for further processing)
	return redirect('/result')
	return render_template('index.html')

	@app.route('/result/')
	def upload_success():
	# file name is the file name that some one uploaded
	# so the image path will be ./static/uploads/<filename>
	# call your model with keras tf of sklearn
	try:
	model = YOLO('./yolov8-custom.pt')
	confidence = 0.6 # 0.0-1.0 # render the message that you wanted to show in message variable
	results = model.predict(source = f"./static/uploads/input.jpg", save = True, show = False, conf = confidence)
	print(results)

	img_main = cv2.imread("./static/uploads/input.jpg")
	r = results[0]
	box = r.boxes[0]
	[left, top, right, bottom] = box.xyxy[0]
	left = int(left)
	top = int(top)
	right = int(right)
	bottom = int(bottom)
	cropped_img = img_main[top+1:bottom-1, left+1:right-1]
	output_path = "./static/uploads/processed.jpg"
	cv2.imwrite(output_path, cropped_img)
	print(f"Largest image saved at {output_path}")
	time.sleep(2)
	print("detection and cropping is successfull. ")
	return redirect('/final_result/')
	except Exception as e :
	print(e)
	return render_template('error.html')

	@app.route('/final_result/')
	def final_result():
	try:
	print("now easy ocr part.")
	reader = easyocr.Reader(['bn'], gpu = False)
	result = reader.readtext("./static/uploads/processed.jpg", detail = 0, paragraph = True)
	print(result)
	return render_template('index.html', message=result[0])
	except Exception as e :
	return render_template('error.html')
	@app.route("/download")
	def download():
	import requests

	# Replace with the URL of the file on GitHub (use raw content URL)
	url = "https://raw.githubusercontent.com/user/repository/branch/file.txt"

	# Send a GET request to download the file
	response = requests.get(url)

	# Check for successful response
	if response.status_code == 200:
	# Get the filename from the URL (optional)
	filename = url.split("/")[-1]

	# Save the content to a file
	with open(filename, "wb") as f:
	f.write(response.content)

	print(f"Downloaded {filename} successfully!")
	else:
	print(f"Failed to download file. Status code: {response.status_code}")


	if __name__ == '__main__':
	app.run(debug=True)