Update app.py

app.py

@@ -32,27 +32,36 @@ def classify_disease(image_path):
 def whatsapp_webhook():
     incoming_msg = request.values.get('Body', '').lower()
     sender = request.values.get('From')
-    
+
+    # Check if an image is attached
     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/'):
             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)
-                
-                # Download and save the image using the media URL
-                response = requests.get(media_url)
-                image_data = response.content
-                
+
+                # Download and save the image
                 with open(filepath, 'wb') as f:
-                    f.write(image_data)
-                
-                # Call the appropriate classification function
+                    f.write(media.fetch())
+
+                # Check downloaded image format and size (optional)
+                print(f"Downloaded image: {filepath}, size: {os.path.getsize(filepath)} bytes")
+                try:
+                    Image.open(filepath)  # Basic check if image can be opened
+                except Exception as e:
+                    print(f"Error opening downloaded image: {e}")
+                    response_text = "Failed to process the image. Please try again."
+                    raise  # Re-raise the exception for further handling
+
                 if 'pest' in incoming_msg:
                     response_text = classify_pest(filepath)
                 elif 'disease' in incoming_msg:
@@ -68,9 +77,9 @@ def whatsapp_webhook():
         response_text = "Please provide the details you'd like to record."
     else:
         response_text = get_agricultural_insights(incoming_msg)
-    
+
     send_message(sender, response_text)
-    return '', 204  # Return an empty response to Twilio
+    return '', 204
 
 def get_agricultural_insights(query):
     # Implement your agricultural insights logic here