Spaces:

usmanyousaf
/

image-to-text

Sleeping

App Files Files Community

usmanyousaf commited on Dec 1, 2024

Commit

665ac31

verified ·

1 Parent(s): 47b9341

Rename app.py to main.py

Browse files

Files changed (2) hide show

app.py +0 -60
main.py +89 -0

app.py DELETED Viewed

@@ -1,60 +0,0 @@
-from flask import Flask, render_template, request, redirect, url_for
-import cv2
-import pytesseract
-import os
-# Initialize Flask app
-app = Flask(__name__)
-UPLOAD_FOLDER = "static/uploads/"
-app.config["UPLOAD_FOLDER"] = UPLOAD_FOLDER
-# Inform pytesseract where Tesseract is installed
-pytesseract.pytesseract.tesseract_cmd = "/usr/local/bin/tesseract"
-# Ensure upload folder exists
-if not os.path.exists(UPLOAD_FOLDER):
-    os.makedirs(UPLOAD_FOLDER)
-# Home route
-@app.route("/", methods=["GET", "POST"])
-def home():
-    if request.method == "POST":
-        if "image" not in request.files:
-            return redirect(request.url)
-        file = request.files["image"]
-        if file.filename == "":
-            return redirect(request.url)
-        # Save the uploaded file
-        if file:
-            filepath = os.path.join(app.config["UPLOAD_FOLDER"], file.filename)
-            file.save(filepath)
-            return render_template("index.html", image_url=filepath)
-    return render_template("index.html")
-# Process image and extract text
-@app.route("/process", methods=["POST"])
-def process_image():
-    if "image_path" in request.form:
-        image_path = request.form["image_path"]
-        img = cv2.imread(image_path)
-        # Check if the image was read successfully
-        if img is None:
-            error_message = "Error: Unable to read the uploaded image. Please upload a valid image."
-            return render_template("index.html", error=error_message)
-        # Extract text from the image
-        extracted_text = pytesseract.image_to_string(img)
-        # Pass extracted text to the result template
-        return render_template("index.html", image_url=image_path, text=extracted_text)
-    # Handle missing form data case
-    error_message = "Error: No image path provided for processing. Please upload an image first."
-    return render_template("index.html", error=error_message)
-# Run Flask app
-if __name__ == "__main__":
-    app.run(debug=True)

main.py ADDED Viewed

	@@ -0,0 +1,89 @@

+from fastapi import FastAPI, File, UploadFile, Form
+from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+import cv2
+import pytesseract
+import os
+# Initialize FastAPI app
+app = FastAPI()
+# Define upload folder
+UPLOAD_FOLDER = "static/uploads/"
+if not os.path.exists(UPLOAD_FOLDER):
+    os.makedirs(UPLOAD_FOLDER)
+# Inform pytesseract where Tesseract is installed
+pytesseract.pytesseract.tesseract_cmd = "/usr/local/bin/tesseract"
+# Serve static files (images, etc.)
+app.mount("/static", StaticFiles(directory="static"), name="static")
+# Home route
+@app.get("/", response_class=HTMLResponse)
+async def home():
+    html_content = """
+    <!DOCTYPE html>
+    <html lang="en">
+    <head>
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <title>Image to Text Converter</title>
+        <style>
+            body { font-family: Arial, sans-serif; background-color: #000000; color: #333; margin: 0; padding: 0; display: flex; justify-content: center; align-items: center; height: 100vh; }
+            .container { text-align: center; background: #fff; padding: 30px; border-radius: 10px; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1); width: 90%; max-width: 600px; }
+            h1 { font-size: 24px; color: #4CAF50; }
+            p { color: #666; }
+            .upload-box { margin: 20px auto; padding: 30px; border: 2px dashed #ccc; border-radius: 10px; background-color: #f9f9f9; cursor: pointer; position: relative; }
+            .upload-box:hover { background-color: #f4f4f4; }
+            .upload-box span { color: #888; font-size: 14px; display: block; }
+            .upload-box input[type="file"] { position: absolute; width: 100%; height: 100%; top: 0; left: 0; opacity: 0; cursor: pointer; }
+            .process-button { background-color: #4CAF50; color: white; padding: 10px 20px; border: none; border-radius: 5px; cursor: pointer; font-size: 16px; }
+            .process-button:hover { background-color: #45a049; }
+            .result-box { margin-top: 20px; text-align: left; }
+            .result-box img { max-width: 100%; border-radius: 5px; margin-top: 10px; }
+            .actions { margin-top: 20px; }
+            .actions button { margin: 5px; background-color: #4CAF50; color: white; padding: 8px 16px; border: none; border-radius: 5px; cursor: pointer; font-size: 14px; }
+            .actions button:hover { background-color: #45a049; }
+            pre { background-color: #f4f4f4; padding: 10px; border-radius: 5px; white-space: pre-wrap; word-wrap: break-word; }
+            .error { color: red; font-weight: bold; margin-bottom: 20px; }
+        </style>
+    </head>
+    <body>
+        <div class="container">
+            <h1>Image to Text Converter</h1>
+            <p>Quickly extract text from your uploaded images!</p>
+            <form action="/upload_image/" method="POST" enctype="multipart/form-data">
+                <div class="upload-box">
+                    <span>Drag & Drop the Images<br>Or Click to Browse</span>
+                    <input type="file" name="image" accept="image/*" id="image-file">
+                </div>
+                <button class="process-button" type="submit">Upload Image</button>
+            </form>
+        </div>
+    </body>
+    </html>
+    """
+    return HTMLResponse(content=html_content)
+# Upload image route
+@app.post("/upload_image/")
+async def upload_image(image: UploadFile = File(...)):
+    file_location = f"{UPLOAD_FOLDER}/{image.filename}"
+    with open(file_location, "wb") as file:
+        file.write(await image.read())
+    return {"image_url": f"/static/uploads/{image.filename}"}
+# Process image and extract text
+@app.post("/process_image/")
+async def process_image(image_path: str = Form(...)):
+    img = cv2.imread(image_path)
+    if img is None:
+        return {"error": "Unable to read the uploaded image. Please upload a valid image."}
+    # Extract text from the image
+    extracted_text = pytesseract.image_to_string(img)
+    return {"extracted_text": extracted_text}