Update app.py

app.py

@@ -58,28 +58,29 @@ def whatsapp_webhook():
         content_type = request.values.get('MediaContentType0')
 
         if content_type.startswith('image/'):
-            r = requests.get(media_url)
+            r = requests.get(media_url, stream=True)
             
             # Generate a unique filename
             filename = f"{uuid.uuid4()}.jpg"
             filepath = os.path.join(UPLOAD_FOLDER, filename)
-            with open(filepath, 'wb') as out_file:
-                shutil.copyfileobj(r.raw, out_file)
+
             try:
+                with open(filepath, 'wb') as out_file:
+                    shutil.copyfileobj(r.raw, out_file)
                 
-                
-                
-                
-                
-                if ('pest' in incoming_msg) or ('Pest' in incoming_msg):
-                    response_text = predict_pest(filepath)
-                elif ('disease' in incoming_msg) or ('Disease' in incoming_msg):
-                    response_text = predict_disease(filepath)
+                # Check file size and existence
+                if os.path.getsize(filepath) == 0:
+                    response_text = "The image file is empty. Please send a valid image."
                 else:
-                    response_text = "Please specify if you want to detect a pest or a disease."
+                    if 'pest' in incoming_msg:
+                        response_text = predict_pest(filepath)
+                    elif 'disease' in incoming_msg:
+                        response_text = predict_disease(filepath)
+                    else:
+                        response_text = "Please specify if you want to detect a pest or a disease."
             except Exception as e:
                 print(f"Error processing image: {e}")
-                response_text = e
+                response_text = 'Error processing image.'
         else:
             response_text = "The attached file is not an image. Please send an image for classification."
     elif 'bookkeeping' in incoming_msg: