Update app.py

app.py

@@ -12,13 +12,6 @@ app = Flask(__name__)
 UPLOAD_FOLDER = '/code/uploads'
 if not os.path.exists(UPLOAD_FOLDER):
     os.makedirs(UPLOAD_FOLDER)
-
-from inference_sdk import InferenceHTTPClient
-import base64
-
-
-
-
     
 
 # Initialize the Flask app
@@ -28,13 +21,6 @@ client = Client(account_sid, auth_token)
 # WhatsApp number to send messages from (your Twilio number)
 from_whatsapp_number = 'whatsapp:+14155238886'
 
-def classify_pest(image_path):
-    # Implement pest classification model here
-    return f"Detected Pest: [Pest Name] for image at {image_path}"
-
-def classify_disease(image_path):
-    # Implement disease classification model here
-    return f"Detected Disease: [Disease Name] for image at {image_path}"
 @app.route('/whatsapp', methods=['POST'])
 def whatsapp_webhook():
     incoming_msg = request.values.get('Body', '').lower()
@@ -48,27 +34,18 @@ def whatsapp_webhook():
         content_type = request.values.get('MediaContentType0')
 
         if content_type.startswith('image/'):
-            '''r = requests.get(media_url)
-            r.raise_for_status()'''
-            response_text=media_url
             
             
-            filepath = os.path.join(UPLOAD_FOLDER, filename)
+            
+            
+            filepath = convert_img(media_url, account_sid, auth_token)
 
-            if 1==1:
-                with open(filepath, 'wb') as out_file:
-                    out_file.write(r.content)
-                
-                # Check file size and existence
-                if os.path.getsize(filepath) == 0:
-                    response_text = "The image file is empty. Please send a valid image."
-                else:
-                    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."'''
+       
+        
+            
+            response_text = predict_disease(filepath)
+            else:
+                response_text = "Please specify if you want to detect a pest or a disease."
             
         else:
             response_text = "The attached file is not an image. Please send an image for classification."