app.py

import os
from flask import Flask, request, redirect, url_for, render_template, flash
from werkzeug.utils import secure_filename
from PIL import Image
import pytesseract
from rankMenu import RankMenu
from evaluateMenu import MenuEvaluator
from readMenu import ReadMenu
import firebase_admin
from firebase_admin import credentials, auth

# Define the upload folder
UPLOAD_FOLDER = './uploads'
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}

app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER

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

# Initialize Firebase Admin SDK
cred = credentials.Certificate('palatemapfirebase.json')
firebase_admin.initialize_app(cred)

app.secret_key = os.urandom(24)  # Generates a random key

def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

def perform_ocr(filepath):
    img = Image.open(filepath)
    text = pytesseract.image_to_string(img)
    return text

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        email = request.form.get('email')
        password = request.form.get('password')

        try:
            if not email:
                raise ValueError("Email is required.")
            if not password:
                raise ValueError("Password is required.")
            
            user = auth.get_user_by_email(email)
            flash("Login successful!", "success")
            return redirect(url_for('homepage'))  # Redirect to homepage after login
        
        except ValueError as ve:
            flash(f"Input Error: {str(ve)}", "danger")

        except firebase_admin._auth_utils.UserNotFoundError:
            flash("No user record found for the provided email.", "danger")

        except Exception as e:
            flash(f"An error occurred: {str(e)}", "danger")

    return render_template('login.html')

@app.route('/register', methods=['GET', 'POST'])
def register():
    if request.method == 'POST':
        email = request.form.get('email')
        password = request.form.get('password')

        if not email or not password:
            flash("Email and password are required.", "danger")
            return redirect(url_for('register'))

        try:
            user = auth.create_user(email=email, password=password)
            flash("Registration successful! Please log in.", "success")
            return redirect(url_for('login'))
        except firebase_admin.auth.EmailAlreadyExistsError:
            flash("Email already in use.", "danger")
        except Exception as e:
            flash(f"Registration failed: {str(e)}", "danger")
    return render_template('register.html')

@app.route('/homepage', methods=['GET', 'POST'])
def homepage():
    if request.method == 'POST':
        # Handle file upload from camera
        if 'file_camera' in request.files:
            file_camera = request.files['file_camera']
            if file_camera.filename != '' and allowed_file(file_camera.filename):
                filename = secure_filename(file_camera.filename)
                filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
                file_camera.save(filepath)
                print(f"Saved camera file: {filename}")
                # Process the file and get results
                try:
                    eval_menu = MenuEvaluator()
                    dish_dict = eval_menu.evaluate_menu(filepath)

                    menuRanker = RankMenu()
                    user_vector = [3, 4, 6, 2, 8, 9]
                    rankedMenu = menuRanker.rankMenu(user_vector, dish_dict)

                    return render_template('result.html', text=rankedMenu, image_url=filepath)
                except Exception as e:
                    print(f"Error processing file: {e}")
                    flash("An error occurred while processing the file.", "danger")
                    return redirect(request.url)

        # Handle file upload from library
        if 'file_library' in request.files:
            file_library = request.files['file_library']
            if file_library.filename != '' and allowed_file(file_library.filename):
                filename = secure_filename(file_library.filename)
                filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
                file_library.save(filepath)
                print(f"Saved library file: {filename}")
                # Process the file and get results
                try:
                    eval_menu = MenuEvaluator()
                    dish_dict = eval_menu.evaluate_menu(filepath)

                    menuRanker = RankMenu()
                    user_vector = [3, 4, 6, 2, 8, 9]
                    rankedMenu = menuRanker.rankMenu(user_vector, dish_dict)

                    return render_template('result.html', text=rankedMenu, image_url=filepath)
                except Exception as e:
                    print(f"Error processing file: {e}")
                    flash("An error occurred while processing the file.", "danger")
                    return redirect(request.url)

    return render_template('homepage.html')

def allowed_file(filename):
    # Check for allowed file extensions
    return '.' in filename and filename.rsplit('.', 1)[1].lower() in {'jpg', 'jpeg', 'png'}



@app.route('/breakfast')
def breakfast():
    return render_template('breakfast.html')

@app.route('/userprofile')
def userprofile():
    return render_template('userprofile.html')

@app.route('/result')
def result():
    return render_template('result.html')

if __name__ == '__main__':
    app.run(host='0.0.0.0', port=5000, debug=True)