Update main.py

main.py

@@ -70,34 +70,40 @@ def brain_tumor():
 @app.route('/resultbt', methods=['POST'])
 def resultbt():
     if request.method == 'POST':
-        firstname = request.form['firstname']
-        lastname = request.form['lastname']
-        email = request.form['email']
-        phone = request.form['phone']
-        gender = request.form['gender']
-        age = request.form['age']
         file = request.files['file']
-
         if file and allowed_file(file.filename):
             filename = secure_filename(file.filename)
-            file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
+            filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+
+            # Ensure upload folder exists
+            if not os.path.exists(app.config['UPLOAD_FOLDER']):
+                os.makedirs(app.config['UPLOAD_FOLDER'])
+
+            file.save(filepath)
             flash('Image successfully uploaded and displayed below')
 
-            img = cv2.imread('static/uploads/' + filename)
-            img = crop_imgs([img])
-            img = img.reshape(img.shape[1:])
-            img = preprocess_imgs([img], (224, 224))
+            # Load and preprocess image
+            try:
+                img = cv2.imread(filepath)
+                if img is None:
+                    raise ValueError("Image file is invalid or cannot be read.")
 
-            pred = braintumor_model.predict(img)
-            if pred < 0.5:
-                pred = 0
-            else:
-                pred = 1
+                img = crop_imgs([img])
+                img = img.reshape(img.shape[1:])
+                img = preprocess_imgs([img], (224, 224))
+                img = img.reshape(1, 224, 224, 3)
 
-            return render_template('resultbt.html', filename=filename, fn=firstname, ln=lastname, age=age, r=pred, gender=gender)
+                pred = braintumor_model.predict(img)
+                pred = 1 if pred >= 0.5 else 0
+
+                return render_template('resultbt.html', filename=filename, r=pred)
+            except Exception as e:
+                flash(f"Error processing image: {str(e)}")
+                return redirect(request.url)
         else:
             flash('Allowed image types are - png, jpg, jpeg')
             return redirect(request.url)
 
+
 if __name__ == '__main__':
     app.run(debug=True)