Update app.py

app.py

@@ -23,12 +23,10 @@ from_whatsapp_number = 'whatsapp:+14155238886'
 # Placeholder functions for image classification
 def classify_pest(image_path):
     # Implement pest classification model here
-    # This function should now accept a file path instead of an image object
     return f"Detected Pest: [Pest Name] for image at {image_path}"
 
 def classify_disease(image_path):
     # Implement disease classification model here
-    # This function should now accept a file path instead of an image object
     return f"Detected Disease: [Disease Name] for image at {image_path}"
 
 @app.route('/whatsapp', methods=['POST'])
@@ -40,20 +38,21 @@ def whatsapp_webhook():
     num_media = int(request.values.get('NumMedia', 0))
     
     if num_media > 0:
-        media_url = request.values.get('MediaUrl0')
+        media_sid = request.values.get('MediaSid0')
         content_type = request.values.get('MediaContentType0')
         
         if content_type.startswith('image/'):
-            # Download and save the image
-            response = requests.get(media_url)
-            if response.status_code == 200:
+            try:
+                # Retrieve the media using Twilio client
+                media = client.messages.media(request.values.get('MessageSid')).list()[0]
+                
                 # Generate a unique filename
                 filename = f"{uuid.uuid4()}.jpg"
                 filepath = os.path.join(UPLOAD_FOLDER, filename)
                 
-                # Save the image
+                # Download and save the image
                 with open(filepath, 'wb') as f:
-                    f.write(response.content)
+                    f.write(media.fetch())
                 
                 if 'pest' in incoming_msg:
                     response_text = classify_pest(filepath)
@@ -61,8 +60,9 @@ def whatsapp_webhook():
                     response_text = classify_disease(filepath)
                 else:
                     response_text = "Please specify if you want to detect a pest or a disease."
-            else:
-                response_text = "Failed to download the image. Please try again."
+            except Exception as e:
+                print(f"Error processing image: {e}")
+                response_text = "Failed to process the image. Please try again."
         else:
             response_text = "The attached file is not an image. Please send an image for classification."
     elif 'bookkeeping' in incoming_msg: