backend.py

from flask import Flask, request, jsonify
import os
import sqlite3
from PIL import Image
import pytesseract
import requests
import datetime

app = Flask(__name__)
UPLOAD_FOLDER = 'uploads'
DB_FILE = 'tag_keeper.db'

# Ensure the upload folder exists
if not os.path.exists(UPLOAD_FOLDER):
    os.makedirs(UPLOAD_FOLDER)

# Initialize SQLite database
def init_db():
    with sqlite3.connect(DB_FILE) as conn:
        cursor = conn.cursor()
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS tags (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                extracted_text TEXT,
                image_path TEXT,
                timestamp TEXT
            )
        ''')
        conn.commit()

# Process image and extract text
def process_image(image_path):
    image = Image.open(image_path)
    image = image.convert('L')
    text = pytesseract.image_to_string(image)
    return text.strip()

# Store extracted data into SQLite
def store_tag_data(text, image_path):
    timestamp = datetime.datetime.now().isoformat()
    with sqlite3.connect(DB_FILE) as conn:
        cursor = conn.cursor()
        cursor.execute("INSERT INTO tags (extracted_text, image_path, timestamp) VALUES (?, ?, ?)",
                       (text, image_path, timestamp))
        conn.commit()
    return cursor.lastrowid

@app.route('/upload', methods=['POST'])
def upload_image():
    if 'image' not in request.files:
        return jsonify({"error": "No image provided"}), 400
    
    file = request.files['image']
    if file.filename == '':
        return jsonify({"error": "No file selected"}), 400
    
    filepath = os.path.join(UPLOAD_FOLDER, file.filename)
    file.save(filepath)
    
    # Process the image using OCR
    tag_text = process_image(filepath)
    
    # Store data in SQLite
    tag_id = store_tag_data(tag_text, filepath)
    
    return jsonify({"tag_id": tag_id, "tag_text": tag_text, "image_path": filepath})

@app.route('/tags', methods=['GET'])
def get_tags():
    with sqlite3.connect(DB_FILE) as conn:
        cursor = conn.cursor()
        cursor.execute("SELECT id, extracted_text, image_path, timestamp FROM tags ORDER BY timestamp DESC")
        tags = cursor.fetchall()
    
    return jsonify([{ "id": row[0], "extracted_text": row[1], "image_path": row[2], "timestamp": row[3] } for row in tags])

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