from flask import Flask, render_template, request, jsonify
import os
from werkzeug.utils import secure_filename

# Folder untuk menyimpan file yang diunggah
UPLOAD_FOLDER = 'static/uploads'
ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}

# Fungsi untuk memeriksa format file yang diunggah
def allowed_file(filename):
    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS

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

# Halaman dasar
@app.route('/')
def base():
    return render_template('base.html')

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

# Halaman aplikasi untuk mendeteksi kerusakan
@app.route('/damageapp')
def damageapp():
    return render_template('damageapp.html')

# Fungsi untuk memproses unggahan dan memprediksi kerusakan
@app.route('/damage', methods=['GET', 'POST'])
def damage():
    if request.method == 'POST':
        
       

        # Simpan file jika formatnya diizinkan
        if file and allowed_file(file.filename):
            filename = secure_filename(file.filename)
            filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)

            # Buat folder jika belum ada
            if not os.path.exists(app.config['UPLOAD_FOLDER']):
                os.makedirs(app.config['UPLOAD_FOLDER'])
            
            file.save(filepath)

            # Panggil model prediksi di sini
            # Anda perlu mengganti bagian ini dengan kode model prediksi Anda
            # Contoh: hasil = model.predict(filepath)
            hasil = "Tidak ada kerusakan"  # Ganti dengan hasil prediksi model