Update app.py

app.py

@@ -1,4 +1,3 @@
-```python
 from flask import Flask, render_template, request, jsonify
 import os
 from PIL import Image
@@ -6,6 +5,10 @@ import numpy as np
 
 app = Flask(__name__)
 
+# Ensure uploads directory exists
+if not os.path.exists('uploads'):
+    os.makedirs('uploads')
+
 # Define the HTML template for the Tree Counter page
 @app.route('/')
 def index():
@@ -14,30 +17,55 @@ def index():
 # Define the API endpoint for uploading images
 @app.route('/upload_image', methods=['POST'])
 def upload_image():
+    if 'image' not in request.files:
+        return jsonify({'message': 'No file part'}), 400
+    
     image = request.files['image']
-    image.save('image.jpg')
-    img = Image.open('image.jpg')
-    width, height = img.size
-    if width > 1024 or height > 1024:
-        return jsonify({'message': 'Image is too large. Please crop it.'})
-    else:
-        # Simulate tree detection using YOLO8 model
-        # Replace this with actual model implementation
-        trees = np.random.randint(0, 100)
-        return jsonify({'message': f'Number of Trees: {trees}'})
+    if image.filename == '':
+        return jsonify({'message': 'No selected file'}), 400
+    
+    try:
+        # Save the uploaded image
+        image_path = os.path.join('uploads', 'image.jpg')
+        image.save(image_path)
+        
+        # Open and process the image
+        img = Image.open(image_path)
+        width, height = img.size
+        if width > 1024 or height > 1024:
+            return jsonify({'message': 'Image is too large. Please crop it.'}), 400
+        else:
+            # Simulate tree detection using YOLO8 model
+            # Replace this with actual model implementation
+            trees = np.random.randint(0, 100)
+            return jsonify({'message': f'Number of Trees: {trees}'})
+    except Exception as e:
+        return jsonify({'message': f'Error processing image: {str(e)}'}), 500
 
 # Define the API endpoint for cropping images
 @app.route('/crop_image', methods=['POST'])
 def crop_image():
+    if 'image' not in request.files:
+        return jsonify({'message': 'No file part'}), 400
+    
     image = request.files['image']
-    image.save('image.jpg')
-    img = Image.open('image.jpg')
-    width, height = img.size
-    # Simulate image cropping
-    # Replace this with actual image cropping implementation
-    cropped_img = img.crop((0, 0, 1024, 1024))
-    cropped_img.save('cropped_image.jpg')
-    return jsonify({'message': 'Image cropped successfully'})
+    if image.filename == '':
+        return jsonify({'message': 'No selected file'}), 400
+
+    try:
+        # Save the uploaded image
+        image_path = os.path.join('uploads', 'image.jpg')
+        image.save(image_path)
+        
+        # Open and crop the image
+        img = Image.open(image_path)
+        cropped_img = img.crop((0, 0, min(1024, img.width), min(1024, img.height)))
+        
+        # Save the cropped image
+        cropped_img.save(os.path.join('uploads', 'cropped_image.jpg'))
+        return jsonify({'message': 'Image cropped successfully'})
+    except Exception as e:
+        return jsonify({'message': f'Error cropping image: {str(e)}'}), 500
 
 if __name__ == '__main__':
-    app.run(debug=True)
+    app.run(debug=True)