Spaces:
Sleeping
Sleeping
from flask import Flask, request | |
from twilio.twiml.messaging_response import MessagingResponse | |
from twilio.rest import Client | |
import os | |
from PIL import Image | |
import io | |
app = Flask(__name__) | |
UPLOAD_FOLDER = 'uploads' | |
if not os.path.exists(UPLOAD_FOLDER): | |
os.makedirs(UPLOAD_FOLDER) | |
# Initialize the Flask app | |
account_sid = os.environ.get('TWILIO_ACCOUNT_SID') | |
auth_token = os.environ.get('TWILIO_AUTH_TOKEN') | |
client = Client(account_sid, auth_token) | |
# WhatsApp number to send messages from (your Twilio number) | |
from_whatsapp_number = 'whatsapp:+14155238886' | |
# Placeholder functions for image classification | |
def classify_pest(image): | |
# Implement pest classification model here | |
return "Detected Pest: [Pest Name]" | |
def classify_disease(image): | |
# Implement disease classification model here | |
return "Detected Disease: [Disease Name]" | |
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') | |
content_type = request.values.get('MediaContentType0') | |
if content_type.startswith('image/'): | |
# Download and process the image | |
image_data = client.request(media_url) | |
image = Image.open(io.BytesIO(image_data.content)) | |
if 'pest' in incoming_msg: | |
response_text = classify_pest(image) | |
elif 'disease' in incoming_msg: | |
response_text = classify_disease(image) | |
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." | |
elif 'bookkeeping' in incoming_msg: | |
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 | |
def get_agricultural_insights(query): | |
# Implement your agricultural insights logic here | |
return query | |
def send_message(to, body): | |
try: | |
message = client.messages.create( | |
from_=from_whatsapp_number, | |
body=body, | |
to=to | |
) | |
print(f"Message sent with SID: {message.sid}") | |
except Exception as e: | |
print(f"Error sending message: {e}") | |
# Function to send an initial message | |
def send_initial_message(to_number): | |
send_message( | |
f'whatsapp:{to_number}', | |
'Welcome to the Agri AI Chatbot! How can I assist you today? You can send an image with "pest" or "disease" to classify it.' | |
) | |
if __name__ == '__main__': | |
# Send an initial message when the script starts | |
send_initial_message('916382792828') # Replace with the recipient's number | |
# Start the Flask server | |
app.run(host='0.0.0.0', port=7860) |