Update app.py

app.py

@@ -11,6 +11,23 @@ app = Flask(__name__)
 UPLOAD_FOLDER = 'uploads'
 if not os.path.exists(UPLOAD_FOLDER):
     os.makedirs(UPLOAD_FOLDER)
+from inference_sdk import InferenceHTTPClient
+def predict_disease(filepath):    
+    CLIENT = InferenceHTTPClient(
+        api_url="https://classify.roboflow.com",
+        api_key="oF1aC4b1FBCDtK8CoKx7"
+    )
+    
+    result = CLIENT.infer(filepath, model_id="plant-disease-detection-iefbi/1")
+    return result['predicted_classes'][0]
+
+def predict_pest(filepath):
+    CLIENT = InferenceHTTPClient(
+    api_url="https://detect.roboflow.com",
+    api_key="oF1aC4b1FBCDtK8CoKx7")
+    
+    result = CLIENT.infer(filepath, model_id="pest-detection-ueoco/1")
+    return result['predicted_classes'][0]
 
 # Initialize the Flask app
 account_sid = os.environ.get('TWILIO_ACCOUNT_SID')
@@ -51,15 +68,15 @@ def whatsapp_webhook():
                 
                 
                 
-                if 'pest' in incoming_msg:
-                    response_text = classify_pest(filepath)
-                elif 'disease' in incoming_msg:
-                    response_text = classify_disease(filepath)
+                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)
                 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 = filepath
+                response_text = 'Invalid image'
         else:
             response_text = "The attached file is not an image. Please send an image for classification."
     elif 'bookkeeping' in incoming_msg: