Spaces:

SayedAi
/

lpd-2

Runtime error

App Files Files Community

coder-black-mamba commited on Apr 18

Commit

dd5050e

•

1 Parent(s): 4ea84da

initial commit

Browse files

Files changed (6) hide show

Dockerfile +11 -0
app.py +97 -0
static/uploads/input.jpg +0 -0
static/uploads/processed.jpg +0 -0
templates/error.html +19 -0
templates/index.html +40 -0

Dockerfile ADDED Viewed

	@@ -0,0 +1,11 @@

+FROM python:3.9
+WORKDIR /code
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+COPY . .
+CMD ["gunicorn", "-b","0.0.0.0:7860","app:app",]

app.py ADDED Viewed

	@@ -0,0 +1,97 @@

+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)

static/uploads/input.jpg ADDED Viewed

static/uploads/processed.jpg ADDED Viewed

templates/error.html ADDED Viewed

	@@ -0,0 +1,19 @@

+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <title>Lost in the Interwebs?</title>
+    <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.3/dist/css/bootstrap.min.css" rel="stylesheet"
+        integrity="sha384-QWTKZyjpPEjISv5WaRU9OFeRpok6YctnYmDr5pNlyT2bRjXh0JMhjY6hW+ALEwIH" crossorigin="anonymous">
+</head>
+<body>
+    <div class="container text-center">
+        <h1 class="mt-5 py-3 text-danger bg-secondary">Opps</h1>
+        <p class="py-5">The page you requested seems to have gotten kidnapped by gremlins. Don't worry,  gremlins are pretty lousy at hiding things. Maybe try searching for it, or head back to the homepage using the button below?  </p>
+        <a cla href="/">Back to Home</a>
+    </div>
+    </body>
+</html>

templates/index.html ADDED Viewed

	@@ -0,0 +1,40 @@

+<!DOCTYPE html>
+<html>
+<head>
+    <title>Upload Image</title>
+    <link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.3/dist/css/bootstrap.min.css" rel="stylesheet"
+        integrity="sha384-QWTKZyjpPEjISv5WaRU9OFeRpok6YctnYmDr5pNlyT2bRjXh0JMhjY6hW+ALEwIH" crossorigin="anonymous">
+    <style>
+    </style>
+</head>
+<body>
+    <div class="container">
+        <div class="row">
+            <div class="col-md-4"></div>
+            <div class="col-md-4 mt-5 pt-5">
+                <div class="wrapper mt-5 bg-secondary p-4 rounded">
+                    <h1>Upload Image</h1>
+                    <form method="POST" enctype="multipart/form-data">
+                        <input type="file" name="image" class="btn btn-danger">
+                        <input type="submit" value="Upload" class="btn btn-danger my-1 mx-auto">
+                    </form>
+                    <div class="my-3">
+                        {% if message %}
+                        <h2>{{ message }}</h2>
+                        <img src="/static/uploads/processed.jpg" alt="">
+                        {% endif %}
+                    </div>
+                </div>
+            </div>
+            <div class="col-md-4"></div>
+        </div>
+    </div>
+</body>
+</html>